browsertrix-crawler/tests/blockrules.test.js

import child_process from "child_process";
import fs from "fs";
import yaml from "js-yaml";

function runCrawl(name, config, commandExtra = "") {
  config.generateCDX = true;
  config.depth = 0;
  config.collection = name;

  const configYaml = yaml.dump(config);

  try {
    const proc = child_process.execSync(
      `docker run -i -v $PWD/test-crawls:/crawls webrecorder/browsertrix-crawler crawl --config stdin ${commandExtra}`,
      { input: configYaml, stdin: "inherit", encoding: "utf8" },
    );

    console.log(proc);
  } catch (error) {
    console.log(error);
  }
}

function doesCDXContain(coll, value) {
  const data = fs.readFileSync(
    `test-crawls/collections/${coll}/indexes/index.cdxj`,
  );
  return data.indexOf(value) >= 0;
}

function checkVideo(coll) {
  return doesCDXContain(coll, '"video/mp4"') || doesCDXContain(coll, '"application/vnd.yt-ump"');
}

// Test Disabled for Brave -- should always be blocked, but seeing inconsistent ci behavior
/*
test("test crawl without block for specific URL", () => {
  const config = {
    "url": "https://www.iana.org/",
    "pageExtraDelay": 10
  };

  runCrawl("block-1-no-block", config);

  // without blocks, URL with add sense is included
  expect(doesCDXContain("block-1-no-block", "https://cse.google.com/adsense/search/async-ads.js")).toBe(true);
});
*/

test("test block rule on specific URL", () => {
  const config = {
    url: "https://www.iana.org/",
    blockRules: [{ url: "adsense" }],
  };

  runCrawl("block-1", config);

  expect(
    doesCDXContain(
      "block-1",
      "https://cse.google.com/adsense/search/async-ads.js",
    ),
  ).toBe(false);
});

test("test block rule based on iframe text, content included due to match", () => {
  const config = {
    url: "https://oembed.link/https://www.youtube.com/watch?v=aT-Up5Y4uRI",
    blockRules: [
      {
        url: "https://www.youtube.com/embed/",
        frameTextMatch:
          '\\\\"channelId\\\\":\\\\"UCrQElMF25VP-1JjhBuFsW_Q\\\\"',
        type: "allowOnly",
      },
    ],
  };

  runCrawl("block-2", config);

  expect(checkVideo("block-2")).toBe(true);
});

test("test block rule based on iframe text, wrong text, content should be excluded", () => {
  const config = {
    url: "https://oembed.link/https://www.youtube.com/watch?v=aT-Up5Y4uRI",
    blockRules: [
      {
        url: "https://www.youtube.com/embed/",
        frameTextMatch:
          '\\\\"channelId\\\\":\\\\"UCrQElMF25VP-1JjhBuFsW_R\\\\"',
        type: "allowOnly",
      },
    ],
  };

  runCrawl("block-3", config);

  expect(checkVideo("block-3")).toBe(false);
});

test("test block rule based on iframe text, block matched", () => {
  const config = {
    url: "https://oembed.link/https://www.youtube.com/watch?v=aT-Up5Y4uRI",
    blockRules: [
      {
        url: "https://www.youtube.com/embed/",
        frameTextMatch:
          '\\\\"channelId\\\\":\\\\"UCrQElMF25VP-1JjhBuFsW_Q\\\\"',
      },
    ],
  };

  runCrawl("block-4", config);

  expect(checkVideo("block-4")).toBe(false);
});

test("test rule based on iframe text not matching, plus allowOnly iframe", () => {
  const config = {
    url: "https://oembed.link/https://www.youtube.com/watch?v=aT-Up5Y4uRI",
    blockRules: [
      {
        url: "example.com/embed/",
        frameTextMatch:
          '\\\\"channelId\\\\":\\\\"UCrQElMF25VP-1JjhBuFsW_Q\\\\"',
        type: "block",
      },
      {
        url: "(youtube.com|example.com)/embed/",
        type: "allowOnly",
        inFrameUrl: "oembed.link/",
      },
    ],
  };

  runCrawl("non-block-5", config);

  expect(checkVideo("non-block-5")).toBe(true);
});

test("test block url in frame url", () => {
  const config = {
    url: "https://oembed.link/https://www.youtube.com/watch?v=aT-Up5Y4uRI",
    blockRules: [
      {
        url: "maxresdefault.jpg",
        type: "block",
        inFrameUrl: "youtube.com/embed",
      },
    ],
  };

  runCrawl("block-6", config);

  expect(
    doesCDXContain(
      "block-6",
      '"https://i.ytimg.com/vi/aT-Up5Y4uRI/maxresdefault.jpg"',
    ),
  ).toBe(false);
});

test("test block rules complex example, block external urls on main frame, but not on youtube", () => {
  const config = {
    seeds: ["https://archiveweb.page/en/troubleshooting/errors/"],
    depth: "0",
    blockRules: [
      {
        url: "(archiveweb.page|www.youtube.com)",
        type: "allowOnly",
        inFrameUrl: "archiveweb.page",
      },
      {
        url: "https://archiveweb.page/assets/js/vendor/lunr.min.js",
        inFrameUrl: "archiveweb.page",
      },
      {
        url: "https://www.youtube.com/embed/",
        type: "allowOnly",
        frameTextMatch:
          '(\\\\"channelId\\\\":\\\\"UCOHO8gYUWpDYFWHXmIwE02g\\\\")',
      },
    ],

    combineWARC: true,

    logging: "stats,debug",
  };

  runCrawl("block-7", config);

  expect(
    doesCDXContain(
      "block-7",
      '"https://archiveweb.page/assets/js/vendor/lunr.min.js"',
    ),
  ).toBe(false);
  expect(checkVideo("block-7")).toBe(true);
});