summaryrefslogtreecommitdiffstats
path: root/browser/components/urlbar/tests/browser/browser_observers_for_strip_on_share.js
blob: 02b404926b7a411ff33d10130fa255360747a3b0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
/* Any copyright is dedicated to the Public Domain.
http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

let listService;

const QPS_PREF = "privacy.query_stripping.enabled";
const STRIP_ON_SHARE_PREF = "privacy.query_stripping.strip_on_share.enabled";

// Tests for the observers for both QPS and Strip on Share
add_setup(async function () {
  // Get the list service so we can wait for it to be fully initialized before running tests.
  listService = Cc["@mozilla.org/query-stripping-list-service;1"].getService(
    Ci.nsIURLQueryStrippingListService
  );

  await listService.testWaitForInit();
});

// Test if Strip on share observers are registered/unregistered depending if the
// Strip on Share Pref is enabled/disabled regardless of the state of QPS Pref
add_task(
  async function checkStripOnShareObserversForVaryingStatesOfQPSAndStripOnShare() {
    for (let queryStrippingEnabled of [false, true]) {
      for (let stripOnShareEnabled of [false, true]) {
        await SpecialPowers.pushPrefEnv({
          set: [
            [QPS_PREF, queryStrippingEnabled],
            [STRIP_ON_SHARE_PREF, stripOnShareEnabled],
          ],
        });

        let areObserservesRegistered;
        await BrowserTestUtils.waitForCondition(function () {
          areObserservesRegistered = listService.testHasStripOnShareObservers();
          return areObserservesRegistered == stripOnShareEnabled;
        }, "waiting for init of URLQueryStrippingListService ensuring observers have time to register if they need");

        if (!stripOnShareEnabled) {
          Assert.ok(!areObserservesRegistered, "Observers are unregistered");
        } else {
          Assert.ok(areObserservesRegistered, "Observers are registered");
        }

        await SpecialPowers.popPrefEnv();
      }
    }
  }
);

// Test if QPS observers are registered/unregistered depending if the QPS
// Pref is enabled/disabled regardless of the state of Strip on Share Pref
add_task(
  async function checkQPSObserversForVaryingStatesOfQPSAndStripOnShare() {
    for (let queryStrippingEnabled of [false, true]) {
      for (let stripOnShareEnabled of [false, true]) {
        await SpecialPowers.pushPrefEnv({
          set: [
            [QPS_PREF, queryStrippingEnabled],
            [STRIP_ON_SHARE_PREF, stripOnShareEnabled],
          ],
        });

        let areObserservesRegistered;
        await BrowserTestUtils.waitForCondition(function () {
          areObserservesRegistered = listService.testHasQPSObservers();
          return areObserservesRegistered == queryStrippingEnabled;
        }, "waiting for init of URLQueryStrippingListService ensuring observers have time to register if they need");

        if (!queryStrippingEnabled) {
          Assert.ok(!areObserservesRegistered, "Observers are unregistered");
        } else {
          Assert.ok(areObserservesRegistered, "Observers are registered");
        }

        await SpecialPowers.popPrefEnv();
      }
    }
  }
);