mirror of
https://github.com/ajayyy/SponsorBlockServer.git
synced 2025-12-22 23:48:27 +03:00
fixed tests, typos and optimized code for ratings
This commit is contained in:
@@ -8,15 +8,19 @@ import { partialDeepEquals } from "../../utils/partialDeepEquals";
|
||||
const endpoint = "/api/ratings/rate/";
|
||||
const getRating = (hash: string, params?: unknown): Promise<AxiosResponse> => client.get(endpoint + hash, { params });
|
||||
|
||||
const videoOneID = "some-likes-and-dislikes";
|
||||
const videoOneIDHash = getHash(videoOneID, 1);
|
||||
const videoOnePartialHash = videoOneIDHash.substr(0, 4);
|
||||
|
||||
describe("getRating", () => {
|
||||
before(async () => {
|
||||
const insertUserNameQuery = 'INSERT INTO "ratings" ("videoID", "service", "type", "count", "hashedVideoID") VALUES (?, ?, ?, ?, ?)';
|
||||
await db.prepare("run", insertUserNameQuery, ["some-likes-and-dislikes", "YouTube", 0, 5, getHash("some-likes-and-dislikes", 1)]); //b3f0
|
||||
await db.prepare("run", insertUserNameQuery, ["some-likes-and-dislikes", "YouTube", 1, 10, getHash("some-likes-and-dislikes", 1)]);
|
||||
await db.prepare("run", insertUserNameQuery, [videoOneID, "YouTube", 0, 5, videoOneIDHash]);
|
||||
await db.prepare("run", insertUserNameQuery, [videoOneID, "YouTube", 1, 10, videoOneIDHash]);
|
||||
});
|
||||
|
||||
it("Should be able to get dislikes and likes by default", (done) => {
|
||||
getRating("b3f0")
|
||||
getRating(videoOnePartialHash)
|
||||
.then(res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expected = [{
|
||||
@@ -33,7 +37,7 @@ describe("getRating", () => {
|
||||
});
|
||||
|
||||
it("Should be able to filter for only dislikes", (done) => {
|
||||
getRating("b3f0", { type: 0 })
|
||||
getRating(videoOnePartialHash, { type: 0 })
|
||||
.then(res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expected = [{
|
||||
@@ -46,4 +50,31 @@ describe("getRating", () => {
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should return 400 for invalid hash", (done) => {
|
||||
getRating("a")
|
||||
.then(res => {
|
||||
assert.strictEqual(res.status, 400);
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should return 404 for nonexitent type", (done) => {
|
||||
getRating(videoOnePartialHash, { type: 100 })
|
||||
.then(res => {
|
||||
assert.strictEqual(res.status, 404);
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should return 404 for nonexistent videoID", (done) => {
|
||||
getRating("aaaa")
|
||||
.then(res => {
|
||||
assert.strictEqual(res.status, 404);
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
});
|
||||
@@ -9,37 +9,43 @@ const endpoint = "/api/ratings/rate/";
|
||||
const postRating = (body: unknown): Promise<AxiosResponse> => client.post(endpoint, body);
|
||||
const queryDatabase = (videoID: string) => db.prepare("all", `SELECT * FROM "ratings" WHERE "videoID" = ?`, [videoID]);
|
||||
|
||||
const videoIDOne = "normal-video";
|
||||
const videoIDTwo = "multiple-rates";
|
||||
const ratingUserID = "rating-testman";
|
||||
|
||||
describe("postRating", () => {
|
||||
before(async () => {
|
||||
const insertUserNameQuery = 'INSERT INTO "ratings" ("videoID", "service", "type", "count", "hashedVideoID") VALUES (?, ?, ?, ?, ?)';
|
||||
await db.prepare("run", insertUserNameQuery, ["multiple-rates", "YouTube", 0, 3, getHash("multiple-rates", 1)]);
|
||||
await db.prepare("run", insertUserNameQuery, [videoIDTwo, "YouTube", 0, 3, getHash(videoIDTwo, 1)]);
|
||||
});
|
||||
|
||||
it("Should be able to vote on a video", (done) => {
|
||||
const videoID = videoIDOne;
|
||||
postRating({
|
||||
userID: "rating-testman",
|
||||
videoID: "normal-video",
|
||||
userID: ratingUserID,
|
||||
videoID,
|
||||
type: 0
|
||||
})
|
||||
.then(async res => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const expected = [{
|
||||
hashedVideoID: getHash("normal-video", 1),
|
||||
videoID: "normal-video",
|
||||
hashedVideoID: getHash(videoID, 1),
|
||||
videoID,
|
||||
type: 0,
|
||||
count: 1,
|
||||
service: "YouTube"
|
||||
}];
|
||||
assert.ok(partialDeepEquals(await queryDatabase("normal-video"), expected));
|
||||
assert.ok(partialDeepEquals(await queryDatabase(videoID), expected));
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should be able to undo a vote on a video", (done) => {
|
||||
const videoID = videoIDOne;
|
||||
postRating({
|
||||
userID: "rating-testman",
|
||||
videoID: "normal-video",
|
||||
userID: ratingUserID,
|
||||
videoID,
|
||||
type: 0,
|
||||
enabled: false
|
||||
})
|
||||
@@ -49,16 +55,17 @@ describe("postRating", () => {
|
||||
type: 0,
|
||||
count: 0
|
||||
}];
|
||||
assert.ok(partialDeepEquals(await queryDatabase("normal-video"), expected));
|
||||
assert.ok(partialDeepEquals(await queryDatabase(videoID), expected));
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should be able to vote after someone else on a video", (done) => {
|
||||
const videoID = videoIDTwo;
|
||||
postRating({
|
||||
userID: "rating-testman",
|
||||
videoID: "multiple-rates",
|
||||
userID: ratingUserID,
|
||||
videoID,
|
||||
type: 0
|
||||
})
|
||||
.then(async res => {
|
||||
@@ -67,16 +74,17 @@ describe("postRating", () => {
|
||||
type: 0,
|
||||
count: 4
|
||||
}];
|
||||
assert.ok(partialDeepEquals(await queryDatabase("multiple-rates"), expected));
|
||||
assert.ok(partialDeepEquals(await queryDatabase(videoID), expected));
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should be able to vote a different type than existing votes on a video", (done) => {
|
||||
const videoID = videoIDTwo;
|
||||
postRating({
|
||||
userID: "rating-testman",
|
||||
videoID: "multiple-rates",
|
||||
userID: ratingUserID,
|
||||
videoID,
|
||||
type: 1
|
||||
})
|
||||
.then(async res => {
|
||||
@@ -88,7 +96,21 @@ describe("postRating", () => {
|
||||
type: 1,
|
||||
count: 1
|
||||
}];
|
||||
assert.ok(partialDeepEquals(await queryDatabase("multiple-rates"), expected));
|
||||
assert.ok(partialDeepEquals(await queryDatabase(videoID), expected));
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
});
|
||||
|
||||
it("Should not be able to vote with nonexistent type", (done) => {
|
||||
const videoID = videoIDOne;
|
||||
postRating({
|
||||
userID: ratingUserID,
|
||||
videoID,
|
||||
type: 100
|
||||
})
|
||||
.then(res => {
|
||||
assert.strictEqual(res.status, 400);
|
||||
done();
|
||||
})
|
||||
.catch(err => done(err));
|
||||
|
||||
Reference in New Issue
Block a user