From 0bbb2aa60d283d8173ea7ebabe5eac087fddf301 Mon Sep 17 00:00:00 2001
From: Ajay Ramachandran <dev@ajay.app>
Date: Thu, 24 Jun 2021 01:36:54 -0400
Subject: [PATCH] Add one more check in tests

---
 test/cases/postPurgeAllSegments.ts | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/test/cases/postPurgeAllSegments.ts b/test/cases/postPurgeAllSegments.ts
index 8161952..45086ec 100644
--- a/test/cases/postPurgeAllSegments.ts
+++ b/test/cases/postPurgeAllSegments.ts
@@ -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]);
 }
 
-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]);
     for (let i = 0, len = seg.length; i < len; i++) {
-        if (seg.hidden !== 1) {
-            return `${seg.UUID} hidden expected to be 1 but found ${seg.hidden}`;
+        if (seg.hidden !== expectdHidden) {
+            return `${seg.UUID} hidden expected to be ${expectdHidden} but found ${seg.hidden}`;
         }
     }
     return;
@@ -37,14 +37,15 @@ describe('postPurgeAllSegments', function () {
 
     before(async function () {
         // startTime and endTime get set in beforeEach for consistency
-        await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid01', 'intro');
-        await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid02', 'sponsor');
-        await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid03', 'interaction');
-        await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 0, 'vsegpurgetest01uuid04', 'outro');
+        await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 1, 'vsegpurgetest01uuid01', 'intro');
+        await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 2, 'vsegpurgetest01uuid02', 'sponsor');
+        await dbSponsorTimesAdd(db, 'vsegpurge01', 0, 3, 'vsegpurgetest01uuid03', 'interaction');
+        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]);
     });
 
-    it('Reject none VIP user', function (done: Done) {
+    it('Reject non-VIP user', function (done: Done) {
         fetch(`${baseURL}${route}`, {
             method: 'POST',
             headers: {
@@ -74,7 +75,7 @@ describe('postPurgeAllSegments', function () {
         })
         .then(async res => {
             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));
     });