/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

const CONFIG = [
  {
    identifier: "get-engine",
    base: {
      urls: {
        search: {
          base: "https://example.com",
          params: [
            {
              name: "config",
              value: "1",
            },
            {
              name: "is_enterprise",
              value: "false",
            },
            {
              name: "is_enterprise",
              enterpriseValue: "true",
            },
          ],
          searchTermParamName: "search",
        },
        suggestions: {
          base: "https://example.com",
          params: [
            {
              name: "config",
              value: "1",
            },
            {
              name: "is_enterprise",
              enterpriseValue: "yes",
            },
          ],
          searchTermParamName: "suggest",
        },
      },
    },
  },
  {
    identifier: "post-engine",
    base: {
      urls: {
        search: {
          base: "https://example.com",
          method: "POST",
          params: [
            {
              name: "config",
              value: "1",
            },
          ],
          searchTermParamName: "search",
        },
        suggestions: {
          base: "https://example.com",
          method: "POST",
          params: [
            {
              name: "config",
              value: "1",
            },
          ],
          searchTermParamName: "suggest",
        },
      },
    },
  },
];

add_setup(async function () {
  SearchTestUtils.setRemoteSettingsConfig(CONFIG);
  await SearchService.init();
});

add_task(async function test_get_extension() {
  let engine = SearchService.getEngineByName("get-engine");
  Assert.notEqual(engine, null, "Should have found an engine");

  let url = engine.getURLOfType(SearchUtils.URL_TYPE.SEARCH);
  Assert.equal(url.method, "GET", "Search URLs method is GET");

  let submission = engine.getSubmission("foo");
  let expectedURL =
    SearchUtils.MODIFIED_APP_CHANNEL == "esr"
      ? "https://example.com/?config=1&is_enterprise=true&search=foo"
      : "https://example.com/?config=1&is_enterprise=false&search=foo";

  Assert.equal(submission.uri.spec, expectedURL, "Search URLs should match");

  let submissionSuggest = engine.getSubmission(
    "bar",
    SearchUtils.URL_TYPE.SUGGEST_JSON
  );
  expectedURL =
    SearchUtils.MODIFIED_APP_CHANNEL == "esr"
      ? "https://example.com/?config=1&is_enterprise=yes&suggest=bar"
      : "https://example.com/?config=1&suggest=bar";

  Assert.equal(
    submissionSuggest.uri.spec,
    expectedURL,
    "Suggest URLs should match"
  );
});

add_task(async function test_post_extension() {
  let engine = SearchService.getEngineByName("post-engine");
  Assert.ok(!!engine, "Should have found an engine");

  let url = engine.getURLOfType(SearchUtils.URL_TYPE.SEARCH);
  Assert.equal(url.method, "POST", "Search URLs method is POST");

  let submission = engine.getSubmission("foo");
  Assert.equal(
    submission.uri.spec,
    "https://example.com/",
    "Search URLs should match"
  );
  Assert.equal(
    submission.postData.data.data,
    "config=1&search=foo",
    "Search postData should match"
  );

  let submissionSuggest = engine.getSubmission(
    "bar",
    SearchUtils.URL_TYPE.SUGGEST_JSON
  );
  Assert.equal(
    submissionSuggest.uri.spec,
    "https://example.com/",
    "Suggest URLs should match"
  );
  Assert.equal(
    submissionSuggest.postData.data.data,
    "config=1&suggest=bar",
    "Suggest postData should match"
  );
});

add_task(async function test_enterprise_params() {
  await enableEnterprise();

  let engine = SearchService.getEngineByName("get-engine");
  Assert.notEqual(engine, null, "Should have found an engine");

  let url = engine.getURLOfType(SearchUtils.URL_TYPE.SEARCH);
  Assert.equal(url.method, "GET", "Search URLs method is GET");

  let submission = engine.getSubmission("foo");
  Assert.equal(
    submission.uri.spec,
    "https://example.com/?config=1&is_enterprise=true&search=foo",
    "Enterprise parameter should override normal param."
  );

  let submissionSuggest = engine.getSubmission(
    "bar",
    SearchUtils.URL_TYPE.SUGGEST_JSON
  );
  Assert.equal(
    submissionSuggest.uri.spec,
    "https://example.com/?config=1&is_enterprise=yes&suggest=bar",
    "Enterprise parameter should be added."
  );
});