YouTube.js/test/index.js

'use strict';

const Fs = require('fs');
const Innertube = require('..');
const NToken = require('../lib/NToken');
const SigDecipher = require('../lib/Sig');
const Constants = require('./constants');

let failed_tests = 0;

async function performTests() {
  const youtube = await new Innertube().catch((error) => error);
  assert(youtube instanceof Error ? false : true, `should retrieve Innertube configuration data`, youtube);

  if (!(youtube instanceof Error)) {
    const search = await youtube.search('Carl Sagan - Documentary').catch((error) => error);
    assert((search instanceof Error ? false : true) && search.videos.length >= 1, `should search videos`, search);

    const details = await youtube.getDetails(Constants.test_video_id).catch((error) => error);
    assert(details instanceof Error ? false : true, `should retrieve details for ${Constants.test_video_id}`, details);

    const comments = await youtube.getComments(Constants.test_video_id).catch((error) => error);
    assert(comments instanceof Error ? false : true, `should retrieve comments for ${Constants.test_video_id}`, comments);

    const video = await downloadVideo(Constants.test_video_id_1, youtube).catch((error) => error);
    assert(video instanceof Error ? false : true, `should download video (${Constants.test_video_id_1})`, video);
  }

  const n_token = new NToken(Constants.n_scramble_sc).transform(Constants.original_ntoken);
  assert(n_token == Constants.expected_ntoken, `should transform n token into ${Constants.expected_ntoken}`, n_token);

  const transformed_url = new SigDecipher(Constants.test_url, Constants.client_version, { sig_decipher_sc: Constants.sig_decipher_sc, ntoken_sc: Constants.n_scramble_sc }).decipher();
  assert(transformed_url == Constants.expected_url, `should correctly decipher signature`, transformed_url);

  if (failed_tests > 0)
    throw new Error('Some tests have failed');
}

function downloadVideo(id, youtube) {
  return new Promise((resolve, reject) => {
    let got_video_info = false;
    const stream = youtube.download(id, { type: 'videoandaudio' });
    stream.pipe(Fs.createWriteStream(`./${id}.mp4`));
    stream.on('end', () => Fs.existsSync(`./${id}.mp4`) && got_video_info && resolve() || reject());
    stream.on('info', () => got_video_info = true);
    stream.on('error', (err) => reject(err));
  });
}

function assert(outcome, description, data) {
  const pass_fail = outcome ? 'pass' : 'fail';
  !outcome && (failed_tests += 1);
  console.info(pass_fail, ':', description, !outcome && `\nError: ${data}` || '');
  return outcome;
};

performTests();