SponsorBlockServer/test/cases/getSkipSegments.js

var request = require('request');
var db = require('../../src/databases/databases.js').db;
var utils = require('../utils.js');
var getHash = require('../../src/utils/getHash.js');


describe('getSkipSegments', () => {
  before(() => {
    let startOfQuery = "INSERT INTO sponsorTimes (videoID, startTime, endTime, votes, UUID, userID, timeSubmitted, views, category, shadowHidden, hashedVideoID) VALUES";
    db.exec(startOfQuery + "('testtesttest', 1, 11, 2, '1-uuid-0', 'testman', 0, 50, 'sponsor', 0, '" + getHash('testtesttest', 1) + "')");
    db.exec(startOfQuery + "('testtesttest', 20, 33, 2, '1-uuid-2', 'testman', 0, 50, 'intro', 0, '" + getHash('testtesttest', 1) + "')");
    db.exec(startOfQuery + "('testtesttest,test', 1, 11, 2, '1-uuid-1', 'testman', 0, 50, 'sponsor', 0, '" + getHash('testtesttest,test', 1) + "')");
    db.exec(startOfQuery + "('test3', 1, 11, 2, '1-uuid-4', 'testman', 0, 50, 'sponsor', 0, '" + getHash('test3', 1) + "')");
    db.exec(startOfQuery + "('test3', 7, 22, -3, '1-uuid-5', 'testman', 0, 50, 'sponsor', 0, '" + getHash('test3', 1) + "')");
    db.exec(startOfQuery + "('multiple', 1, 11, 2, '1-uuid-6', 'testman', 0, 50, 'intro', 0, '" + getHash('multiple', 1) + "')");
    db.exec(startOfQuery + "('multiple', 20, 33, 2, '1-uuid-7', 'testman', 0, 50, 'intro', 0, '" + getHash('multiple', 1) + "')");
  }); 
  

  it('Should be able to get a time by category 1', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/skipSegments?videoID=testtesttest&category=sponsor", null, 
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.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") {
                    done();
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Should be able to get a time by category 2', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/skipSegments?videoID=testtesttest&category=intro", null, 
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.body);
            if (data.length === 1 && data[0].segment[0] === 20 && data[0].segment[1] === 33
                && data[0].category === "intro" && data[0].UUID === "1-uuid-2") {
                    done();
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Should be able to get a time by categories array', (done) => {
    request.get(utils.getbaseURL() 
    + "/api/skipSegments?videoID=testtesttest&categories=[\"sponsor\"]", null, 
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.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") {
                    done();
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Should be able to get a time by categories array 2', (done) => {
    request.get(utils.getbaseURL() 
    + "/api/skipSegments?videoID=testtesttest&categories=[\"intro\"]", null,
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.body);
            if (data.length === 1 && data[0].segment[0] === 20 && data[0].segment[1] === 33
                && data[0].category === "intro" && data[0].UUID === "1-uuid-2") {
                    done();
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Should be able to get multiple times by category', (done) => {
    request.get(utils.getbaseURL() 
    + "/api/skipSegments?videoID=multiple&categories=[\"intro\"]", null,
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.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) done();
                else done("Received incorrect body: " + res.body);
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Should be able to get multiple times by multiple categories', (done) => {
    request.get(utils.getbaseURL() 
      + "/api/skipSegments?videoID=testtesttest&categories=[\"sponsor\", \"intro\"]", null,
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.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) done();
                else done("Received incorrect body: " + res.body);
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Should be possible to send unexpected query parameters', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/skipSegments?videoID=testtesttest&fakeparam=hello&category=sponsor", null, 
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.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") {
                    done();
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Low voted submissions should be hidden', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/skipSegments?videoID=test3&category=sponsor", null, 
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = JSON.parse(res.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") {
                    done();
            } else {
                done("Received incorrect body: " + res.body);
            }
        }
      });
  });

  it('Should return 404 if no segment found', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/skipSegments?videoID=notarealvideo", null, 
      (err, res, body) => {
        if (err) done("couldn't call endpoint");
        else if (res.statusCode !== 404) done("non 404 respone code: " + res.statusCode);
        else done(); // pass
      });
  });
  

  it('Should be able send a comma in a query param', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/skipSegments?videoID=testtesttest,test&category=sponsor", null, 
      (err, res, body) => {
            if (err) done("Couldn't call endpoint");
            else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
            else {
                let data = JSON.parse(res.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") {
                        done();
                } else {
                    done("Received incorrect body: " + res.body);
                }
            }
      });
  });

});