mirror of
https://github.com/ajayyy/SponsorBlockServer.git
synced 2025-12-08 04:27:09 +03:00
everything after postClearCache
This commit is contained in:
Michael C
@@ -1,46 +1,47 @@
|
||||
import fetch from "node-fetch";
|
||||
import {Done, getbaseURL} from "../utils";
|
||||
import {Done, getbaseURL, postJSON} from "../utils";
|
||||
import {db} from "../../src/databases/databases";
|
||||
import {getHash} from "../../src/utils/getHash";
|
||||
import {IDatabase} from "../../src/databases/IDatabase";
|
||||
import assert from "assert";
|
||||
|
||||
async function dbSponsorTimesAdd(db: IDatabase, videoID: string, startTime: number, endTime: number, UUID: string, category: string) {
|
||||
const votes = 0,
|
||||
userID = 0,
|
||||
timeSubmitted = 0,
|
||||
views = 0,
|
||||
shadowHidden = 0,
|
||||
hashedVideoID = `hash_${UUID}`;
|
||||
await db.prepare("run", `INSERT INTO
|
||||
"sponsorTimes" ("videoID", "startTime", "endTime", "votes", "UUID",
|
||||
"userID", "timeSubmitted", "views", "category", "shadowHidden", "hashedVideoID")
|
||||
VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||
[videoID, startTime, endTime, votes, UUID, userID, timeSubmitted, views, category, shadowHidden, hashedVideoID]);
|
||||
}
|
||||
describe("segmentShift", function () {
|
||||
// functions
|
||||
async function dbSponsorTimesAdd(db: IDatabase, videoID: string, startTime: number, endTime: number, UUID: string, category: string) {
|
||||
const votes = 0,
|
||||
userID = 0,
|
||||
timeSubmitted = 0,
|
||||
views = 0,
|
||||
shadowHidden = 0,
|
||||
hashedVideoID = `hash_${UUID}`;
|
||||
await db.prepare("run", `INSERT INTO
|
||||
"sponsorTimes" ("videoID", "startTime", "endTime", "votes", "UUID",
|
||||
"userID", "timeSubmitted", "views", "category", "shadowHidden", "hashedVideoID")
|
||||
VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||
[videoID, startTime, endTime, votes, UUID, userID, timeSubmitted, views, category, shadowHidden, hashedVideoID]);
|
||||
}
|
||||
|
||||
async function dbSponsorTimesSetByUUID(db: IDatabase, UUID: string, startTime: number, endTime: number) {
|
||||
await db.prepare("run", `UPDATE "sponsorTimes" SET "startTime" = ?, "endTime" = ? WHERE "UUID" = ?`, [startTime, endTime, UUID]);
|
||||
}
|
||||
async function dbSponsorTimesSetByUUID(db: IDatabase, UUID: string, startTime: number, endTime: number) {
|
||||
await db.prepare("run", `UPDATE "sponsorTimes" SET "startTime" = ?, "endTime" = ? WHERE "UUID" = ?`, [startTime, endTime, UUID]);
|
||||
}
|
||||
|
||||
async function dbSponsorTimesCompareExpect(db: IDatabase, expect: any): Promise<void> {
|
||||
for (let i = 0, len = expect.length; i < len; i++) {
|
||||
const expectSeg = expect[i];
|
||||
const seg = await db.prepare("get", `SELECT "startTime", "endTime" FROM "sponsorTimes" WHERE "UUID" = ?`, [expectSeg.UUID]);
|
||||
if ("removed" in expect) {
|
||||
assert.ok(expect.removed);
|
||||
assert.strictEqual(seg.votes, -2);
|
||||
assert.deepStrictEqual(seg, expectSeg);
|
||||
assert.strictEqual(seg.startTime, expectSeg.startTime);
|
||||
assert.strictEqual(seg.endTime, expectSeg.endTime);
|
||||
async function dbSponsorTimesCompareExpect(db: IDatabase, expect: any): Promise<void> {
|
||||
for (let i = 0, len = expect.length; i < len; i++) {
|
||||
const expectSeg = expect[i];
|
||||
const seg = await db.prepare("get", `SELECT "startTime", "endTime" FROM "sponsorTimes" WHERE "UUID" = ?`, [expectSeg.UUID]);
|
||||
if ("removed" in expect) {
|
||||
assert.ok(expect.removed);
|
||||
assert.strictEqual(seg.votes, -2);
|
||||
assert.deepStrictEqual(seg, expectSeg);
|
||||
assert.strictEqual(seg.startTime, expectSeg.startTime);
|
||||
assert.strictEqual(seg.endTime, expectSeg.endTime);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
describe("segmentShift", function () {
|
||||
// constants
|
||||
const privateVipUserID = "VIPUser-segmentShift";
|
||||
const vipUserID = getHash(privateVipUserID);
|
||||
const baseURL = getbaseURL();
|
||||
const endpoint = `${getbaseURL()}/api/segmentShift`;
|
||||
|
||||
before(async function () {
|
||||
// startTime and endTime get set in beforeEach for consistency
|
||||
@@ -60,11 +61,8 @@ describe("segmentShift", function () {
|
||||
});
|
||||
|
||||
it("Reject none VIP user", function (done: Done) {
|
||||
fetch(`${baseURL}/api/segmentShift`, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
fetch(endpoint, {
|
||||
...postJSON,
|
||||
body: JSON.stringify({
|
||||
videoID: "vsegshift01",
|
||||
userID: "segshift_randomuser001",
|
||||
@@ -80,11 +78,8 @@ describe("segmentShift", function () {
|
||||
});
|
||||
|
||||
it("Shift is outside segments", function (done: Done) {
|
||||
fetch(`${baseURL}/api/segmentShift`, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
fetch(endpoint, {
|
||||
...postJSON,
|
||||
body: JSON.stringify({
|
||||
videoID: "vsegshift01",
|
||||
userID: privateVipUserID,
|
||||
@@ -94,39 +89,31 @@ describe("segmentShift", function () {
|
||||
})
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expect = [
|
||||
{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 50,
|
||||
endTime: 80,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 30,
|
||||
endTime: 35,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
},
|
||||
];
|
||||
const expect = [{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 50,
|
||||
endTime: 80,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 30,
|
||||
endTime: 35,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
}];
|
||||
done(await dbSponsorTimesCompareExpect(db, expect));
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Shift is inside segment", function (done: Done) {
|
||||
fetch(`${baseURL}/api/segmentShift`, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
fetch(endpoint, {
|
||||
...postJSON,
|
||||
body: JSON.stringify({
|
||||
videoID: "vsegshift01",
|
||||
userID: privateVipUserID,
|
||||
@@ -136,39 +123,31 @@ describe("segmentShift", function () {
|
||||
})
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expect = [
|
||||
{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 60,
|
||||
endTime: 80,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 40,
|
||||
endTime: 45,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
},
|
||||
];
|
||||
const expect = [{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 60,
|
||||
endTime: 80,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 40,
|
||||
endTime: 45,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
}];
|
||||
done(await dbSponsorTimesCompareExpect(db, expect));
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Shift is overlaping startTime of segment", function (done: Done) {
|
||||
fetch(`${baseURL}/api/segmentShift`, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
fetch(endpoint, {
|
||||
...postJSON,
|
||||
body: JSON.stringify({
|
||||
videoID: "vsegshift01",
|
||||
userID: privateVipUserID,
|
||||
@@ -178,39 +157,31 @@ describe("segmentShift", function () {
|
||||
})
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expect = [
|
||||
{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 50,
|
||||
endTime: 80,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 32,
|
||||
endTime: 35,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
},
|
||||
];
|
||||
const expect = [{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 50,
|
||||
endTime: 80,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 32,
|
||||
endTime: 35,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
}];
|
||||
done(await dbSponsorTimesCompareExpect(db, expect));
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Shift is overlaping endTime of segment", function (done: Done) {
|
||||
fetch(`${baseURL}/api/segmentShift`, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
fetch(endpoint, {
|
||||
...postJSON,
|
||||
body: JSON.stringify({
|
||||
videoID: "vsegshift01",
|
||||
userID: privateVipUserID,
|
||||
@@ -220,39 +191,31 @@ describe("segmentShift", function () {
|
||||
})
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expect = [
|
||||
{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 60,
|
||||
endTime: 85,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 40,
|
||||
endTime: 45,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
},
|
||||
];
|
||||
const expect = [{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 60,
|
||||
endTime: 85,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 40,
|
||||
endTime: 45,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 110,
|
||||
endTime: 130,
|
||||
}];
|
||||
done(await dbSponsorTimesCompareExpect(db, expect));
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Shift is overlaping segment", function (done: Done) {
|
||||
fetch(`${baseURL}/api/segmentShift`, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
fetch(endpoint, {
|
||||
...postJSON,
|
||||
body: JSON.stringify({
|
||||
videoID: "vsegshift01",
|
||||
userID: privateVipUserID,
|
||||
@@ -262,29 +225,24 @@ describe("segmentShift", function () {
|
||||
})
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expect = [
|
||||
{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 40,
|
||||
endTime: 70,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 40,
|
||||
endTime: 45,
|
||||
removed: true,
|
||||
},
|
||||
{
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 100,
|
||||
endTime: 120,
|
||||
},
|
||||
];
|
||||
const expect = [{
|
||||
UUID: "vsegshifttest01uuid01",
|
||||
startTime: 0,
|
||||
endTime: 10,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid02",
|
||||
startTime: 40,
|
||||
endTime: 70,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid03",
|
||||
startTime: 40,
|
||||
endTime: 45,
|
||||
removed: true,
|
||||
}, {
|
||||
UUID: "vsegshifttest01uuid04",
|
||||
startTime: 100,
|
||||
endTime: 120,
|
||||
}];
|
||||
done(await dbSponsorTimesCompareExpect(db, expect));
|
||||
})
|
||||
.catch(err => done(err));
|
||||
|
||||
Reference in New Issue
Block a user