SponsorBlockServer/test/test.ts

import Mocha from "mocha";
import fs from "fs";
import path from "path";
import { config } from "../src/config";
import { createServer } from "../src/app";
import { createMockServer } from "./mocks";
import { Logger } from "../src/utils/logger";
import { initDb } from "../src/databases/databases";
import { ImportMock } from "ts-mock-imports";
import * as rateLimitMiddlewareModule from "../src/middleware/requestRateLimit";
import rateLimit from "express-rate-limit";
import redis from "../src/utils/redis";
import { resetRedis, resetPostgres } from "./utils/reset";

async function init() {
    ImportMock.mockFunction(rateLimitMiddlewareModule, "rateLimitMiddleware", rateLimit({
        skip: () => true
    }));

    // delete old test database
    if (fs.existsSync(config.db)) fs.unlinkSync(config.db);
    if (fs.existsSync(config.privateDB)) fs.unlinkSync(config.privateDB);
    if (config?.redis?.enabled) await resetRedis();
    if (config?.postgres) await resetPostgres();

    await initDb();

    const dbMode = config.postgres ? "postgres"
        : "sqlite";
    Logger.info(`Database Mode: ${dbMode}`);

    // set commit at headCommit
    (global as any).HEADCOMMIT = "test";

    // Instantiate a Mocha instance.
    const mocha = new Mocha();

    const testDirs = ["./test/cases"];

    // Add each .ts file to the mocha instance
    testDirs.forEach(testDir => {
        fs.readdirSync(testDir)
            .filter((file) =>
                // Only keep the .ts files
                file.slice(-3) === ".ts"
            )
            .forEach(function(file) {
                mocha.addFile(
                    path.join(testDir, file)
                );
            });
    });

    const mockServer = createMockServer(() => {
        Logger.info("Started mock HTTP Server");
        const server = createServer(() => {
            Logger.info("Started main HTTP server");
            // Run the tests.
            mocha.run((failures) => {
                mockServer.close();
                server.close();
                redis.quit();
                process.exitCode = failures ? 1 : 0; // exit with non-zero status if there were failures
                process.exit();
            });
        });
    });
}

init();