mirror of
https://github.com/ajayyy/SponsorBlockServer.git
synced 2024-11-10 17:17:50 +01:00
231 lines
11 KiB
TypeScript
231 lines
11 KiB
TypeScript
import fetch from 'node-fetch';
|
|
import {db} from '../../src/databases/databases';
|
|
import {Done, getbaseURL} from '../utils';
|
|
import {getHash} from '../../src/utils/getHash';
|
|
|
|
describe('getSkipSegments', () => {
|
|
before(async () => {
|
|
let startOfQuery = 'INSERT INTO "sponsorTimes" ("videoID", "startTime", "endTime", "votes", "locked", "UUID", "userID", "timeSubmitted", views, category, "shadowHidden", "hashedVideoID") VALUES';
|
|
await db.prepare("run", startOfQuery + "('testtesttest', 1, 11, 2, 0, '1-uuid-0', 'testman', 0, 50, 'sponsor', 0, '" + getHash('testtesttest', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('testtesttest', 20, 33, 2, 0, '1-uuid-2', 'testman', 0, 50, 'intro', 0, '" + getHash('testtesttest', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('testtesttest,test', 1, 11, 2, 0, '1-uuid-1', 'testman', 0, 50, 'sponsor', 0, '" + getHash('testtesttest,test', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('test3', 1, 11, 2, 0, '1-uuid-4', 'testman', 0, 50, 'sponsor', 0, '" + getHash('test3', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('test3', 7, 22, -3, 0, '1-uuid-5', 'testman', 0, 50, 'sponsor', 0, '" + getHash('test3', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('multiple', 1, 11, 2, 0, '1-uuid-6', 'testman', 0, 50, 'intro', 0, '" + getHash('multiple', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('multiple', 20, 33, 2, 0, '1-uuid-7', 'testman', 0, 50, 'intro', 0, '" + getHash('multiple', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('locked', 20, 33, 2, 1, '1-uuid-locked-8', 'testman', 0, 50, 'intro', 0, '" + getHash('locked', 1) + "')");
|
|
await db.prepare("run", startOfQuery + "('locked', 20, 34, 100000, 0, '1-uuid-9', 'testman', 0, 50, 'intro', 0, '" + getHash('locked', 1) + "')");
|
|
|
|
return;
|
|
});
|
|
|
|
|
|
it('Should be able to get a time by category 1', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=testtesttest&category=sponsor")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const data = await res.json();
|
|
if (data.length === 1 && data[0].segment[0] === 1 && data[0].segment[1] === 11
|
|
&& data[0].category === "sponsor" && data[0].UUID === "1-uuid-0") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + (await res.text()));
|
|
}
|
|
}
|
|
})
|
|
.catch(err => "Couldn't call endpoint");
|
|
});
|
|
|
|
it('Should be able to get a time by category 2', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=testtesttest&category=intro")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const data = await res.json();
|
|
if (data.length === 1 && data[0].segment[0] === 20 && data[0].segment[1] === 33
|
|
&& data[0].category === "intro" && data[0].UUID === "1-uuid-2") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + (await res.text()));
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get a time by categories array', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=testtesttest&categories=[\"sponsor\"]")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const data = await res.json();
|
|
if (data.length === 1 && data[0].segment[0] === 1 && data[0].segment[1] === 11
|
|
&& data[0].category === "sponsor" && data[0].UUID === "1-uuid-0") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + (await res.text()));
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get a time by categories array 2', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=testtesttest&categories=[\"intro\"]")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const data = await res.json();
|
|
if (data.length === 1 && data[0].segment[0] === 20 && data[0].segment[1] === 33
|
|
&& data[0].category === "intro" && data[0].UUID === "1-uuid-2") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + (await res.text()));
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get multiple times by category', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=multiple&categories=[\"intro\"]")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
const data = JSON.parse(body);
|
|
if (data.length === 2) {
|
|
let success = true;
|
|
for (const segment of data) {
|
|
if ((segment.segment[0] !== 20 || segment.segment[1] !== 33
|
|
|| segment.category !== "intro" || segment.UUID !== "1-uuid-7") &&
|
|
(segment.segment[0] !== 1 || segment.segment[1] !== 11
|
|
|| segment.category !== "intro" || segment.UUID !== "1-uuid-6")) {
|
|
success = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (success) return;
|
|
else return ("Received incorrect body: " + body);
|
|
} else {
|
|
return ("Received incorrect body: " + body);
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint\n\n" + err));
|
|
});
|
|
|
|
it('Should be able to get multiple times by multiple categories', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=testtesttest&categories=[\"sponsor\", \"intro\"]")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
const data = JSON.parse(body);
|
|
if (data.length === 2) {
|
|
|
|
let success = true;
|
|
for (const segment of data) {
|
|
if ((segment.segment[0] !== 20 || segment.segment[1] !== 33
|
|
|| segment.category !== "intro" || segment.UUID !== "1-uuid-2") &&
|
|
(segment.segment[0] !== 1 || segment.segment[1] !== 11
|
|
|| segment.category !== "sponsor" || segment.UUID !== "1-uuid-0")) {
|
|
success = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (success) return;
|
|
else return ("Received incorrect body: " + body);
|
|
} else {
|
|
return ("Received incorrect body: " + body);
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be possible to send unexpected query parameters', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=testtesttest&fakeparam=hello&category=sponsor")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
const data = JSON.parse(body);
|
|
if (data.length === 1 && data[0].segment[0] === 1 && data[0].segment[1] === 11
|
|
&& data[0].category === "sponsor" && data[0].UUID === "1-uuid-0") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + body);
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Low voted submissions should be hidden', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=test3&category=sponsor")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
const data = JSON.parse(body);
|
|
if (data.length === 1 && data[0].segment[0] === 1 && data[0].segment[1] === 11
|
|
&& data[0].category === "sponsor" && data[0].UUID === "1-uuid-4") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + body);
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should return 404 if no segment found', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=notarealvideo")
|
|
.then(res => {
|
|
if (res.status !== 404) return ("non 404 respone code: " + res.status);
|
|
else return; // pass
|
|
})
|
|
.catch(err => ("couldn't call endpoint"));
|
|
});
|
|
|
|
|
|
it('Should be able send a comma in a query param', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=testtesttest,test&category=sponsor")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
const data = JSON.parse(body);
|
|
if (data.length === 1 && data[0].segment[0] === 1 && data[0].segment[1] === 11
|
|
&& data[0].category === "sponsor" && data[0].UUID === "1-uuid-1") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + body);
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should always get locked segment', () => {
|
|
fetch(getbaseURL() + "/api/skipSegments?videoID=locked&category=intro")
|
|
.then(async res => {
|
|
if (res.status !== 200) return ("Status code was: " + res.status);
|
|
else {
|
|
const data = await res.json();
|
|
if (data.length === 1 && data[0].segment[0] === 20 && data[0].segment[1] === 33
|
|
&& data[0].category === "intro" && data[0].UUID === "1-uuid-locked-8") {
|
|
return;
|
|
} else {
|
|
return ("Received incorrect body: " + (await res.text()));
|
|
}
|
|
}
|
|
})
|
|
.catch(err => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
});
|