You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
813 B
TypeScript
29 lines
813 B
TypeScript
import 'hydrooj';
|
|
jest.mock('hydrooj/src/service/db');
|
|
|
|
export async function connect() {
|
|
await new Promise((resolve) => {
|
|
const db = require('hydrooj/src/service/db');
|
|
db.bus.once('connect', resolve);
|
|
});
|
|
const modelSystem = require('hydrooj/src/model/system');
|
|
const scripts = require('hydrooj/src/upgrade');
|
|
let dbVer = (await modelSystem.get('db.ver')) || 0;
|
|
const expected = scripts.length;
|
|
while (dbVer < expected) {
|
|
await scripts[dbVer]();
|
|
dbVer++;
|
|
await modelSystem.set('db.ver', dbVer);
|
|
}
|
|
}
|
|
|
|
export async function dispose() {
|
|
const db = require('hydrooj/src/service/db');
|
|
await Promise.all([
|
|
db.getClient().close(),
|
|
db.getClient2().close(),
|
|
db.getDb().close(),
|
|
db.getDb2().close(),
|
|
]);
|
|
}
|