SponsorBlockServer/test/cases/voteOnSponsorTime.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('voteOnSponsorTime', () => {
  before(() => {
    let startOfQuery = "INSERT INTO sponsorTimes (videoID, startTime, endTime, votes, UUID, userID, timeSubmitted, views, category, shadowHidden) VALUES";
    db.exec(startOfQuery + "('vote-testtesttest', 1, 11, 2, 'vote-uuid-0', 'testman', 0, 50, 'sponsor', 0)");
    db.exec(startOfQuery + "('vote-testtesttest', 20, 33, 10, 'vote-uuid-2', 'testman', 0, 50, 'intro', 0)");
    db.exec(startOfQuery + "('vote-testtesttest,test', 1, 11, 100, 'vote-uuid-3', 'testman', 0, 50, 'sponsor', 0)");
    db.exec(startOfQuery + "('vote-test3', 1, 11, 2, 'vote-uuid-4', 'testman', 0, 50, 'sponsor', 0)");
    db.exec(startOfQuery + "('vote-test3', 7, 22, -3, 'vote-uuid-5', 'testman', 0, 50, 'intro', 0)");
    db.exec(startOfQuery + "('vote-multiple', 1, 11, 2, 'vote-uuid-6', 'testman', 0, 50, 'intro', 0)");
    db.exec(startOfQuery + "('vote-multiple', 20, 33, 2, 'vote-uuid-7', 'testman', 0, 50, 'intro', 0)");

    db.exec("INSERT INTO vipUsers (userID) VALUES ('" + getHash("VIPUser") + "')");
  }); 
  

  it('Should be able to upvote a segment', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=randomID&UUID=vote-uuid-0&type=1", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 200) {
          let row = db.prepare("SELECT votes FROM sponsorTimes WHERE UUID = ?").get("vote-uuid-0");
          if (row.votes === 3) {
            done()
          } else {
            done("Vote did not succeed. Submission went from 2 votes to " + row.votes);
          }
        } else {
          done("Status code was " + res.statusCode);
        }
    });
  });

  it('Should be able to downvote a segment', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=randomID2&UUID=vote-uuid-2&type=0", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 200) {
          let row = db.prepare("SELECT votes FROM sponsorTimes WHERE UUID = ?").get("vote-uuid-2");
          if (row.votes < 10) {
            done()
          } else {
            done("Vote did not succeed. Submission went from 10 votes to " + row.votes);
          }
        } else {
          done("Status code was " + res.statusCode);
        }
    });
  });

  it('VIP should be able to completely downvote a segment', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=VIPUser&UUID=vote-uuid-3&type=0", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 200) {
          let row = db.prepare("SELECT votes FROM sponsorTimes WHERE UUID = ?").get("vote-uuid-3");
          if (row.votes <= -2) {
            done()
          } else {
            done("Vote did not succeed. Submission went from 100 votes to " + row.votes);
          }
        } else {
          done("Status code was " + res.statusCode);
        }
    });
  });

  it('Should be able to vote for a category and it should immediately change (for now)', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=randomID2&UUID=vote-uuid-4&category=intro", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 200) {
          let row = db.prepare("SELECT category FROM sponsorTimes WHERE UUID = ?").get("vote-uuid-4");
          if (row.category === "intro") {
            done()
          } else {
            done("Vote did not succeed. Submission went from sponsor to " + row.category);
          }
        } else {
          done("Status code was " + res.statusCode);
        }
    });
  });

  it('Should be able to change your vote for a category and it should immediately change (for now)', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=randomID2&UUID=vote-uuid-4&category=outro", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 200) {
          let row = db.prepare("SELECT category FROM sponsorTimes WHERE UUID = ?").get("vote-uuid-4");
          if (row.category === "outro") {
            done()
          } else {
            done("Vote did not succeed. Submission went from intro to " + row.category);
          }
        } else {
          done("Status code was " + res.statusCode);
        }
    });
  });

  it('VIP should be able to vote for a category and it should immediately change', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=VIPUser&UUID=vote-uuid-5&category=outro", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 200) {
          let row = db.prepare("SELECT category FROM sponsorTimes WHERE UUID = ?").get("vote-uuid-5");
          let row2 = db.prepare("SELECT votes FROM categoryVotes WHERE UUID = ? and category = ?").get("vote-uuid-5", "outro");
          if (row.category === "outro" && row2.votes === 500) {
            done()
          } else {
            done("Vote did not succeed. Submission went from intro to " + row.category + ". Category votes are " + row2.votes + " and should be 500.");
          }
        } else {
          done("Status code was " + res.statusCode);
        }
    });
  });

  it('Should not be able to category-vote on an invalid UUID submission', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=randomID3&UUID=invalid-uuid&category=intro", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 400) {
          done();
        } else {
          done("Status code was " + res.statusCode + " instead of 400.");
        }
    });
  });

  it('Non-VIP should not be able to upvote "dead" submission', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=randomID2&UUID=vote-uuid-5&type=1", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 403) {
          done();
        } else {
          done("Status code was " + res.statusCode + " instead of 403");
        }
    });
  });

  it('VIP should be able to upvote "dead" submission', (done) => {
    request.get(utils.getbaseURL() 
     + "/api/voteOnSponsorTime?userID=VIPUser&UUID=vote-uuid-5&type=1", null, 
      (err, res, body) => {
        if (err) done(err);
        else if (res.statusCode === 200) {
          let row = db.prepare("SELECT votes FROM sponsorTimes WHERE UUID = ?").get("vote-uuid-5");
          if (row.votes > -3) {
            done()
          } else {
            done("Vote did not succeed. Votes raised from -3 to " + row.votes);
          }
        } else {
          done("Status code was " + res.statusCode);
        }
    });
  });

});