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

Merge

Browse Source
This commit is contained in:
Ajay Ramachandran
2020-04-06 20:05:26 -04:00
parent 761fb7dafe
commit 9807d3e9c7
12 changed files with 529 additions and 126 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/cases/getSavedTimeForUser.js
View File

@@ -5,16 +5,16 @@ var getHash = require('../../src/utils/getHash.js');
describe('getSavedTimeForUser', () => {
before(() => {
db.exec("INSERT INTO sponsorTimes VALUES ('getSavedTimeForUser', 1, 11, 2, 'abc1239999', '"+getHash("testman")+"', 0, 50, 0)");
db.exec("INSERT INTO sponsorTimes VALUES ('getSavedTimeForUser', 1, 11, 2, 'abc1239999', '" + getHash("testman") + "', 0, 50, 'sponsor', 0)");
});
it('Should be able to get a 200', (done) => {
request.get(utils.getbaseURL()
+ "/api/getSavedTimeForUser?userID=testman", null,
(err, res, body) => {
if (err) done(false);
if (err) done("couldn't call endpoint");
else if (res.statusCode !== 200) done("non 200");
else done();
else done(); // pass
});
});
});

25
test/cases/getSponsorTime.js
View File

@@ -19,17 +19,17 @@ var utils = require('../utils.js');
describe('getVideoSponsorTime', () => {
before(() => {
db.exec("INSERT INTO sponsorTimes VALUES ('testtesttest', 1, 11, 2, 'uuid-0', 'testman', 0, 50, 0)");
db.exec("INSERT INTO sponsorTimes VALUES ('testtesttest,test', 1, 11, 2, 'uuid-1', 'testman', 0, 50, 0)");
db.exec("INSERT INTO sponsorTimes VALUES ('testtesttest', 1, 11, 2, 'uuid-0', 'testman', 0, 50, 'sponsor', 0)");
db.exec("INSERT INTO sponsorTimes VALUES ('testtesttest,test', 1, 11, 2, 'uuid-1', 'testman', 0, 50, 'sponsor', 0)");
});
it('Should be able to get a time', (done) => {
request.get(utils.getbaseURL()
+ "/api/getVideoSponsorTimes?videoID=testtesttest", null,
(err, res, body) => {
if (err) done(false);
if (err) done("Couldn't call endpoint");
else if (res.statusCode !== 200) done("non 200");
else done();
else done(); // pass
});
});
@@ -37,9 +37,9 @@ describe('getVideoSponsorTime', () => {
request.get(utils.getbaseURL()
+ "/api/getVideoSponsorTimes?videoID=notarealvideo", null,
(err, res, body) => {
if (err) done(false);
if (err) done("couldn't call endpoint");
else if (res.statusCode !== 404) done("non 404 respone code: " + res.statusCode);
else done();
else done(); // pass
});
});
@@ -48,9 +48,9 @@ describe('getVideoSponsorTime', () => {
request.get(utils.getbaseURL()
+ "/api/getVideoSponsorTimes?videoID=testtesttest&fakeparam=hello", null,
(err, res, body) => {
if (err) done(false);
if (err) done("couldn't callendpoint");
else if (res.statusCode !== 200) done("non 200");
else done();
else done(); // pass
});
});
@@ -58,9 +58,10 @@ describe('getVideoSponsorTime', () => {
request.get(utils.getbaseURL()
+ "/api/getVideoSponsorTimes?videoID=testtesttest,test", null,
(err, res, body) => {
if (err) done(false);
if (err) done("couln't call endpoint");
else if (res.statusCode !== 200) done("non 200 response: " + res.statusCode);
else (JSON.parse(body).UUIDs[0] === 'uuid-1') && done();
else if (JSON.parse(body).UUIDs[0] === 'uuid-1') done(); // pass
else done("couldn't parse response");
});
});
@@ -68,14 +69,14 @@ describe('getVideoSponsorTime', () => {
request.get(utils.getbaseURL()
+ "/api/getVideoSponsorTimes?videoID=testtesttest", null,
(err, res, body) => {
if (err) done(false);
if (err) done("couldn't call endpoint");
else if (res.statusCode !== 200) done("non 200");
else {
let parsedBody = JSON.parse(body);
if (parsedBody.sponsorTimes[0][0] === 1
&& parsedBody.sponsorTimes[0][1] === 11
&& parsedBody.UUIDs[0] === 'uuid-0') {
done();
done(); // pass
} else {
done("Wrong data was returned + " + parsedBody);
}

32
test/cases/oldSubmitSponsorTimes.js
View File

@@ -9,35 +9,37 @@ var db = databases.db;
describe('postVideoSponsorTime (Old submission method)', () => {
it('Should be able to submit a time (GET)', (done) => {
request.get(utils.getbaseURL()
+ "/api/postVideoSponsorTimes?videoID=djgofQKWmXc&startTime=1&endTime=10&userID=test", null,
+ "/api/postVideoSponsorTimes?videoID=dQw4w9WgXcQ&startTime=1&endTime=10&userID=test", null,
(err, res, body) => {
if (err) done(false);
if (err) done(err);
else if (res.statusCode === 200) {
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get(videoID);
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get("dQw4w9WgXcQ");
if (row.startTime === 1 && row.endTime === 10 && row.category === "sponsor") {
done()
return;
} else {
done("Submitted times were not saved. Actual submission: " + JSON.stringify(row));
}
} else {
done("Status code was " + res.statusCode);
}
done(false);
});
});
it('Should be able to submit a time (POST)', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes?videoID=djgofQKWmXc&startTime=1&endTime=10&userID=test", null,
+ "/api/postVideoSponsorTimes?videoID=dQw4w9WgXcE&startTime=1&endTime=11&userID=test", null,
(err, res, body) => {
if (err) done(false);
if (err) done(err);
else if (res.statusCode === 200) {
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get(videoID);
if (row.startTime === 1 && row.endTime === 10 && row.category === "sponsor") {
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get("dQw4w9WgXcE");
if (row.startTime === 1 && row.endTime === 11 && row.category === "sponsor") {
done()
return;
} else {
done("Submitted times were not saved. Actual submission: " + JSON.stringify(row));
}
} else {
done("Status code was " + res.statusCode);
}
done(false);
});
});
@@ -45,9 +47,9 @@ describe('postVideoSponsorTime (Old submission method)', () => {
request.get(utils.getbaseURL()
+ "/api/postVideoSponsorTimes?startTime=1&endTime=10&userID=test", null,
(err, res, body) => {
if (err) done(false);
if (err) done(err);
if (res.statusCode === 400) done();
else done(false);
else done("Status code was: " + res.statusCode);
});
});
});

180
test/cases/postSkipSegments.js
View File

@@ -9,172 +9,204 @@ var db = databases.db;
describe('postSkipSegments', () => {
it('Should be able to submit a single time (Params method)', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes?videoID=djgofQKWmXc&startTime=1&endTime=10&userID=test&category=sponsor", null,
+ "/api/postVideoSponsorTimes?videoID=dQw4w9WgXcR&startTime=2&endTime=10&userID=test&category=sponsor", null,
(err, res, body) => {
if (err) done(false);
if (err) done(err);
else if (res.statusCode === 200) {
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get(videoID);
if (row.startTime === 1 && row.endTime === 10 && row.category === "sponsor") {
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get("dQw4w9WgXcR");
if (row.startTime === 2 && row.endTime === 10 && row.category === "sponsor") {
done()
return;
} else {
done("Submitted times were not saved. Actual submission: " + JSON.stringify(row));
}
} else {
done("Status code was " + res.statusCode);
}
done(false);
});
});
it('Should be able to submit a single time (JSON method)', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes", JSON.stringify({
body: {
videoID: "djgofQKWmXc",
+ "/api/postVideoSponsorTimes", {
json: {
userID: "test",
videoID: "dQw4w9WgXcF",
segments: [{
segment: [0, 10],
category: "sponsor"
}]
}
}),
},
(err, res, body) => {
if (err) done(false);
if (err) done(err);
else if (res.statusCode === 200) {
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get(videoID);
let row = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").get("dQw4w9WgXcF");
if (row.startTime === 0 && row.endTime === 10 && row.category === "sponsor") {
done()
return;
} else {
done("Submitted times were not saved. Actual submission: " + JSON.stringify(row));
}
} else {
done("Status code was " + res.statusCode);
}
done(false);
});
});
it('Should be able to submit multiple times (JSON method)', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes", JSON.stringify({
body: {
videoID: "djgofQKWmXc",
+ "/api/postVideoSponsorTimes", {
json: {
userID: "test",
videoID: "dQw4w9WgXcQ",
segments: [{
segment: [0, 10],
segment: [3, 10],
category: "sponsor"
}, {
segment: [30, 60],
category: "intro"
}]
}
}),
},
(err, res, body) => {
if (err) done(false);
if (err) done(err);
else if (res.statusCode === 200) {
let rows = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").all(videoID);
if (rows.length !== 2) done(false);
for (const row of rows) {
if (row.startTime !== 1 || row.endTime !== 10 || row.category !== "sponsor") {
done(false)
return;
let rows = db.prepare("SELECT startTime, endTime, category FROM sponsorTimes WHERE videoID = ?").all("dQw4w9WgXcR");
let success = true;
if (rows.length === 2) {
for (const row of rows) {
if ((row.startTime !== 3 || row.endTime !== 10 || row.category !== "sponsor") &&
(row.startTime !== 30 || row.endTime !== 60 || row.category !== "intro")) {
success = false;
break;
}
}
}
done()
if (success) done();
else done("Submitted times were not saved. Actual submissions: " + JSON.stringify(row));
} else {
done("Status code was " + res.statusCode);
}
done(false);
});
});
it('Should be rejected if over 80% of the video', (done) => {
request.get(utils.getbaseURL()
+ "/api/postVideoSponsorTimes?videoID=qqwerty&startTime=30&endTime=1000000&userID=testing", null,
(err, res, body) => {
if (err) done("Couldn't call endpoint");
else if (res.statusCode === 403) done(); // pass
else done("non 403 status code: " + res.statusCode + " ("+body+")");
});
});
it('Should be rejected if not a valid videoID', (done) => {
request.get(utils.getbaseURL()
+ "/api/postVideoSponsorTimes?videoID=knownWrongID&startTime=30&endTime=1000000&userID=testing", null,
(err, res, body) => {
if (err) done("Couldn't call endpoint");
else if (res.statusCode === 403) done(); // pass
else done("non 403 status code: " + res.statusCode + " ("+body+")");
});
});
it('Should return 400 for missing params (Params method)', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes?startTime=1&endTime=10&userID=test", null,
+ "/api/postVideoSponsorTimes?startTime=9&endTime=10&userID=test", null,
(err, res, body) => {
if (err) done(false);
if (err) done(true);
if (res.statusCode === 400) done();
else done(false);
else done(true);
});
});
it('Should return 400 for missing params (JSON method) 1', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes", JSON.stringify({
body: {
+ "/api/postVideoSponsorTimes", {
json: {
userID: "test",
segments: [{
segment: [0, 10],
segment: [9, 10],
category: "sponsor"
}, {
segment: [30, 60],
segment: [31, 60],
category: "intro"
}]
}
}),
},
(err, res, body) => {
if (err) done(false);
else if (res.statusCode === 200) done();
else done(false);
if (err) done(true);
else if (res.statusCode === 400) done();
else done(true);
});
});
it('Should return 400 for missing params (JSON method) 2', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes", JSON.stringify({
body: {
videoID: "djgofQKWmXc"
+ "/api/postVideoSponsorTimes", {
json: {
userID: "test",
videoID: "dQw4w9WgXcQ"
}
}),
},
(err, res, body) => {
if (err) done(false);
else if (res.statusCode === 200) done();
else done(false);
if (err) done(true);
else if (res.statusCode === 400) done();
else done(true);
});
});
it('Should return 400 for missing params (JSON method) 3', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes", JSON.stringify({
body: {
videoID: "djgofQKWmXc",
+ "/api/postVideoSponsorTimes", {
json: {
userID: "test",
videoID: "dQw4w9WgXcQ",
segments: [{
segment: [0],
category: "sponsor"
}, {
segment: [30, 60],
segment: [31, 60],
category: "intro"
}]
}
}),
},
(err, res, body) => {
if (err) done(false);
else if (res.statusCode === 200) done();
else done(false);
if (err) done(true);
else if (res.statusCode === 400) done();
else done(true);
});
});
it('Should return 400 for missing params (JSON method) 4', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes", JSON.stringify({
body: {
videoID: "djgofQKWmXc",
+ "/api/postVideoSponsorTimes", {
json: {
userID: "test",
videoID: "dQw4w9WgXcQ",
segments: [{
segment: [0, 10]
segment: [9, 10]
}, {
segment: [30, 60],
segment: [31, 60],
category: "intro"
}]
}
}),
},
(err, res, body) => {
if (err) done(false);
else if (res.statusCode === 200) done();
else done(false);
if (err) done(true);
else if (res.statusCode === 400) done();
else done(true);
});
});
it('Should return 400 for missing params (JSON method) 5', (done) => {
request.post(utils.getbaseURL()
+ "/api/postVideoSponsorTimes", JSON.stringify({
body: {
videoID: "djgofQKWmXc"
+ "/api/postVideoSponsorTimes", {
json: {
userID: "test",
videoID: "dQw4w9WgXcQ"
}
}),
},
(err, res, body) => {
if (err) done(false);
else if (res.statusCode === 200) done();
else done(false);
if (err) done(true);
else if (res.statusCode === 400) done();
else done(true);
});
});
});

