122 lines
4.4 KiB
TypeScript
122 lines
4.4 KiB
TypeScript
/*
|
|
* This file is a part of "NMIG" - the database migration tool.
|
|
*
|
|
* Copyright (C) 2016 - present, Anatoly Khaytovich <anatolyuss@gmail.com>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program (please see the "LICENSE.md" file).
|
|
* If not, see <http://www.gnu.org/licenses/gpl.txt>.
|
|
*
|
|
* @author Anatoly Khaytovich <anatolyuss@gmail.com>
|
|
*/
|
|
import TestSchemaProcessor from './TestSchemaProcessor';
|
|
import Conversion from '../../src/Conversion';
|
|
import DBAccess from '../../src/DBAccess';
|
|
import DBVendors from '../../src/DBVendors';
|
|
import DBAccessQueryResult from '../../src/DBAccessQueryResult';
|
|
import { Test } from 'tape';
|
|
|
|
/**
|
|
* Returns `table_a` column types.
|
|
*/
|
|
async function getColumnTypes(testSchemaProcessor: TestSchemaProcessor): Promise<any[]> {
|
|
const logTitle: string = 'ColumnTypesTest::getColumnTypes';
|
|
const sql: string = `SELECT column_name, data_type
|
|
FROM information_schema.columns
|
|
WHERE table_catalog = '${ (<Conversion>testSchemaProcessor.conversion)._targetConString.database }'
|
|
AND table_schema = '${ (<Conversion>testSchemaProcessor.conversion)._schema }'
|
|
AND table_name = 'table_a';`;
|
|
|
|
const result: DBAccessQueryResult = await (<DBAccess>testSchemaProcessor.dbAccess).query(
|
|
logTitle,
|
|
sql,
|
|
DBVendors.PG,
|
|
false,
|
|
false
|
|
);
|
|
|
|
if (result.error) {
|
|
await testSchemaProcessor.processFatalError(result.error);
|
|
}
|
|
|
|
return result.data.rows;
|
|
}
|
|
|
|
/**
|
|
* Returns expected column types map.
|
|
*/
|
|
function getExpectedColumnTypes(): Map<string, string> {
|
|
return new Map<string, string>([
|
|
['id_test_sequence', 'bigint'],
|
|
['id_test_unique_index', 'integer'],
|
|
['id_test_composite_unique_index_1', 'integer'],
|
|
['id_test_composite_unique_index_2', 'integer'],
|
|
['id_test_index', 'integer'],
|
|
['int_test_not_null', 'integer'],
|
|
['id_test_composite_index_1', 'integer'],
|
|
['id_test_composite_index_2', 'integer'],
|
|
['json_test_comment', 'json'],
|
|
['bit', 'bit varying'],
|
|
['year', 'smallint'],
|
|
['tinyint_test_default', 'smallint'],
|
|
['smallint', 'smallint'],
|
|
['mediumint', 'integer'],
|
|
['bigint', 'bigint'],
|
|
['float', 'real'],
|
|
['double', 'double precision'],
|
|
['double_precision', 'double precision'],
|
|
['numeric', 'numeric'],
|
|
['decimal', 'numeric'],
|
|
['decimal_19_2', 'numeric'],
|
|
['char_5', 'character'],
|
|
['varchar_5', 'character varying'],
|
|
['date', 'date'],
|
|
['time', 'time without time zone'],
|
|
['datetime', 'timestamp without time zone'],
|
|
['timestamp', 'timestamp without time zone'],
|
|
['enum', 'character varying'],
|
|
['set', 'character varying'],
|
|
['tinytext', 'text'],
|
|
['mediumtext', 'text'],
|
|
['longtext', 'text'],
|
|
['text', 'text'],
|
|
['blob', 'bytea'],
|
|
['longblob', 'bytea'],
|
|
['mediumblob', 'bytea'],
|
|
['tinyblob', 'bytea'],
|
|
['varbinary', 'bytea'],
|
|
['binary', 'bytea']
|
|
]);
|
|
}
|
|
|
|
/**
|
|
* The data content testing.
|
|
*/
|
|
export default async function(testSchemaProcessor: TestSchemaProcessor, tape: Test): Promise<void> {
|
|
const data: any[] = await getColumnTypes(testSchemaProcessor);
|
|
const expectedColumnTypesMap: Map<string, string> = getExpectedColumnTypes();
|
|
const autoTimeoutMs: number = 3 * 1000; // 3 seconds.
|
|
const numberOfPlannedAssertions: number = data.length;
|
|
|
|
tape.plan(numberOfPlannedAssertions);
|
|
tape.timeoutAfter(autoTimeoutMs);
|
|
|
|
for (let i: number = 0; i < numberOfPlannedAssertions; ++i) {
|
|
const columnName: string = data[i].column_name;
|
|
const actualColumnType: string = data[i].data_type;
|
|
const expectedColumnType: string = <string>expectedColumnTypesMap.get(columnName);
|
|
|
|
tape.comment(`Test ${ columnName } column type`);
|
|
tape.equal(actualColumnType, expectedColumnType);
|
|
}
|
|
}
|