"use strict";

AddonTestUtils.init(this);
AddonTestUtils.overrideCertDB();
AddonTestUtils.createAppInfo(
  "xpcshell@tests.mozilla.org",
  "XPCShell",
  "1",
  "43"
);

add_setup(async () => {
  await AddonTestUtils.promiseStartupManager();
});

add_task(async function test_simple() {
  let extensionData = {
    manifest: {
      name: "Simple extension test",
      version: "1.0",
      manifest_version: 2,
      description: "",
    },
  };

  let extension = ExtensionTestUtils.loadExtension(extensionData);
  await extension.startup();
  await extension.unload();
});

add_task(async function test_manifest_unsupported_version() {
  let extensionData = {
    manifest: {
      manifest_version: 1,
    },
  };

  let extension = ExtensionTestUtils.loadExtension(extensionData);
  await Assert.rejects(
    extension.startup(),
    /Property "manifest_version" is unsupported in Manifest Version 1/,
    "manifest version 1 is rejected"
  );
});

add_task(async function test_manifest_V3_enabled() {
  let extensionData = {
    manifest: {
      manifest_version: 3,
    },
  };

  let extension = ExtensionTestUtils.loadExtension(extensionData);
  await extension.startup();
  equal(extension.extension.manifest.manifest_version, 3, "manifest V3 loads");
  await extension.unload();
});

add_task(async function test_background() {
  function background() {
    browser.test.log("running background script");

    browser.test.onMessage.addListener((x, y) => {
      browser.test.assertEq(x, 10, "x is 10");
      browser.test.assertEq(y, 20, "y is 20");

      browser.test.notifyPass("background test passed");
    });

    browser.test.sendMessage("running", 1);
  }

  let extensionData = {
    background,
    manifest: {
      name: "Simple extension test",
      version: "1.0",
      manifest_version: 2,
      description: "",
    },
  };

  let extension = ExtensionTestUtils.loadExtension(extensionData);

  let [, x] = await Promise.all([
    extension.startup(),
    extension.awaitMessage("running"),
  ]);
  equal(x, 1, "got correct value from extension");

  extension.sendMessage(10, 20);
  await extension.awaitFinish();
  await extension.unload();
});

add_task(async function test_extensionTypes() {
  let extensionData = {
    background: function () {
      browser.test.assertEq(
        typeof browser.extensionTypes,
        "object",
        "browser.extensionTypes exists"
      );
      browser.test.assertEq(
        typeof browser.extensionTypes.RunAt,
        "object",
        "browser.extensionTypes.RunAt exists"
      );
      browser.test.notifyPass("extentionTypes test passed");
    },
  };

  let extension = ExtensionTestUtils.loadExtension(extensionData);

  await extension.startup();
  await extension.awaitFinish();
  await extension.unload();
});

add_task(async function test_policy_temporarilyInstalled() {
  let extensionData = {
    manifest: {
      manifest_version: 2,
    },
  };

  async function runTest(useAddonManager) {
    let extension = ExtensionTestUtils.loadExtension({
      ...extensionData,
      useAddonManager,
    });

    const expected = useAddonManager === "temporary";
    await extension.startup();
    const { temporarilyInstalled } = WebExtensionPolicy.getByID(extension.id);
    equal(
      temporarilyInstalled,
      expected,
      `Got the expected WebExtensionPolicy.temporarilyInstalled value on "${useAddonManager}"`
    );
    await extension.unload();
  }

  await runTest("temporary");
  await runTest("permanent");
});

add_task(async function test_manifest_allowInsecureRequests() {
  let extensionData = {
    allowInsecureRequests: true,
    manifest: {
      manifest_version: 3,
    },
  };

  let extension = ExtensionTestUtils.loadExtension(extensionData);
  await extension.startup();
  equal(
    extension.extension.manifest.content_security_policy.extension_pages,
    `script-src 'self'`,
    "insecure allowed"
  );
  await extension.unload();
});

add_task(async function test_manifest_allowInsecureRequests_throws() {
  let extensionData = {
    allowInsecureRequests: true,
    manifest: {
      manifest_version: 3,
      content_security_policy: {
        extension_pages: `script-src 'self'`,
      },
    },
  };

  await Assert.throws(
    () => ExtensionTestUtils.loadExtension(extensionData),
    /allowInsecureRequests cannot be used with manifest.content_security_policy/,
    "allowInsecureRequests with content_security_policy cannot be loaded"
  );
});

add_task(async function test_gecko_android_key_in_applications() {
  const xpiFile = AddonTestUtils.createTempWebExtensionFile({
    manifest: {
      manifest_version: 2,
      applications: {
        gecko_android: {},
      },
    },
  });
  ExtensionTestUtils.failOnSchemaWarnings(false);
  let { messages } = await promiseConsoleOutput(async () => {
    const { AddonManager } = ChromeUtils.importESModule(
      "resource://gre/modules/AddonManager.sys.mjs"
    );
    await Assert.rejects(
      AddonManager.installTemporaryAddon(xpiFile),
      /Extension is invalid/,
      "Install failed with privileged permission"
    );
  });
  ExtensionTestUtils.failOnSchemaWarnings(true);
  AddonTestUtils.checkMessages(
    messages,
    {
      expected: [
        {
          message:
            /applications: Property "gecko_android" is unsupported by Firefox/,
        },
      ],
    },
    true
  );
  xpiFile.remove(true);
});
