SponsorBlockServer/test/cases/getUserStats.ts

import fetch from "node-fetch";
import {Done, getbaseURL} from "../utils";
import {db} from "../../src/databases/databases";
import {getHash} from "../../src/utils/getHash";
import assert from "assert";

describe("getUserStats", () => {
    before(async () => {
        const insertUserNameQuery = 'INSERT INTO "userNames" ("userID", "userName") VALUES(?, ?)';
        await db.prepare("run", insertUserNameQuery, [getHash("getuserstats_user_01"), "Username user 01"]);

        const sponsorTimesQuery = 'INSERT INTO "sponsorTimes" ("videoID", "startTime", "endTime", "votes", "UUID", "userID", "timeSubmitted", views, category, "shadowHidden") VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?)';
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, 0, "getuserstatsuuid1", getHash("getuserstats_user_01"), 1, 1, "sponsor", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, 0, "getuserstatsuuid2", getHash("getuserstats_user_01"), 2, 2, "selfpromo", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, 0, "getuserstatsuuid3", getHash("getuserstats_user_01"), 3, 3, "interaction", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, 0, "getuserstatsuuid4", getHash("getuserstats_user_01"), 4, 4, "intro", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, 0, "getuserstatsuuid5", getHash("getuserstats_user_01"), 5, 5, "outro", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, 0, "getuserstatsuuid6", getHash("getuserstats_user_01"), 6, 6, "preview", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, 0, "getuserstatsuuid7", getHash("getuserstats_user_01"), 7, 7, "music_offtopic", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 11, 11, 0, "getuserstatsuuid8", getHash("getuserstats_user_01"), 8, 8, "poi_highlight", 0]);
        await db.prepare("run", sponsorTimesQuery, ["getuserstats1", 0, 60, -2, "getuserstatsuuid9", getHash("getuserstats_user_02"), 8, 2, "sponsor", 0]);

    });

    it("Should be able to get a 400 (No userID parameter)", (done: Done) => {
        fetch(`${getbaseURL()}/api/userStats`)
            .then(res => {
                assert.strictEqual(res.status, 400);
                done();
            })
            .catch(err => done(err));
    });

    it("Should be able to get user info", (done: Done) => {
        fetch(`${getbaseURL()}/api/userStats?userID=getuserstats_user_01`)
            .then(async res => {
                assert.strictEqual(res.status, 200);
                const expected = {
                    userName: "Username user 01",
                    userID: getHash("getuserstats_user_01"),
                    sponsor: {
                        minutesSaved: 1, segmentCount: 1,
                    }, selfpromo: {
                        minutesSaved: 2, segmentCount: 1,
                    }, interaction: {
                        minutesSaved: 3, segmentCount: 1,
                    }, intro: {
                        minutesSaved: 4, segmentCount: 1,
                    }, outro: {
                        minutesSaved: 5, segmentCount: 1,
                    }, preview: {
                        minutesSaved: 6, segmentCount: 1,
                    }, music_offtopic: {
                        minutesSaved: 7, segmentCount: 1,
                    }, poi_highlight: {
                        minutesSaved: 0, segmentCount: 1,
                    },
                };
                const data = await res.json();
                assert.deepStrictEqual(data, expected);
                done();
            })
            .catch(err => done(err));
    });

    it("Should be able to get all zeroes for invalid userid", (done: Done) => {
        fetch(`${getbaseURL()}/api/userStats?userID=getuserstats_user_invalid`)
            .then(async res => {
                assert.strictEqual(res.status, 200);
                const data = await res.json();
                for (const value in data) {
                    if (data[value]?.minutesSaved || data[value]?.segmentCount) {
                        done(`returned non-zero for ${value}`);
                    }
                }
                done();
            })
            .catch(err => done(err));
    });

    it("Should be able to get all zeroes for only ignored segments", (done: Done) => {
        fetch(`${getbaseURL()}/api/userStats?userID=getuserstats_user_02`)
            .then(async res => {
                assert.strictEqual(res.status, 200);
                const data = await res.json();
                for (const value in data) {
                    if (data[value]?.minutesSaved || data[value]?.segmentCount) {
                        done(`returned non-zero for ${value}`);
                    }
                }
                done();
            })
            .catch(err => done(err));
    });
});