wehub-resource-sync
62 lines
1.5 KiB
TypeScript
62 lines
1.5 KiB
TypeScript
import retry from 'async-retry';
|
|
import type { MySql2Database } from 'drizzle-orm/mysql2';
|
|
import { drizzle } from 'drizzle-orm/mysql2';
|
|
import * as mysql from 'mysql2/promise';
|
|
import { afterAll, beforeAll, beforeEach } from 'vitest';
|
|
import { createDockerDB, tests } from './mysql-common';
|
|
import { TestCache, TestGlobalCache, tests as cacheTests } from './mysql-common-cache';
|
|
|
|
const ENABLE_LOGGING = false;
|
|
|
|
let db: MySql2Database;
|
|
let dbGlobalCached: MySql2Database;
|
|
let cachedDb: MySql2Database;
|
|
let client: mysql.Connection;
|
|
|
|
beforeAll(async () => {
|
|
let connectionString;
|
|
if (process.env['MYSQL_CONNECTION_STRING']) {
|
|
connectionString = process.env['MYSQL_CONNECTION_STRING'];
|
|
} else {
|
|
const { connectionString: conStr } = await createDockerDB();
|
|
connectionString = conStr;
|
|
}
|
|
client = await retry(async () => {
|
|
client = await mysql.createConnection({
|
|
uri: connectionString!,
|
|
supportBigNumbers: true,
|
|
});
|
|
await client.connect();
|
|
return client;
|
|
}, {
|
|
retries: 20,
|
|
factor: 1,
|
|
minTimeout: 250,
|
|
maxTimeout: 250,
|
|
randomize: false,
|
|
onRetry() {
|
|
client?.end();
|
|
},
|
|
});
|
|
db = drizzle(client, { logger: ENABLE_LOGGING });
|
|
cachedDb = drizzle(client, { logger: ENABLE_LOGGING, cache: new TestCache() });
|
|
dbGlobalCached = drizzle(client, { logger: ENABLE_LOGGING, cache: new TestGlobalCache() });
|
|
});
|
|
|
|
afterAll(async () => {
|
|
await client?.end();
|
|
});
|
|
|
|
beforeEach((ctx) => {
|
|
ctx.mysql = {
|
|
db,
|
|
};
|
|
ctx.cachedMySQL = {
|
|
db: cachedDb,
|
|
dbGlobalCached,
|
|
};
|
|
});
|
|
|
|
cacheTests();
|
|
tests();
|