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';

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

    // delete old test database
    if (fs.existsSync(config.db)) fs.unlinkSync(config.db);
    if (fs.existsSync(config.privateDB)) fs.unlinkSync(config.privateDB);

    await initDb();

    const dbMode = config.mysql ? 'mysql'
        : config.postgres ? 'postgres'
        : 'sqlite';
    Logger.info('Database Mode: ' + dbMode);

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

    const testDir = './test/cases';

    // Add each .ts file to the mocha instance
    fs.readdirSync(testDir)
        .filter(function(file) {
            // Only keep the .ts files
            return file.substr(-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();
                process.exitCode = failures ? 1 : 0; // exit with non-zero status if there were failures
            });
        });
    });
}

init();