SponsorBlockServer/test/utils/reset.ts
2023-02-22 01:15:49 -05:00

22 lines
No EOL
826 B
TypeScript

// drop postgres tables
// reset redis cache
import { config } from "../../src/config";
import { createClient } from "redis";
import { Pool } from "pg";
import { Logger } from "../../src/utils/logger";
export async function resetRedis() {
if (config?.redis?.enabled && config.mode === "test") {
const client = createClient(config.redis);
await client.connect();
await client.flushAll();
}
}
export async function resetPostgres() {
if (process.env.TEST_POSTGRES && config.mode == "test" && config.postgres) {
const pool = new Pool({ ...config.postgres });
await pool.query(`DROP DATABASE IF EXISTS "sponsorTimes"`);
await pool.query(`DROP DATABASE IF EXISTS "privateDB"`);
await pool.end().catch(err => Logger.error(`closing db (postgres): ${err}`));
}
}