Merge pull request #439 from mchangrh/channelVipTests

additional tests for tempVIP

src/app.ts

@@ -45,7 +45,7 @@ import { youtubeApiProxy } from "./routes/youtubeApiProxy";
 import { getChapterNames } from "./routes/getChapterNames";
 import { postRating } from "./routes/ratings/postRating";
 import { getRating } from "./routes/ratings/getRating";
-import { postClearCache as ratingPostClearCache } from "./routes/ratings/postClearCache"
+import { postClearCache as ratingPostClearCache } from "./routes/ratings/postClearCache";
 import { getTopCategoryUsers } from "./routes/getTopCategoryUsers";
 import { addUserAsTempVIP } from "./routes/addUserAsTempVIP";
 

src/routes/addUserAsTempVIP.ts

@@ -37,7 +37,7 @@ export async function addUserAsTempVIP(req: AddUserAsTempVIPRequest, res: Respon
     const enabled = req.query?.enabled === "true";
     const channelVideoID = req.query?.channelVideoID as VideoID;
 
-    if (!userID || !adminUserID || !channelVideoID ) {
+    if ((!userID || !adminUserID || (!channelVideoID && enabled))) {
         // invalid request
         return res.sendStatus(400);
     }
@@ -57,15 +57,19 @@ export async function addUserAsTempVIP(req: AddUserAsTempVIPRequest, res: Respon
     }
 
     const startTime = Date.now();
+
+    if (enabled) {
         const dayInSeconds = 86400;
         const channelInfo = await getChannelInfo(channelVideoID);
-
+        if (!channelInfo?.id) {
-    await privateDB.prepare("run", `INSERT INTO "tempVipLog" VALUES (?, ?, ?, ?)`, [adminUserID, userID, + enabled, startTime]);
+            return res.status(404).send(`No channel found for videoID ${channelVideoID}`);
-    if (enabled) { // add to redis
+        }
         await redis.setAsyncEx(tempVIPKey(userID), channelInfo?.id, dayInSeconds);
-    } else { // delete key
+        await privateDB.prepare("run", `INSERT INTO "tempVipLog" VALUES (?, ?, ?, ?)`, [adminUserID, userID, + enabled, startTime]);
-        await redis.delAsync(tempVIPKey(userID));
+        return res.status(200).send(`Temp VIP added on channel ${channelInfo?.name}`);
     }
 
-    return res.sendStatus(200);
+    await redis.delAsync(tempVIPKey(userID));
+    await privateDB.prepare("run", `INSERT INTO "tempVipLog" VALUES (?, ?, ?, ?)`, [adminUserID, userID, + enabled, startTime]);
+    return res.status(200).send(`Temp VIP removed`);
 }

test/cases/tempVip.ts

@@ -10,22 +10,25 @@ import assert from "assert";
 // helpers
 const getSegment = (UUID: string) => db.prepare("get", `SELECT "votes", "locked", "category" FROM "sponsorTimes" WHERE "UUID" = ?`, [UUID]);
 
-const permVIP = "tempVipPermOne";
+const permVIP1 = "tempVipPermOne";
-const publicPermVIP = getHash(permVIP) as HashedUserID;
+const publicPermVIP1 = getHash(permVIP1) as HashedUserID;
+const permVIP2 = "tempVipPermOne";
+const publicPermVIP2 = getHash(permVIP2) as HashedUserID;
+
 const tempVIPOne = "tempVipTempOne";
 const publicTempVIPOne = getHash(tempVIPOne) as HashedUserID;
 const UUID0 = "tempvip-uuid0";
 const UUID1 = "tempvip-uuid1";
 
 const tempVIPEndpoint = "/api/addUserAsTempVIP";
-const addTempVIP = (enabled: boolean) => client({
+const addTempVIP = (enabled: string, adminUserID: string, userID: HashedUserID, channelVideoID = "channelid-convert") => client({
     url: tempVIPEndpoint,
     method: "POST",
     params: {
-        userID: publicTempVIPOne,
+        userID,
-        adminUserID: permVIP,
+        adminUserID,
-        channelVideoID: "channelid-convert",
+        channelVideoID,
-        enabled: enabled
+        enabled
     }
 });
 const voteEndpoint = "/api/voteOnSponsorTime";
@@ -47,8 +50,8 @@ const postVoteCategory = (userID: string, UUID: string, category: string) => cli
         category
     }
 });
-const checkUserVIP = async () => {
+const checkUserVIP = async (publicID: HashedUserID) => {
-    const { reply } = await redis.getAsync(tempVIPKey(publicTempVIPOne));
+    const { reply } = await redis.getAsync(tempVIPKey(publicID));
     return reply;
 };
 
@@ -62,7 +65,8 @@ describe("tempVIP test", function() {
         await db.prepare("run", insertSponsorTimeQuery, ["otherchannel",        1, 9, 0, 1, UUID1, "testman", 0, 50, "sponsor", 0]);
 
 
-        await db.prepare("run", 'INSERT INTO "vipUsers" ("userID") VALUES (?)', [publicPermVIP]);
+        await db.prepare("run", 'INSERT INTO "vipUsers" ("userID") VALUES (?)', [publicPermVIP1]);
+        await db.prepare("run", 'INSERT INTO "vipUsers" ("userID") VALUES (?)', [publicPermVIP2]);
         // clear redis if running consecutive tests
         await redis.delAsync(tempVIPKey(publicTempVIPOne));
     });
@@ -74,10 +78,14 @@ describe("tempVIP test", function() {
             });
     });
     it("User should not already be temp VIP", (done) => {
-        checkUserVIP()
+        checkUserVIP(publicTempVIPOne)
             .then(result => {
                 assert.ok(!result);
-                done(result);
+            })
+            .then(async () => {
+                const row = await privateDB.prepare("get", `SELECT * FROM "tempVipLog" WHERE "targetUserID" = ?`, [publicTempVIPOne]);
+                assert.ok(!row?.enabled);
+                done();
             })
             .catch(err => done(err));
     });