4
test/databases/_private.db.sql
View File

@@ -1,8 +1,4 @@
BEGIN TRANSACTION;
DROP TABLE IF EXISTS "shadowBannedUsers";
DROP TABLE IF EXISTS "votes";
DROP TABLE IF EXISTS "sponsorTimes";
CREATE TABLE IF NOT EXISTS "shadowBannedUsers" (
"userID" TEXT NOT NULL
);

5
test/databases/_sponsorTimes.db.sql
View File

@@ -1,8 +1,4 @@
BEGIN TRANSACTION;
DROP TABLE IF EXISTS "vipUsers";
DROP TABLE IF EXISTS "sponsorTimes";
DROP TABLE IF EXISTS "userNames";
CREATE TABLE IF NOT EXISTS "vipUsers" (
"userID" TEXT NOT NULL
);
@@ -15,6 +11,7 @@ CREATE TABLE IF NOT EXISTS "sponsorTimes" (
"userID" TEXT NOT NULL,
"timeSubmitted" INTEGER NOT NULL,
"views" INTEGER NOT NULL,
"category" TEXT NOT NULL,
"shadowHidden" INTEGER NOT NULL
);
CREATE TABLE IF NOT EXISTS "userNames" (

38
test/youtubeMock.js Normal file
View File

@@ -0,0 +1,38 @@
/*
YouTubeAPI.videos.list({
part: "snippet",
id: videoID
}, function (err, data) {});
*/
// https://developers.google.com/youtube/v3/docs/videos
const YouTubeAPI = {
videos: {
list: (obj, callback) => {
if (obj.videoID === "knownWrongID") {
callback(undefined, {
pageInfo: {
totalResults: 0
},
items: []
});
} else {
callback(undefined, {
pageInfo: {
totalResults: 1
},
items: [
{
contentDetails: {
duration: "PT1H23M30S"
}
}
]
});
}
}
}
};
module.exports = YouTubeAPI;