fune/toolkit/components/extensions/test/xpcshell/test_ext_contentscript_scriptCreated.js

"use strict";

const server = createHttpServer();
server.registerDirectory("/data/", do_get_file("data"));

const BASE_URL = `http://localhost:${server.identity.primaryPort}/data`;

// ExtensionContent.jsm needs to know when it's running from xpcshell,
// to use the right timeout for content scripts executed at document_idle.
ExtensionTestUtils.mockAppInfo();

// Test that document_start content scripts don't block script-created
// parsers.
add_task(async function test_contentscript_scriptCreated() {
  let extensionData = {
    manifest: {
      content_scripts: [
        {
          matches: ["http://*/*/file_document_write.html"],
          js: ["content_script.js"],
          run_at: "document_start",
          match_about_blank: true,
          all_frames: true,
        },
      ],
    },

    files: {
      "content_script.js": function () {
        if (window === top) {
          addEventListener(
            "message",
            msg => {
              browser.test.assertEq(
                "ok",
                msg.data,
                "document.write() succeeded"
              );
              browser.test.sendMessage("content-script-done");
            },
            { once: true }
          );
        }
      },
    },
  };

  let extension = ExtensionTestUtils.loadExtension(extensionData);

  await extension.startup();

  let contentPage = await ExtensionTestUtils.loadContentPage(
    `${BASE_URL}/file_document_write.html`
  );

  await extension.awaitMessage("content-script-done");

  await contentPage.close();

  await extension.unload();
});