SponsorBlockServer/test/cases/getSkipSegments.js

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

/*
 *CREATE TABLE IF NOT EXISTS "sponsorTimes" (
	"videoID"	TEXT NOT NULL,
	"startTime"	REAL NOT NULL,
	"endTime"	REAL NOT NULL,
	"votes"	INTEGER NOT NULL,
	"UUID"	TEXT NOT NULL UNIQUE,
	"userID"	TEXT NOT NULL,
	"timeSubmitted"	INTEGER NOT NULL,
	"views"	INTEGER NOT NULL,
	"shadowHidden"	INTEGER NOT NULL
);
 */

describe('getSkipSegments', () => {
  before(() => {
    db.exec("INSERT INTO sponsorTimes VALUES ('testtesttest', 1, 11, 2, '1-uuid-0', 'testman', 0, 50, 'sponsor', 0)");
    db.exec("INSERT INTO sponsorTimes VALUES ('testtesttest', 20, 33, 2, '1-uuid-2', 'testman', 0, 50, 'intro', 0)");
    db.exec("INSERT INTO sponsorTimes VALUES ('testtesttest,test', 1, 11, 2, '1-uuid-1', 'testman', 0, 50, 'sponsor', 0)");
    db.exec("INSERT INTO sponsorTimes VALUES ('test3', 1, 11, 2, '1-uuid-4', 'testman', 0, 50, 'sponsor', 0)");
    db.exec("INSERT INTO sponsorTimes VALUES ('test3', 7, 22, -3, '1-uuid-5', 'testman', 0, 50, 'sponsor', 0)");
    db.exec("INSERT INTO sponsorTimes VALUES ('multiple', 1, 11, 2, '1-uuid-6', 'testman', 0, 50, 'intro', 0)");
    db.exec("INSERT INTO sponsorTimes VALUES ('multiple', 20, 33, 2, '1-uuid-7', 'testman', 0, 50, 'intro', 0)");
  }); 
  

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

  it('Should be able to get a time by category (JSON Method) 2', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/skipSegments", {
        json: {
           videoID: "testtesttest",
           categories: ["intro"]
        }
      }, 
      (err, res, body) => {
        if (err) done("Couldn't call endpoint");
        else if (res.statusCode !== 200) done("Status code was: " + res.statusCode);
        else {
            let data = 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: " + JSON.stringify(res.body));
            }
        }
      });
  });

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

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

});