多维表格
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

141 lines
4.5 KiB

import { knex, Knex } from 'knex';
import { promises as fs } from 'fs';
import { getKnexConfig } from '../tests/utils/config';
async function dropAndCreateDb(kn: Knex, dbName: string) {
await kn.raw(`DROP DATABASE IF EXISTS ??`, [dbName]);
await kn.raw(`CREATE DATABASE ??`, [dbName]);
}
export async function initializeSakilaPg(database: string) {
{
const kn = knex(getKnexConfig({ dbName: 'postgres', dbType: 'pg' }));
await dropAndCreateDb(kn, database);
await kn.destroy();
}
{
const kn = knex(getKnexConfig({ dbName: database, dbType: 'pg' }));
const testsDir = __dirname.replace('/tests/playwright/setup', '/packages/nocodb/tests');
const schemaFile = await fs.readFile(`${testsDir}/pg-sakila-db/01-postgres-sakila-schema.sql`);
await kn.raw(schemaFile.toString());
const trx = await kn.transaction();
const dataFile = await fs.readFile(`${testsDir}/pg-sakila-db/02-postgres-sakila-insert-data.sql`);
await trx.raw(dataFile.toString());
await trx.commit();
await kn.destroy();
}
}
export async function resetSakilaPg(database: string) {
try {
await initializeSakilaPg(database);
} catch (e) {
console.error(`Error resetting pg sakila db: Worker ${database}`, e);
}
}
export async function createTableWithDateTimeColumn(database: string, dbName: string, setTz = false) {
if (database === 'pg') {
{
const pgknex = knex(getKnexConfig({ dbName: 'postgres', dbType: 'pg' }));
await dropAndCreateDb(pgknex, dbName);
await pgknex.destroy();
}
{
const pgknex = knex(getKnexConfig({ dbName, dbType: 'pg' }));
try {
await pgknex.raw(`
CREATE TABLE my_table (
title serial PRIMARY KEY,
datetime_without_tz TIMESTAMP WITHOUT TIME ZONE,
datetime_with_tz TIMESTAMP WITH TIME ZONE
);
-- SET timezone = 'Asia/Hong_Kong';
-- SELECT pg_sleep(1);
INSERT INTO my_table (datetime_without_tz, datetime_with_tz)
VALUES
('2023-04-27 10:00:00', '2023-04-27 10:00:00'),
('2023-04-27 10:00:00+05:30', '2023-04-27 10:00:00+05:30');
`);
} catch (e) {
console.error(`Error resetting pg sakila db: Worker ${dbName}`);
}
await pgknex.destroy();
}
} else if (database === 'mysql') {
{
const mysqlknex = knex(getKnexConfig({ dbName: 'sakila', dbType: 'mysql' }));
await dropAndCreateDb(mysqlknex, dbName);
if (setTz) {
await mysqlknex.raw(`SET GLOBAL time_zone = '+08:00'`);
// wait for 1 second for the timezone to be set
await mysqlknex.raw(`SELECT SLEEP(1)`);
}
await mysqlknex.destroy();
}
{
const mysqlknex = knex(getKnexConfig({ dbName, dbType: 'mysql' }));
try {
await mysqlknex.raw(`
CREATE TABLE my_table (
title INT AUTO_INCREMENT PRIMARY KEY,
datetime_without_tz DATETIME,
datetime_with_tz TIMESTAMP
);
INSERT INTO my_table (datetime_without_tz, datetime_with_tz)
VALUES
('2023-04-27 10:00:00', '2023-04-27 10:00:00'),
('2023-04-27 10:00:00+05:30', '2023-04-27 10:00:00+05:30');
`);
} catch (e) {
console.error(`Error resetting mysql sakila db: Worker ${dbName}`);
}
await mysqlknex.destroy();
}
} else if (database === 'sqlite') {
const sqliteknex = knex(getKnexConfig({ dbName, dbType: 'sqlite' }));
try {
await sqliteknex.raw(`DROP TABLE IF EXISTS my_table`);
await sqliteknex.raw(`
CREATE TABLE my_table (
title INTEGER PRIMARY KEY AUTOINCREMENT,
datetime_without_tz DATETIME,
datetime_with_tz DATETIME )`);
const datetimeData = [
['2023-04-27 10:00:00', '2023-04-27 10:00:00'],
['2023-04-27 10:00:00+05:30', '2023-04-27 10:00:00+05:30'],
];
for (const [datetime_without_tz, datetime_with_tz] of datetimeData) {
await sqliteknex.raw(
`
INSERT INTO my_table (datetime_without_tz, datetime_with_tz)
VALUES (?, ?)`,
[datetime_without_tz, datetime_with_tz]
);
}
} catch (e) {
console.error(`Error resetting sqlite sakila db: Worker ${dbName}`);
}
await sqliteknex.destroy();
}
}
export async function mysqlTz() {
const mysqlknex = knex(getKnexConfig({ dbName: 'sakila', dbType: 'mysql' }));
await mysqlknex.raw(`SET GLOBAL time_zone = '+00:00'`);
await mysqlknex.destroy();
}