firefox/toolkit/components/contentprefs/tests/unit_cps2/test_setGet.js

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

add_task(async function resetBeforeTests() {
  await reset();
});

add_task(async function get_nonexistent() {
  await getOK(["a.com", "foo"], undefined);
  await getGlobalOK(["foo"], undefined);
  await reset();
});

add_task(async function isomorphicDomains() {
  await set("a.com", "foo", 1);
  await dbOK([["a.com", "foo", 1]]);
  await getOK(["a.com", "foo"], 1);
  await getOK(["http://a.com/huh", "foo"], 1, "a.com");

  await set("http://a.com/huh", "foo", 2);
  await dbOK([["a.com", "foo", 2]]);
  await getOK(["a.com", "foo"], 2);
  await getOK(["http://a.com/yeah", "foo"], 2, "a.com");
  await reset();
});

add_task(async function names() {
  await set("a.com", "foo", 1);
  await dbOK([["a.com", "foo", 1]]);
  await getOK(["a.com", "foo"], 1);

  await set("a.com", "bar", 2);
  await dbOK([
    ["a.com", "foo", 1],
    ["a.com", "bar", 2],
  ]);
  await getOK(["a.com", "foo"], 1);
  await getOK(["a.com", "bar"], 2);

  await setGlobal("foo", 3);
  await dbOK([
    ["a.com", "foo", 1],
    ["a.com", "bar", 2],
    [null, "foo", 3],
  ]);
  await getOK(["a.com", "foo"], 1);
  await getOK(["a.com", "bar"], 2);
  await getGlobalOK(["foo"], 3);

  await setGlobal("bar", 4);
  await dbOK([
    ["a.com", "foo", 1],
    ["a.com", "bar", 2],
    [null, "foo", 3],
    [null, "bar", 4],
  ]);
  await getOK(["a.com", "foo"], 1);
  await getOK(["a.com", "bar"], 2);
  await getGlobalOK(["foo"], 3);
  await getGlobalOK(["bar"], 4);
  await reset();
});

add_task(async function subdomains() {
  await set("a.com", "foo", 1);
  await set("b.a.com", "foo", 2);
  await dbOK([
    ["a.com", "foo", 1],
    ["b.a.com", "foo", 2],
  ]);
  await getOK(["a.com", "foo"], 1);
  await getOK(["b.a.com", "foo"], 2);
  await reset();
});

add_task(async function privateBrowsing() {
  await set("a.com", "foo", 1);
  await set("a.com", "bar", 2);
  await setGlobal("foo", 3);
  await setGlobal("bar", 4);
  await set("b.com", "foo", 5);

  let context = privateLoadContext;
  await set("a.com", "foo", 6, context);
  await setGlobal("foo", 7, context);
  await dbOK([
    ["a.com", "foo", 1],
    ["a.com", "bar", 2],
    [null, "foo", 3],
    [null, "bar", 4],
    ["b.com", "foo", 5],
  ]);
  await getOK(["a.com", "foo", context], 6, "a.com");
  await getOK(["a.com", "bar", context], 2);
  await getGlobalOK(["foo", context], 7);
  await getGlobalOK(["bar", context], 4);
  await getOK(["b.com", "foo", context], 5);

  await getOK(["a.com", "foo"], 1);
  await getOK(["a.com", "bar"], 2);
  await getGlobalOK(["foo"], 3);
  await getGlobalOK(["bar"], 4);
  await getOK(["b.com", "foo"], 5);
  await reset();
});

add_task(async function set_erroneous() {
  do_check_throws(() => cps.set(null, "foo", 1, null));
  do_check_throws(() => cps.set("", "foo", 1, null));
  do_check_throws(() => cps.set("a.com", "", 1, null));
  do_check_throws(() => cps.set("a.com", null, 1, null));
  do_check_throws(() => cps.set("a.com", "foo", undefined, null));
  do_check_throws(() => cps.set("a.com", "foo", 1, null, "bogus"));
  do_check_throws(() => cps.setGlobal("", 1, null));
  do_check_throws(() => cps.setGlobal(null, 1, null));
  do_check_throws(() => cps.setGlobal("foo", undefined, null));
  do_check_throws(() => cps.setGlobal("foo", 1, null, "bogus"));
  await reset();
});

