Mirror/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2025-12-26 17:38:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add one more check in tests

Browse Source
This commit is contained in:
Ajay Ramachandran
2021-06-24 01:36:54 -04:00
parent 9b43ce0ab7
commit 0bbb2aa60d
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
test/cases/postPurgeAllSegments.ts
View File

@@ -19,11 +19,11 @@ async function dbSponsorTimesAdd(db: IDatabase, videoID: string, startTime: numb
[videoID, startTime, endTime, votes, UUID, userID, timeSubmitted, views, category, shadowHidden, hashedVideoID, hidden]); [videoID, startTime, endTime, votes, UUID, userID, timeSubmitted, views, category, shadowHidden, hashedVideoID, hidden]);
} }
async function dbSponsorTimesCompareExpect(db: IDatabase, videoId: string) { async function dbSponsorTimesCompareExpect(db: IDatabase, videoId: string, expectdHidden: number) {
let seg = await db.prepare('get', `SELECT "hidden", "UUID" FROM "sponsorTimes" WHERE "videoId" = ?`, [videoId]); let seg = await db.prepare('get', `SELECT "hidden", "UUID" FROM "sponsorTimes" WHERE "videoId" = ?`, [videoId]);
for (let i = 0, len = seg.length; i < len; i++) { for (let i = 0, len = seg.length; i < len; i++) {
if (seg.hidden !== 1) { if (seg.hidden !== expectdHidden) {
return `${seg.UUID} hidden expected to be 1 but found ${seg.hidden}`; return `${seg.UUID} hidden expected to be ${expectdHidden} but found ${seg.hidden}`;
} }
} }
return; return;
@@ -37,14 +37,15 @@ describe('postPurgeAllSegments', function () {
before(async function () { before(async function () {
// startTime and endTime get set in beforeEach for consistency // startTime and endTime get set in beforeEach for consistency
await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid01', 'intro'); await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 1, 'vsegpurgetest01uuid01', 'intro');
await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid02', 'sponsor'); await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 2, 'vsegpurgetest01uuid02', 'sponsor');
await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid03', 'interaction'); await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 3, 'vsegpurgetest01uuid03', 'interaction');
await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid04', 'outro'); await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 4, 'vsegpurgetest01uuid04', 'outro');
await dbSponsorTimesAdd(db, 'vseg-not-purged01', 0, 5, 'vsegpurgetest01uuid05', 'outro');
await db.prepare("run", `INSERT INTO "vipUsers" ("userID") VALUES (?)`, [vipUserID]); await db.prepare("run", `INSERT INTO "vipUsers" ("userID") VALUES (?)`, [vipUserID]);
}); });
it('Reject none VIP user', function (done: Done) { it('Reject non-VIP user', function (done: Done) {
fetch(`${baseURL}${route}`, { fetch(`${baseURL}${route}`, {
method: 'POST', method: 'POST',
headers: { headers: {
@@ -74,7 +75,7 @@ describe('postPurgeAllSegments', function () {
}) })
.then(async res => { .then(async res => {
if (res.status !== 200) return done(`Status code was ${res.status}`); if (res.status !== 200) return done(`Status code was ${res.status}`);
done(await dbSponsorTimesCompareExpect(db, 'vsegpurge01')); done(await dbSponsorTimesCompareExpect(db, 'vsegpurge01', 1) || await dbSponsorTimesCompareExpect(db, 'vseg-not-purged01', 0));
}) })
.catch(err => done(err)); .catch(err => done(err));
}); });