mirror of
https://github.com/ajayyy/SponsorBlockServer.git
synced 2025-12-13 06:57:05 +03:00
Merge branch 'master' of https://github.com/ajayyy/SponsorBlockServer into getLockReason
This commit is contained in:
Michael C
29
test/cases/getService.ts
Normal file
29
test/cases/getService.ts
Normal file
@@ -0,0 +1,29 @@
|
||||
import { getService } from "../../src/utils/getService";
|
||||
import { Service } from "../../src/types/segments.model";
|
||||
|
||||
import assert from "assert";
|
||||
|
||||
describe("getService", () => {
|
||||
it("Should return youtube if not match", () => {
|
||||
assert.strictEqual(getService(), Service.YouTube);
|
||||
assert.strictEqual(getService(""), Service.YouTube);
|
||||
assert.strictEqual(getService("test", "not exist"), Service.YouTube);
|
||||
assert.strictEqual(getService(null, null), Service.YouTube);
|
||||
assert.strictEqual(getService(undefined, undefined), Service.YouTube);
|
||||
assert.strictEqual(getService(undefined), Service.YouTube);
|
||||
});
|
||||
|
||||
it("Should return Youtube", () => {
|
||||
assert.strictEqual(getService("youtube"), Service.YouTube);
|
||||
assert.strictEqual(getService(" Youtube "), Service.YouTube);
|
||||
assert.strictEqual(getService(" YouTube "), Service.YouTube);
|
||||
assert.strictEqual(getService(undefined, " YouTube "), Service.YouTube);
|
||||
});
|
||||
|
||||
it("Should return PeerTube", () => {
|
||||
assert.strictEqual(getService("PeerTube"), Service.PeerTube);
|
||||
assert.strictEqual(getService(" PeerTube "), Service.PeerTube);
|
||||
assert.strictEqual(getService(" peertube "), Service.PeerTube);
|
||||
assert.strictEqual(getService(undefined, " PeerTube "), Service.PeerTube);
|
||||
});
|
||||
});
|
||||
124
test/cases/getUserStats.ts
Normal file
124
test/cases/getUserStats.ts
Normal file
@@ -0,0 +1,124 @@
|
||||
import fetch from "node-fetch";
|
||||
import {Done, getbaseURL, partialDeepEquals} 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 all user info", (done: Done) => {
|
||||
fetch(`${getbaseURL()}/api/userStats?userID=getuserstats_user_01&fetchCategoryStats=true&fetchActionTypeStats=true`)
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expected = {
|
||||
userName: "Username user 01",
|
||||
userID: getHash("getuserstats_user_01"),
|
||||
categoryCount: {
|
||||
sponsor: 1,
|
||||
selfpromo: 1,
|
||||
interaction: 1,
|
||||
intro: 1,
|
||||
outro: 1,
|
||||
preview: 1,
|
||||
music_offtopic: 1,
|
||||
poi_highlight: 1,
|
||||
},
|
||||
actionTypeCount: {
|
||||
mute: 0,
|
||||
skip: 8
|
||||
},
|
||||
overallStats: {
|
||||
minutesSaved: 28,
|
||||
segmentCount: 8
|
||||
}
|
||||
};
|
||||
const data = await res.json();
|
||||
assert.ok(partialDeepEquals(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.overallStats) {
|
||||
if (data[value]) {
|
||||
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.overallStats) {
|
||||
if (data[value]) {
|
||||
done(`returned non-zero for ${value}`);
|
||||
}
|
||||
}
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should not get extra stats if not requested", (done: Done) => {
|
||||
fetch(`${getbaseURL()}/api/userStats?userID=getuserstats_user_01`)
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const data = await res.json();
|
||||
// check for categoryCount
|
||||
if (data.categoryCount || data.actionTypeCount) {
|
||||
done("returned extra stats");
|
||||
}
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should get parts of extra stats if not requested", (done: Done) => {
|
||||
fetch(`${getbaseURL()}/api/userStats?userID=getuserstats_user_01&fetchActionTypeStats=true`)
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const data = await res.json();
|
||||
// check for categoryCount
|
||||
if (data.categoryCount && !data.actionTypeCount) {
|
||||
done("returned extra stats");
|
||||
}
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user