SponsorBlockServer/test/cases/highLoad.ts

import sinon from "sinon";
import { db } from "../../src/databases/databases";
import assert from "assert";
import { client } from "../utils/httpClient";
client.defaults.validateStatus = (status) => status < 600;

describe("High load test", () => {
    before(() => {
        sinon.stub(db, "highLoad").returns(true);
    });

    after(() => {
        sinon.restore();
    });

    it("Should return 503 on getTopUsers", () =>
        client.get("/api/getTopUsers?sortType=0")
            .then(res => assert.strictEqual(res.status, 503))
    );

    it("Should return 503 on getTopCategoryUsers", () =>
        client.get("/api/getTopCategoryUsers?sortType=0&category=sponsor")
            .then(res => assert.strictEqual(res.status, 503))
    );

    it("Should return 0 on getTotalStats", () =>
        client.get("/api/getTotalStats")
            .then(res => assert.strictEqual(res.status, 200))
    );
});