mirror of
https://github.com/ajayyy/SponsorBlockServer.git
synced 2025-12-09 13:07:02 +03:00
348 lines
19 KiB
TypeScript
348 lines
19 KiB
TypeScript
import fetch from 'node-fetch';
|
|
import {db} from '../../src/databases/databases';
|
|
import {Done, getbaseURL} from '../utils';
|
|
import {getHash} from '../../src/utils/getHash';
|
|
import {ImportMock,} from 'ts-mock-imports';
|
|
import * as YouTubeAPIModule from '../../src/utils/youtubeApi';
|
|
import {YouTubeApiMock} from '../youtubeMock';
|
|
|
|
const mockManager = ImportMock.mockStaticClass(YouTubeAPIModule, 'YouTubeAPI');
|
|
const sinonStub = mockManager.mock('listVideos');
|
|
sinonStub.callsFake(YouTubeApiMock.listVideos);
|
|
|
|
describe('getSegmentsByHash', () => {
|
|
before(async () => {
|
|
const query = 'INSERT INTO "sponsorTimes" ("videoID", "startTime", "endTime", "votes", "UUID", "userID", "timeSubmitted", views, category, "actionType", "service", "hidden", "shadowHidden", "hashedVideoID") VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)';
|
|
await db.prepare("run", query, ['getSegmentsByHash-0', 1, 10, 2, 'getSegmentsByHash-0-0', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 0, 0, 'fdaff4dee1043451faa7398324fb63d8618ebcd11bddfe0491c488db12c6c910']);
|
|
await db.prepare("run", query, ['getSegmentsByHash-0', 1, 10, 2, 'getSegmentsByHash-0-0-1', 'testman', 0, 50, 'sponsor', 'skip', 'PeerTube', 0, 0, 'fdaff4dee1043451faa7398324fb63d8618ebcd11bddfe0491c488db12c6c910']);
|
|
await db.prepare("run", query, ['getSegmentsByHash-0', 20, 30, 2, 'getSegmentsByHash-0-1', 'testman', 100, 150, 'intro', 'skip', 'YouTube', 0, 0, 'fdaff4dee1043451faa7398324fb63d8618ebcd11bddfe0491c488db12c6c910']);
|
|
await db.prepare("run", query, ['getSegmentsByHash-0', 40, 50, 2, 'getSegmentsByHash-0-2', 'testman', 0, 50, 'sponsor', 'mute', 'YouTube', 0, 0, 'fdaff4dee1043451faa7398324fb63d8618ebcd11bddfe0491c488db12c6c910']);
|
|
await db.prepare("run", query, ['getSegmentsByHash-noMatchHash', 40, 50, 2, 'getSegmentsByHash-noMatchHash', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 0, 0, 'fdaffnoMatchHash']);
|
|
await db.prepare("run", query, ['getSegmentsByHash-1', 60, 70, 2, 'getSegmentsByHash-1', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 0, 0, '3272fa85ee0927f6073ef6f07ad5f3146047c1abba794cfa364d65ab9921692b']);
|
|
await db.prepare("run", query, ['onlyHidden', 60, 70, 2, 'onlyHidden', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 1, 0, 'f3a199e1af001d716cdc6599360e2b062c2d2b3fa2885f6d9d2fd741166cbbd3']);
|
|
await db.prepare("run", query, ['highlightVid', 60, 60, 2, 'highlightVid-1', 'testman', 0, 50, 'highlight', 'skip', 'YouTube', 0, 0, getHash('highlightVid', 1)]);
|
|
await db.prepare("run", query, ['highlightVid', 70, 70, 2, 'highlightVid-2', 'testman', 0, 50, 'highlight', 'skip', 'YouTube', 0, 0, getHash('highlightVid', 1)]);
|
|
await db.prepare("run", query, ['requiredSegmentVid', 60, 70, 2, 'requiredSegmentVid-1', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 0, 0, 'd51822c3f681e07aef15a8855f52ad12db9eb9cf059e65b16b64c43359557f61']);
|
|
await db.prepare("run", query, ['requiredSegmentVid', 60, 70, -2, 'requiredSegmentVid-2', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 0, 0, 'd51822c3f681e07aef15a8855f52ad12db9eb9cf059e65b16b64c43359557f61']);
|
|
await db.prepare("run", query, ['requiredSegmentVid', 80, 90, -2, 'requiredSegmentVid-3', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 0, 0, 'd51822c3f681e07aef15a8855f52ad12db9eb9cf059e65b16b64c43359557f61']);
|
|
await db.prepare("run", query, ['requiredSegmentVid', 80, 90, 2, 'requiredSegmentVid-4', 'testman', 0, 50, 'sponsor', 'skip', 'YouTube', 0, 0, 'd51822c3f681e07aef15a8855f52ad12db9eb9cf059e65b16b64c43359557f61']);
|
|
});
|
|
|
|
it('Should be able to get a 200', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/3272f?categories=["sponsor", "intro"]')
|
|
.then(res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else done(); // pass
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should return 404 if no segments are found even if a video for the given hash is known', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/3272f?categories=["shilling"]')
|
|
.then(async res => {
|
|
if (res.status !== 404) done("non 404 status code, was " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
if (body === '[]') done(); // pass
|
|
else done("Response had videos");
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get an empty array if no videos', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/11111?categories=["shilling"]')
|
|
.then(async res => {
|
|
if (res.status !== 404) done("non 404 status code, was " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
if (JSON.parse(body).length === 0 && body === '[]') done(); // pass
|
|
else done("non empty array returned");
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get an empty array if only hidden videos', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/f3a1?categories=["sponsor"]')
|
|
.then(async res => {
|
|
if (res.status !== 404) done("non 404 status code, was " + res.status);
|
|
else {
|
|
const body = await res.text();
|
|
if (JSON.parse(body).length === 0 && body === '[]') done(); // pass
|
|
else done("non empty array returned");
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should return 400 prefix too short', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/11?categories=["shilling"]')
|
|
.then(res => {
|
|
if (res.status !== 400) done("non 400 status code, was " + res.status);
|
|
else done(); // pass
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should return 400 prefix too long', (done: Done) => {
|
|
const prefix = new Array(50).join('1');
|
|
if (prefix.length <= 32) { // default value, config can change this
|
|
done('failed to generate a long enough string for the test ' + prefix.length);
|
|
return;
|
|
}
|
|
fetch(getbaseURL() + '/api/skipSegments/' + prefix + '?categories=["shilling"]')
|
|
.then(res => {
|
|
if (res.status !== 400) done("non 400 status code, was " + res.status);
|
|
else done(); // pass
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should not return 400 prefix in range', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/11111?categories=["shilling"]')
|
|
.then(res => {
|
|
if (res.status === 400) done("prefix length 5 gave 400 " + res.status);
|
|
else done(); // pass
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should return 400 for no hash', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/?categories=["shilling"]')
|
|
.then(res => {
|
|
if (res.status !== 400) done("expected 400, got " + res.status);
|
|
else done(); // pass
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should return 400 for bad format categories', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/fdaf?categories=shilling')
|
|
.then(res => {
|
|
if (res.status !== 400) done("expected 400 got " + res.status);
|
|
else done(); // pass
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get multiple videos', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/fdaf?categories=["sponsor","intro"]')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 2) done("expected 2 videos, got " + body.length);
|
|
else if (body[0].segments.length !== 2) done("expected 2 segments for first video, got " + body[0].segments.length);
|
|
else if (body[1].segments.length !== 1) done("expected 1 segment for second video, got " + body[1].segments.length);
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get 200 for no categories (default sponsor)', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/fdaf')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 2) done("expected 2 videos, got " + body.length);
|
|
else if (body[0].segments.length !== 1) done("expected 1 segments for first video, got " + body[0].segments.length);
|
|
else if (body[1].segments.length !== 1) done("expected 1 segments for second video, got " + body[1].segments.length);
|
|
else if (body[0].segments[0].category !== 'sponsor'
|
|
|| body[0].segments[0].UUID !== 'getSegmentsByHash-0-0'
|
|
|| body[1].segments[0].category !== 'sponsor') done("both segments are not sponsor");
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get 200 for no categories (default sponsor) with action type', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/fdaf?actionType=skip')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 2) done("expected 2 videos, got " + body.length);
|
|
else if (body[0].segments.length !== 1) done("expected 1 segments for first video, got " + body[0].segments.length);
|
|
else if (body[1].segments.length !== 1) done("expected 1 segments for second video, got " + body[1].segments.length);
|
|
else if (body[0].segments[0].category !== 'sponsor'
|
|
|| body[0].segments[0].UUID !== 'getSegmentsByHash-0-0'
|
|
|| body[1].segments[0].category !== 'sponsor') done("both segments are not sponsor");
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get 200 for no categories (default sponsor) with multiple action types', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/fdaf?actionType=skip&actionType=mute')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 2) done("expected 2 videos, got " + body.length);
|
|
else if (body[0].segments.length !== 2) done("expected 2 segments for first video, got " + body[0].segments.length);
|
|
else if (body[1].segments.length !== 1) done("expected 1 segments for second video, got " + body[1].segments.length);
|
|
else if (body[0].segments[0].category !== 'sponsor'
|
|
|| body[0].segments[0].UUID !== 'getSegmentsByHash-0-0'
|
|
|| body[0].segments[1].UUID !== 'getSegmentsByHash-0-2'
|
|
|| body[1].segments[0].category !== 'sponsor') done("both segments are not sponsor");
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get 200 for no categories (default sponsor) with multiple action types (JSON array)', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/fdaf?actionTypes=["skip","mute"]')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 2) done("expected 2 videos, got " + body.length);
|
|
else if (body[0].segments.length !== 2) done("expected 2 segments for first video, got " + body[0].segments.length);
|
|
else if (body[1].segments.length !== 1) done("expected 1 segments for second video, got " + body[1].segments.length);
|
|
else if (body[0].segments[0].category !== 'sponsor'
|
|
|| body[0].segments[0].UUID !== 'getSegmentsByHash-0-0'
|
|
|| body[0].segments[1].UUID !== 'getSegmentsByHash-0-2'
|
|
|| body[1].segments[0].category !== 'sponsor') done("both segments are not sponsor");
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get 200 for no categories (default sponsor) for a non YouTube service', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/fdaf?service=PeerTube')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 1) done("expected 1 video, got " + body.length);
|
|
else if (body[0].segments.length !== 1) done("expected 1 segments for first video, got " + body[0].segments.length);
|
|
else if (body[0].segments[0].UUID !== 'getSegmentsByHash-0-0-1') done("both segments are not sponsor");
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should only return one segment when fetching highlight segments', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/c962?category=highlight')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 1) done("expected 1 video, got " + body.length);
|
|
else if (body[0].segments.length !== 1) done("expected 1 segment, got " + body[0].segments.length);
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to post a segment and get it using endpoint', (done: Done) => {
|
|
const testID = 'abc123goodVideo';
|
|
fetch(getbaseURL() + "/api/postVideoSponsorTimes", {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
},
|
|
body: JSON.stringify({
|
|
userID: "test-qwertyuiopasdfghjklzxcvbnm",
|
|
videoID: testID,
|
|
segments: [{
|
|
segment: [13, 17],
|
|
category: "sponsor",
|
|
}],
|
|
}),
|
|
})
|
|
.then(async res => {
|
|
if (res.status === 200) {
|
|
fetch(getbaseURL() + '/api/skipSegments/' + getHash(testID, 1).substring(0, 3))
|
|
.then(async res => {
|
|
if (res.status !== 200) done("(get) non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 1) done("(get) expected 1 video, got " + body.length);
|
|
else if (body[0].segments.length !== 1) done("(get) expected 1 segments for first video, got " + body[0].segments.length);
|
|
else if (body[0].segments[0].category !== 'sponsor') done("(get) segment should be sponsor, was " + body[0].segments[0].category);
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("(get) Couldn't call endpoint"));
|
|
} else {
|
|
done("(post) non 200 status code, was " + res.status);
|
|
}
|
|
})
|
|
.catch(err => done('(post) ' + err));
|
|
});
|
|
|
|
it('Should be able to get multiple categories with repeating parameters', (done: Done) => {
|
|
fetch(getbaseURL() + "/api/skipSegments/fdaff4?&category=sponsor&category=intro")
|
|
.then(async res => {
|
|
if (res.status !== 200) done("Status code was: " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 1) done("expected 1 video, got " + body.length);
|
|
|
|
const data = body[0].segments;
|
|
if (data.length === 2) {
|
|
let success = true;
|
|
for (const segment of data) {
|
|
if ((segment.segment[0] !== 1 || segment.segment[1] !== 10
|
|
|| segment.category !== "sponsor" || segment.UUID !== "getSegmentsByHash-0-0") &&
|
|
(segment.segment[0] !== 20 || segment.segment[1] !== 30
|
|
|| segment.category !== "intro" || segment.UUID !== "getSegmentsByHash-0-1")) {
|
|
success = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (success) done();
|
|
else done("Received incorrect body: " + JSON.stringify(body));
|
|
} else {
|
|
done("Received incorrect body: " + JSON.stringify(body));
|
|
}
|
|
}
|
|
})
|
|
.catch(() => ("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get specific segments with requiredSegments', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/d518?requiredSegments=["requiredSegmentVid-2","requiredSegmentVid-3"]')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 1) done("expected 1 video, got " + body.length);
|
|
else if (body[0].segments.length !== 2) done("expected 2 segments for video, got " + body[0].segments.length);
|
|
else if (body[0].segments[0].UUID !== 'requiredSegmentVid-2'
|
|
|| body[0].segments[1].UUID !== 'requiredSegmentVid-3') done("Did not recieve the correct segments\n" + JSON.stringify(body, null, 2));
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
|
|
it('Should be able to get specific segments with repeating requiredSegment', (done: Done) => {
|
|
fetch(getbaseURL() + '/api/skipSegments/d518?requiredSegment=requiredSegmentVid-2&requiredSegment=requiredSegmentVid-3')
|
|
.then(async res => {
|
|
if (res.status !== 200) done("non 200 status code, was " + res.status);
|
|
else {
|
|
const body = await res.json();
|
|
if (body.length !== 1) done("expected 1 video, got " + body.length);
|
|
else if (body[0].segments.length !== 2) done("expected 2 segments for video, got " + body[0].segments.length);
|
|
else if (body[0].segments[0].UUID !== 'requiredSegmentVid-2'
|
|
|| body[0].segments[1].UUID !== 'requiredSegmentVid-3') done("Did not recieve the correct segments\n" + JSON.stringify(body, null, 2));
|
|
else done();
|
|
}
|
|
})
|
|
.catch(() => done("Couldn't call endpoint"));
|
|
});
|
|
});
|
|
|