|
|
|
import { NcContext } from '.';
|
|
|
|
const mysql = require('mysql2');
|
|
|
|
const { Client } = require('pg');
|
|
|
|
|
|
|
|
import { PromisedDatabase } from 'promised-sqlite3';
|
|
|
|
const sqliteDb = new PromisedDatabase();
|
|
|
|
|
|
|
|
const isMysql = (context: NcContext) => context.dbType === 'mysql';
|
|
|
|
|
|
|
|
const isSqlite = (context: NcContext) => context.dbType === 'sqlite';
|
|
|
|
|
|
|
|
const isPg = (context: NcContext) => context.dbType === 'pg';
|
|
|
|
|
|
|
|
// hardwired for hub; this has to be configured to false in nocodb
|
|
|
|
// consider reading this from environment variable
|
|
|
|
const isHub = () => true;
|
|
|
|
const isEE = () => process.env.EE === 'true';
|
|
|
|
|
|
|
|
const pg_credentials = (context: NcContext) => ({
|
|
|
|
user: 'postgres',
|
|
|
|
host: 'localhost',
|
|
|
|
// todo: Hack to resolve issue with pg resetting
|
|
|
|
database: `sakila_${context.workerId}`,
|
|
|
|
password: 'password',
|
|
|
|
port: 5432,
|
|
|
|
});
|
|
|
|
|
|
|
|
const pgExec = async (query: string, context: NcContext) => {
|
|
|
|
// open pg client connection
|
|
|
|
const client = new Client(pg_credentials(context));
|
|
|
|
await client.connect();
|
|
|
|
|
|
|
|
await client.query(query);
|
|
|
|
await client.end();
|
|
|
|
};
|
|
|
|
|
|
|
|
const mysqlExec = async query => {
|
|
|
|
// creates a new mysql connection
|
|
|
|
const connection = mysql.createConnection({
|
|
|
|
host: 'localhost',
|
|
|
|
user: 'root',
|
|
|
|
password: 'password',
|
|
|
|
database: `test_sakila_${process.env.TEST_PARALLEL_INDEX}`,
|
|
|
|
});
|
|
|
|
// start connection to db
|
|
|
|
connection.connect();
|
|
|
|
// exec query + disconnect to db as a Promise
|
|
|
|
return new Promise((resolve, reject) => {
|
|
|
|
connection.query(query, (error, results) => {
|
|
|
|
if (error) reject(error);
|
|
|
|
else {
|
|
|
|
connection.end();
|
|
|
|
// console.log(results)
|
|
|
|
return resolve(results);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
});
|
|
|
|
};
|
|
|
|
|
|
|
|
async function sqliteExec(query) {
|
|
|
|
const parallelIndex = process.env.TEST_PARALLEL_INDEX;
|
|
|
|
const rootProjectDir = __dirname.replace('/tests/playwright/setup', '');
|
|
|
|
await sqliteDb.open(`${rootProjectDir}/packages/nocodb/test_sakila_${parallelIndex}.db`);
|
|
|
|
|
|
|
|
await sqliteDb.run(query);
|
|
|
|
await sqliteDb.close();
|
|
|
|
}
|
|
|
|
|
|
|
|
export { sqliteExec, mysqlExec, isMysql, isSqlite, isPg, pgExec, isHub, isEE };
|