SponsorBlockServer/test/cases/tokenUtils.ts

import assert from "assert";
import { config } from "../../src/config";
import axios from "axios";
import * as tokenUtils from "../../src/utils/tokenUtils";
import MockAdapter from "axios-mock-adapter";
import { validateLicenseKeyRegex } from "../../src/routes/verifyToken";
let mock: MockAdapter;
import * as patreon from "../mocks/patreonMock";

const validateToken = validateLicenseKeyRegex;

describe("tokenUtils test", function() {
    before(function() {
        mock = new MockAdapter(axios, { onNoMatch: "throwException" });
        mock.onPost("https://www.patreon.com/api/oauth2/token").reply(200, patreon.fakeOauth);
        mock.onGet(/identity/).reply(200, patreon.activeIdentity);
    });

    it("Should be able to create patreon token", async function () {
        if (!config?.patreon) return this.skip();
        const licenseKey = await tokenUtils.createAndSaveToken(tokenUtils.TokenType.patreon, "test_code");
        assert.ok(validateToken(licenseKey[0]));
    });
    it("Should be able to create local token", async () => {
        const licenseKey = await tokenUtils.createAndSaveToken(tokenUtils.TokenType.local);
        assert.ok(validateToken(licenseKey[0]));
    });
    it("Should be able to get patreon identity", async function () {
        if (!config?.patreon) return this.skip();
        const result = await tokenUtils.getPatreonIdentity("fake_access_token");
        assert.deepEqual(result, patreon.activeIdentity);
    });
    it("Should be able to refresh token", async function () {
        if (!config?.patreon) return this.skip();
        const result = await tokenUtils.refreshToken(tokenUtils.TokenType.patreon, "fake-licence-Key", "fake_refresh_token");
        assert.strictEqual(result, true);
    });

    after(function () {
        mock.restore();
    });
});

describe("tokenUtils failing tests", function() {
    before(function() {
        mock = new MockAdapter(axios, { onNoMatch: "throwException" });
        mock.onPost("https://www.patreon.com/api/oauth2/token").reply(204, patreon.fakeOauth);
        mock.onGet(/identity/).reply(204, patreon.activeIdentity);
    });

    it("Should fail if patreon is not correctly stubbed", async () => {
        const licenseKey = await tokenUtils.createAndSaveToken(tokenUtils.TokenType.patreon, "test_code");
        assert.strictEqual(licenseKey, null);
    });
    it("Should fail if token type is invalid", async () => {
        const licenseKey = await tokenUtils.createAndSaveToken("invalidTokenType" as tokenUtils.TokenType);
        assert.strictEqual(licenseKey, null);
    });

    after(function () {
        mock.restore();
    });
});