Hydro/packages/hydrooj/test/db.ts

import cluster from 'cluster';
import 'hydrooj/src/loader';
import * as bus from 'hydrooj/src/service/bus';

jest.mock('hydrooj/src/service/db');

export async function connect() {
    const db = require('hydrooj/src/service/db');
    await db.start({});
    const scripts = require('hydrooj/src/upgrade').default;
    let dbVer = 0;
    const expected = scripts.length;
    while (dbVer < expected) {
        const func = scripts[dbVer];
        dbVer++;
        if (typeof func !== 'function' || func.toString().includes('_FRESH_INSTALL_IGNORE')) continue;
        // eslint-disable-next-line no-await-in-loop
        await func();
    }
}

export async function dispose() {
    const db = require('hydrooj/src/service/db');
    await db.stop();
    bus.emit('app/exit');
    for (const key in cluster.workers) cluster.workers[key].destroy('SIGINT');
}