mirror of
https://github.com/ToolJet/ToolJet
synced 2026-05-24 09:28:31 +00:00
39 lines
1.3 KiB
TypeScript
39 lines
1.3 KiB
TypeScript
/**
|
|||
* Truncates all test DB tables (except migrations/instance_settings).
|
|||
* Shared by: jest-global-setup.ts, run-e2e.sh pre-reset.
|
|||
*
|
|||
* Usage:
|
|||
* npx ts-node -r tsconfig-paths/register --transpile-only scripts/truncate-test-db.ts
|
|||
*/
|
|||
require('dotenv').config({ path: require('path').resolve(__dirname, '../../.env.test') });
|
|||
|
|||
const { Client } = require('pg');
|
|||
|
|||
(async () => {
|
|||
const client = new Client({
|
|||
host: process.env.PG_HOST || 'localhost',
|
|||
port: Number(process.env.PG_PORT) || 5432,
|
|||
user: process.env.PG_USER,
|
|||
password: process.env.PG_PASS || '',
|
|||
database: process.env.PG_DB,
|
|||
});
|
|||
await client.connect();
|
|||
|
|||
const { rows } = await client.query(
|
|||
`SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'`
|
|||
);
|
|||
const skip = new Set(['instance_settings', 'migrations', 'typeorm_metadata']);
|
|||
const tables = rows
|
|||
.map((r: { table_name: string }) => r.table_name)
|
|||
.filter((t: string) => !skip.has(t))
|
|||
.map((t: string) => `"${t}"`);
|
|||
|
|||
if (tables.length) {
|
|||
await client.query(`TRUNCATE ${tables.join(', ')} RESTART IDENTITY CASCADE`);
|
|||
}
|
|||
await client.query(`UPDATE "instance_settings" SET value='true' WHERE key='ALLOW_PERSONAL_WORKSPACE'`);
|
|||
await client.end();
|
|||
|
|||
console.log(`Truncated ${tables.length} tables`);
|
|||
})();
|
