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

const engines = require('../engines');
const requireEngineDriver = require('dbgate-api/src/utility/requireEngineDriver');
const crypto = require('crypto');

global.DBGATE_TOOLS = require('dbgate-tools');

function randomDbName() {
  const generatedKey = crypto.randomBytes(6);
  const newKey = generatedKey.toString('hex');
  return `db${newKey}`;
}

async function connect(engine, database) {
  const { connection } = engine;
  const driver = requireEngineDriver(connection);

  if (engine.generateDbFile) {
    const conn = await driver.connect({
      ...connection,
      databaseFile: `dbtemp/${database}`,
    });
    return conn;
  } else {
    const conn = await driver.connect(connection);
    await driver.query(conn, `CREATE DATABASE ${database}`);
    await driver.close(conn);

    const res = await driver.connect({
      ...connection,
      database,
    });
    return res;
  }
}

describe('Analyse tests', () => {
  test.each(engines.map(engine => [engine.label, engine]))(
    'Table structure - full analysis (%s)',
    async (label, engine) => {
      const conn = await connect(engine, randomDbName());
      const driver = requireEngineDriver(engine.connection);

      await driver.query(conn, 'CREATE TABLE t1 (id int not null primary key)');

      const structure = await driver.analyseFull(conn);

      expect(structure.tables.length).toEqual(1);
      expect(structure.tables[0]).toEqual(
        expect.objectContaining({
          pureName: 't1',
          columns: [
            expect.objectContaining({
              columnName: 'id',
            }),
          ],
        })
      );
      await driver.close(conn);
    }
  );

  test.each(engines.map(engine => [engine.label, engine]))(
    'Table add - incremental analysis (%s)',
    async (label, engine) => {
      const conn = await connect(engine, randomDbName());
      const driver = requireEngineDriver(engine.connection);

      await driver.query(conn, 'CREATE TABLE t1 (id int)');
      const structure1 = await driver.analyseFull(conn);
      await driver.query(conn, 'CREATE TABLE t2 (id2 int not null primary key)');
      const structure2 = await driver.analyseIncremental(conn, structure1);

      expect(structure2.tables.length).toEqual(2);
      expect(structure2.tables.find(x => x.pureName == 't2')).toEqual(
        expect.objectContaining({
          pureName: 't2',
          columns: [
            expect.objectContaining({
              columnName: 'id2',
            }),
          ],
        })
      );
      await driver.close(conn);
    }
  );
});