dbgate/integration-tests/__tests__/object-analyse.spec.js

const { connect, randomDbName, testWrapper } = require('../tools');
const requireEngineDriver = require('dbgate-api/src/utility/requireEngineDriver');
const engines = require('../engines');
const _ = require('lodash');

const initSql = ['CREATE TABLE t1 (id int)', 'CREATE TABLE t2 (id int)'];

function flatSource() {
  return _.flatten(
    engines.map(engine => (engine.objects || []).map(object => [engine.label, engine, object.type, object]))
  );
}

const obj1Match = expect.objectContaining({
  pureName: 'obj1',
});

describe('Object analyse', () => {
  test.each(flatSource())(
    'Full analysis (%s - %s)',
    testWrapper(async (conn, driver, engine, type, object) => {
      for (const sql of initSql) await driver.query(conn, sql);

      await driver.query(conn, object.create1);
      const structure = await driver.analyseFull(conn);

      expect(structure[type].length).toEqual(1);
      expect(structure[type][0]).toEqual(obj1Match);
    })
  );

  test.each(flatSource())(
    'Incremental analysis - add (%s - %s)',
    testWrapper(async (conn, driver, engine, type, object) => {
      for (const sql of initSql) await driver.query(conn, sql);

      await driver.query(conn, object.create2);
      const structure1 = await driver.analyseFull(conn);
      await driver.query(conn, object.create1);
      const structure2 = await driver.analyseIncremental(conn, structure1);

      expect(structure2[type].length).toEqual(2);
      expect(structure2[type].find(x => x.pureName == 'obj1')).toEqual(obj1Match);
    })
  );

  test.each(flatSource())(
    'Incremental analysis - drop (%s - %s)',
    testWrapper(async (conn, driver, engine, type, object) => {
      for (const sql of initSql) await driver.query(conn, sql);

      await driver.query(conn, object.create1);
      await driver.query(conn, object.create2);
      const structure1 = await driver.analyseFull(conn);
      await driver.query(conn, object.drop2);
      const structure2 = await driver.analyseIncremental(conn, structure1);

      expect(structure2[type].length).toEqual(1);
      expect(structure2[type][0]).toEqual(obj1Match);
    })
  );
});