diff --git a/src/routes/getSearchSegments.ts b/src/routes/getSearchSegments.ts index a73f9ca..5a86b55 100644 --- a/src/routes/getSearchSegments.ts +++ b/src/routes/getSearchSegments.ts @@ -12,8 +12,8 @@ type searchSegmentResponse = { async function getSegmentsFromDBByVideoID(videoID: VideoID, service: Service): Promise { return db.prepare( "all", - `SELECT "UUID", "timeSubmitted", "startTime", "endTime", "category", "actionType", "votes", "views", "locked", "hidden", "shadowHidden", FROM "sponsorTimes" - WHERE "videoID" = ? AND "service" = ? ORDER BY "UUID"`, + `SELECT "UUID", "timeSubmitted", "startTime", "endTime", "category", "actionType", "votes", "views", "locked", "hidden", "shadowHidden" FROM "sponsorTimes" + WHERE "videoID" = ? AND "service" = ? ORDER BY "timeSubmitted"`, [videoID, service] ) as Promise; } @@ -35,14 +35,13 @@ async function handleGetSegments(req: Request, res: Response): Promise) { +function filterSegments(segments: DBSegment[], page: number, filters: Record) { const startIndex = 0+(page*segmentsPerPage); const endIndex = segmentsPerPage+(page*segmentsPerPage); const filteredSegments = segments.filter((segment) => - (!(segment.votes <= filters.minVotes || segment.votes >= filters.maxVotes) - || (segment.views <= filters.minViews || segment.views >= filters.maxViews) - || (filters.locked && segment.locked) - || (filters.hidden && segment.hidden) - || (filters.ignored && (segment.hidden || segment.shadowHidden)) - ) + !((segment.votes < filters.minVotes || segment.votes > filters.maxVotes) + || (segment.views < filters.minViews || segment.views > filters.maxViews) + || (!filters.locked && segment.locked) + || (!filters.hidden && segment.hidden) + || (!filters.ignored && (segment.hidden || segment.shadowHidden)) + || (filters.categories.length > 0 && !filters.categories.includes(segment.category))) // return false if any of the conditions are met // return true if none of the conditions are met ); diff --git a/test/cases/getSearchSegments.ts b/test/cases/getSearchSegments.ts new file mode 100644 index 0000000..8f3aa61 --- /dev/null +++ b/test/cases/getSearchSegments.ts @@ -0,0 +1,254 @@ +import fetch from "node-fetch"; +import {db} from "../../src/databases/databases"; +import {Done, getbaseURL} from "../utils"; +import assert from "assert"; + +describe("getSearchSegments", () => { + before(async () => { + const query = 'INSERT INTO "sponsorTimes" ("videoID", "startTime", "endTime", "votes", "views", "locked", "hidden", "shadowHidden", "timeSubmitted", "UUID", "userID", "category", "actionType") VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)'; + await db.prepare("run", query, ["searchTest0", 0, 1, 2, 0, 0, 0, 0, 1, "search-normal", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest0", 0, 2, -2, 0, 0, 0, 0, 2, "search-downvote", "testman", "selfpromo", "skip",]); + await db.prepare("run", query, ["searchTest0", 0, 3, 1, 0, 1, 0, 0, 3, "search-locked", "testman", "interaction", "skip"]); + await db.prepare("run", query, ["searchTest0", 0, 4, 1, 0, 0, 1, 0, 4, "search-hidden", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest0", 0, 5, 1, 0, 0, 0, 1, 5, "search-shadowhidden", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest1", 1, 2, 1, 5, 0, 0, 0, 6, "search-lowview", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest1", 1, 3, 1, 50, 0, 0, 0, 7, "search-highview", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest2", 1, 4, -1, 0, 0, 0, 0, 8, "search-lowvote", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest2", 2, 3, 0, 0, 0, 0, 0, 9, "search-zerovote", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest2", 2, 4, 50, 0, 0, 0, 0, 10, "search-highvote", "testman", "sponsor", "skip"]); + // page + await db.prepare("run", query, ["searchTest4", 3, 4, 1, 0, 0, 0, 0, 10, "search-page1-1", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 5, 1, 0, 0, 0, 0, 11, "search-page1-2", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 6, 1, 0, 0, 0, 0, 12, "search-page1-3", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 7, 1, 0, 0, 0, 0, 13, "search-page1-4", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 8, 1, 0, 0, 0, 0, 14, "search-page1-5", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 9, 1, 0, 0, 0, 0, 15, "search-page1-6", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 10, 1, 0, 0, 0, 0, 16, "search-page1-7", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 11, 1, 0, 0, 0, 0, 17, "search-page1-8", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 12, 1, 0, 0, 0, 0, 18, "search-page1-9", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 13, 1, 0, 0, 0, 0, 19, "search-page1-10", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 14, 1, 0, 0, 0, 0, 20, "search-page2-1", "testman", "sponsor", "skip"]); + await db.prepare("run", query, ["searchTest4", 3, 15, 1, 0, 0, 0, 0, 21, "search-page2-2", "testman", "sponsor", "skip"]); + return; + }); + + it("Should be able to show all segments under searchTest0", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest0`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 5); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-normal"); + assert.strictEqual(segments[1].UUID, "search-downvote"); + assert.strictEqual(segments[2].UUID, "search-locked"); + assert.strictEqual(segments[3].UUID, "search-hidden"); + assert.strictEqual(segments[4].UUID, "search-shadowhidden"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter by category", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest0&category=selfpromo`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 1); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-downvote"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter by category", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest0&category=selfpromo`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 1); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-downvote"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter by lock status", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest0&locked=false`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 4); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-normal"); + assert.strictEqual(segments[1].UUID, "search-downvote"); + assert.strictEqual(segments[2].UUID, "search-hidden"); + assert.strictEqual(segments[3].UUID, "search-shadowhidden"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter by hide status", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest0&hidden=false`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 4); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-normal"); + assert.strictEqual(segments[1].UUID, "search-downvote"); + assert.strictEqual(segments[2].UUID, "search-locked"); + assert.strictEqual(segments[3].UUID, "search-shadowhidden"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter by ignored status", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest0&ignored=false`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 3); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-normal"); + assert.strictEqual(segments[1].UUID, "search-downvote"); + assert.strictEqual(segments[2].UUID, "search-locked"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter segments by min views", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest1&minViews=6`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 1); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-highview"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter segments by max views", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest1&maxViews=10`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 1); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-lowview"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter segments by min and max views", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest1&maxViews=10&minViews=1`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 1); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-lowview"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter segments by min votes", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest2&minVotes=0`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 2); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-zerovote"); + assert.strictEqual(segments[1].UUID, "search-highvote"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter segments by max votes", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest2&maxVotes=10`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 2); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-lowvote"); + assert.strictEqual(segments[1].UUID, "search-zerovote"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to filter segments by both min and max votes", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest2&maxVotes=10&minVotes=0`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 1); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-zerovote"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to get first page of results", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest4`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 12); + assert.strictEqual(data.page, 0); + assert.strictEqual(segments[0].UUID, "search-page1-1"); + assert.strictEqual(segments[1].UUID, "search-page1-2"); + assert.strictEqual(segments[2].UUID, "search-page1-3"); + assert.strictEqual(segments[3].UUID, "search-page1-4"); + assert.strictEqual(segments[4].UUID, "search-page1-5"); + assert.strictEqual(segments[5].UUID, "search-page1-6"); + assert.strictEqual(segments[6].UUID, "search-page1-7"); + assert.strictEqual(segments[7].UUID, "search-page1-8"); + assert.strictEqual(segments[8].UUID, "search-page1-9"); + assert.strictEqual(segments[9].UUID, "search-page1-10"); + done(); + }) + .catch(err => done(err)); + }); + + it("Should be able to get second page of results", (done: Done) => { + fetch(`${getbaseURL()}/api/searchSegments?videoID=searchTest4&page=1`) + .then(async res => { + assert.strictEqual(res.status, 200); + const data = await res.json(); + const segments = data.segments; + assert.strictEqual(data.segmentCount, 12); + assert.strictEqual(data.page, 1); + assert.strictEqual(segments[0].UUID, "search-page2-1"); + assert.strictEqual(segments[1].UUID, "search-page2-2"); + done(); + }) + .catch(err => done(err)); + }); +});