mirror of
https://github.com/dbgate/dbgate
synced 2024-11-07 20:26:23 +00:00
125 lines
3.9 KiB
JavaScript
125 lines
3.9 KiB
JavaScript
const stableStringify = require('json-stable-stringify');
|
|
const _ = require('lodash');
|
|
const fp = require('lodash/fp');
|
|
const uuidv1 = require('uuid/v1');
|
|
const { testWrapper } = require('../tools');
|
|
const engines = require('../engines');
|
|
const { getAlterTableScript, extendDatabaseInfo, generateDbPairingId } = require('dbgate-tools');
|
|
|
|
function pickImportantTableInfo(table) {
|
|
return {
|
|
pureName: table.pureName,
|
|
columns: table.columns
|
|
.filter(x => x.columnName != 'rowid')
|
|
.map(fp.pick(['columnName', 'notNull', 'autoIncrement'])),
|
|
};
|
|
}
|
|
|
|
function checkTableStructure(t1, t2) {
|
|
// expect(t1.pureName).toEqual(t2.pureName)
|
|
expect(pickImportantTableInfo(t1)).toEqual(pickImportantTableInfo(t2));
|
|
}
|
|
|
|
async function testTableDiff(conn, driver, mangle) {
|
|
await driver.query(conn, `create table t0 (id int not null primary key)`);
|
|
|
|
await driver.query(
|
|
conn,
|
|
`create table t1 (
|
|
col_pk int not null primary key,
|
|
col_std int null,
|
|
col_def int null default 12,
|
|
col_fk int null references t0(id),
|
|
col_idx int null,
|
|
col_uq int null unique,
|
|
col_ref int null unique
|
|
)`
|
|
);
|
|
|
|
await driver.query(conn, `create index idx1 on t1(col_idx)`);
|
|
|
|
await driver.query(conn, `create table t2 (id int not null primary key, fkval int null references t1(col_ref))`);
|
|
|
|
const tget = x => x.tables.find(y => y.pureName == 't1');
|
|
const structure1 = generateDbPairingId(extendDatabaseInfo(await driver.analyseFull(conn)));
|
|
let structure2 = _.cloneDeep(structure1);
|
|
mangle(tget(structure2));
|
|
structure2 = extendDatabaseInfo(structure2);
|
|
|
|
const { sql } = getAlterTableScript(tget(structure1), tget(structure2), {}, structure1, structure2, driver);
|
|
console.log('RUNNING ALTER SQL', driver.engine, ':', sql);
|
|
|
|
await driver.script(conn, sql);
|
|
|
|
const structure2Real = extendDatabaseInfo(await driver.analyseFull(conn));
|
|
|
|
checkTableStructure(tget(structure2Real), tget(structure2));
|
|
// expect(stableStringify(structure2)).toEqual(stableStringify(structure2Real));
|
|
}
|
|
|
|
const TESTED_COLUMNS = ['col_pk', 'col_std', 'col_def', 'col_fk', 'col_ref', 'col_idx', 'col_uq'];
|
|
// const TESTED_COLUMNS = ['col_pk'];
|
|
// const TESTED_COLUMNS = ['col_idx'];
|
|
// const TESTED_COLUMNS = ['col_def'];
|
|
// const TESTED_COLUMNS = ['col_std'];
|
|
// const TESTED_COLUMNS = ['col_ref'];
|
|
|
|
function engines_columns_source() {
|
|
return _.flatten(engines.map(engine => TESTED_COLUMNS.map(column => [engine.label, column, engine])));
|
|
}
|
|
|
|
describe('Alter table', () => {
|
|
test.each(engines.map(engine => [engine.label, engine]))(
|
|
'Add column - %s',
|
|
testWrapper(async (conn, driver, engine) => {
|
|
await testTableDiff(conn, driver, tbl => {
|
|
tbl.columns.push({
|
|
columnName: 'added',
|
|
dataType: 'int',
|
|
pairingId: uuidv1(),
|
|
notNull: false,
|
|
autoIncrement: false,
|
|
});
|
|
});
|
|
})
|
|
);
|
|
|
|
test.each(engines_columns_source())(
|
|
'Drop column - %s - %s',
|
|
testWrapper(async (conn, driver, column, engine) => {
|
|
await testTableDiff(conn, driver, tbl => (tbl.columns = tbl.columns.filter(x => x.columnName != column)));
|
|
})
|
|
);
|
|
|
|
test.each(engines_columns_source())(
|
|
'Change nullability - %s - %s',
|
|
testWrapper(async (conn, driver, column, engine) => {
|
|
await testTableDiff(
|
|
conn,
|
|
driver,
|
|
tbl => (tbl.columns = tbl.columns.map(x => (x.columnName == column ? { ...x, notNull: true } : x)))
|
|
);
|
|
})
|
|
);
|
|
|
|
test.each(engines_columns_source())(
|
|
'Rename column - %s - %s',
|
|
testWrapper(async (conn, driver, column, engine) => {
|
|
await testTableDiff(
|
|
conn,
|
|
driver,
|
|
tbl => (tbl.columns = tbl.columns.map(x => (x.columnName == column ? { ...x, columnName: 'col_renamed' } : x)))
|
|
);
|
|
})
|
|
);
|
|
|
|
test.each(engines.map(engine => [engine.label, engine]))(
|
|
'Drop index - %s',
|
|
testWrapper(async (conn, driver, engine) => {
|
|
await testTableDiff(conn, driver, tbl => {
|
|
tbl.indexes = [];
|
|
});
|
|
})
|
|
);
|
|
});
|