@@ -92,11 +100,16 @@ describe("tempVIP test", function() {
             .catch(err => done(err));
     });
     it("Should be able to add tempVIP", (done) => {
-        addTempVIP(true)
+        addTempVIP("true", permVIP1, publicTempVIPOne)
             .then(async res => {
                 assert.strictEqual(res.status, 200);
-                const vip = await checkUserVIP();
+                // check redis
-                assert.ok(vip == "ChannelID");
+                const vip = await checkUserVIP(publicTempVIPOne);
+                assert.strictEqual(vip, "ChannelID");
+                assert.strictEqual(res.data, "Temp VIP added on channel ChannelAuthor");
+                // check privateDB
+                const row = await privateDB.prepare("get", `SELECT * FROM "tempVipLog" WHERE "targetUserID" = ?`, [publicTempVIPOne]);
+                assert.ok(row.enabled);
                 done();
             })
             .catch(err => done(err));
@@ -134,11 +147,13 @@ describe("tempVIP test", function() {
             .catch(err => done(err));
     });
     it("Should be able to remove tempVIP prematurely", (done) => {
-        addTempVIP(false)
+        addTempVIP("false", permVIP1, publicTempVIPOne, null)
             .then(async res => {
                 assert.strictEqual(res.status, 200);
-                const vip = await checkUserVIP();
+                const vip = await checkUserVIP(publicTempVIPOne);
-                done(vip);
+                assert.strictEqual(res.data, "Temp VIP removed");
+                assert.ok(!vip, "Should be no listed channelID");
+                done();
             })
             .catch(err => done(err));
     });
@@ -162,4 +177,85 @@ describe("tempVIP test", function() {
             })
             .catch(err => done(err));
     });
+    // error code testing
+    it("Should be able to add tempVIP after removal", (done) => {
+        addTempVIP("true", permVIP1, publicTempVIPOne)
+            .then(async res => {
+                assert.strictEqual(res.status, 200);
+                const vip = await checkUserVIP(publicTempVIPOne);
+                assert.strictEqual(vip, "ChannelID");
+                done();
+            })
+            .catch(err => done(err));
+    });
+    it("Should not be able to add VIP without existing VIP (403)", (done) => {
+        const privateID = "non-vip-privateid";
+        addTempVIP("true", privateID, publicTempVIPOne)
+            .then(async res => {
+                assert.strictEqual(res.status, 403);
+                const vip = await checkUserVIP(getHash(privateID) as HashedUserID);
+                assert.ok(!vip, "Should be no listed channelID");
+                done();
+            })
+            .catch(err => done(err));
+    });
+    it("Should not be able to add permanant VIP as temporary VIP (409)", (done) => {
+        addTempVIP("true", permVIP1, publicPermVIP2)
+            .then(async res => {
+                assert.strictEqual(res.status, 409);
+                const vip = await checkUserVIP(publicPermVIP2);
+                assert.ok(!vip, "Should be no listed channelID");
+                done();
+            })
+            .catch(err => done(err));
+    });
+    it("Temp VIP should not be able to add another Temp VIP (403)", (done) => {
+        const privateID = "non-vip-privateid";
+        const publicID = getHash(privateID) as HashedUserID;
+        addTempVIP("true", tempVIPOne, publicID)
+            .then(async res => {
+                assert.strictEqual(res.status, 403);
+                const vip = await checkUserVIP(publicID);
+                assert.ok(!vip, "Should be no listed channelID");
+                done();
+            })
+            .catch(err => done(err));
+    });
+    // error 40X testing
+    it("Should return 404 with invalid videoID", (done) => {
+        const privateID = "non-vip-privateid";
+        const publicID = getHash(privateID) as HashedUserID;
+        addTempVIP("true", permVIP1, publicID, "knownWrongID")
+            .then(async res => {
+                assert.strictEqual(res.status, 404);
+                const vip = await checkUserVIP(publicID);
+                assert.ok(!vip, "Should be no listed channelID");
+                done();
+            })
+            .catch(err => done(err));
+    });
+    it("Should return 400 with invalid userID", (done) => {
+        addTempVIP("true", permVIP1, "" as HashedUserID, "knownWrongID")
+            .then(res => {
+                assert.strictEqual(res.status, 400);
+                done();
+            })
+            .catch(err => done(err));
+    });
+    it("Should return 400 with invalid adminUserID", (done) => {
+        addTempVIP("true", "", publicTempVIPOne)
+            .then(res => {
+                assert.strictEqual(res.status, 400);
+                done();
+            })
+            .catch(err => done(err));
+    });
+    it("Should return 400 with invalid channelID", (done) => {
+        addTempVIP("true", permVIP1, publicTempVIPOne, "")
+            .then(res => {
+                assert.strictEqual(res.status, 400);
+                done();
+            })
+            .catch(err => done(err));
+    });
 });