add_task(async function get_erroneous() {
  do_check_throws(() => cps.getByDomainAndName(null, "foo", null, {}));
  do_check_throws(() => cps.getByDomainAndName("", "foo", null, {}));
  do_check_throws(() => cps.getByDomainAndName("a.com", "", null, {}));
  do_check_throws(() => cps.getByDomainAndName("a.com", null, null, {}));
  do_check_throws(() => cps.getByDomainAndName("a.com", "foo", null, null));
  do_check_throws(() => cps.getGlobal("", null, {}));
  do_check_throws(() => cps.getGlobal(null, null, {}));
  do_check_throws(() => cps.getGlobal("foo", null, null));
  await reset();
});

add_task(async function set_invalidateCache() {
  // (1) Set a pref and wait for it to finish.
  await set("a.com", "foo", 1);

  // (2) It should be cached.
  getCachedOK(["a.com", "foo"], true, 1);

  // (3) Set the pref to a new value but don't wait for it to finish.
  cps.set("a.com", "foo", 2, null, {
    handleCompletion() {
      // (6) The pref should be cached after setting it.
      getCachedOK(["a.com", "foo"], true, 2);
    },
  });

  // (4) Group "a.com" and name "foo" should no longer be cached.
  getCachedOK(["a.com", "foo"], false);

  // (5) Call getByDomainAndName.
  let fetchedPref;
  let getPromise = new Promise(resolve =>
    cps.getByDomainAndName("a.com", "foo", null, {
      handleResult(pref) {
        fetchedPref = pref;
      },
      handleCompletion() {
        // (7) Finally, this callback should be called after set's above.
        Assert.ok(!!fetchedPref);
        Assert.equal(fetchedPref.value, 2);
        resolve();
      },
    })
  );

  await getPromise;
  await reset();
});

add_task(async function get_nameOnly() {
  await set("a.com", "foo", 1);
  await set("a.com", "bar", 2);
  await set("b.com", "foo", 3);
  await setGlobal("foo", 4);

  await getOKEx(
    "getByName",
    ["foo", undefined],
    [
      { domain: "a.com", name: "foo", value: 1 },
      { domain: "b.com", name: "foo", value: 3 },
      { domain: null, name: "foo", value: 4 },
    ]
  );

  let context = privateLoadContext;
  await set("b.com", "foo", 5, context);

  await getOKEx(
    "getByName",
    ["foo", context],
    [
      { domain: "a.com", name: "foo", value: 1 },
      { domain: null, name: "foo", value: 4 },
      { domain: "b.com", name: "foo", value: 5 },
    ]
  );
  await reset();
});

add_task(async function setSetsCurrentDate() {
  // Because Date.now() is not guaranteed to be monotonically increasing
  // we just do here rough sanity check with one minute tolerance.
  const MINUTE = 60 * 1000;
  let now = Date.now();
  let start = now - MINUTE;
  let end = now + MINUTE;
  await set("a.com", "foo", 1);
  let timestamp = await getDate("a.com", "foo");
  Assert.lessOrEqual(
    start,
    timestamp,
    "Timestamp is not too early (" + start + "<=" + timestamp + ")."
  );
  Assert.lessOrEqual(
    timestamp,
    end,
    "Timestamp is not too late (" + timestamp + "<=" + end + ")."
  );
  await reset();
});

add_task(async function maxLength() {
  const MAX_LENGTH = Ci.nsIContentPrefService2.GROUP_NAME_MAX_LENGTH;

  let count = Math.ceil(MAX_LENGTH / "subdomain.".length);
  const HTTP_URL = `http://${"subdomain.".repeat(count)}.com`;
  await set(HTTP_URL, "foo", 1);
  await getOK(
    [HTTP_URL, "foo"],
    1,
    HTTP_URL.substr("http://".length, MAX_LENGTH - 1)
  );
});

add_task(async function coalesceDataURLs() {
  const DATA_URL1 = "data:text/plain,Hello";
  const DATA_URL2 = "data:text/plain,Goodbye";
  const DATA_URL3 = "data:,Another test";
  const DATA_URL4 = "data:text/html,Another test";
  await set(DATA_URL4, "foo", 4);
  await set(DATA_URL1, "foo", 1);
  await getOK([DATA_URL1, "foo"], 1, "data:text/plain");
  await getOK([DATA_URL2, "foo"], 1, "data:text/plain");
  await getOK([DATA_URL3, "foo"], 1, "data:text/plain");
  await getOK([DATA_URL4, "foo"], 4, "data:text/html");
});