const POLARIS_ENABLED = "browser.polaris.enabled";
const PREF_DNT = "privacy.donottrackheader.enabled";
const PREF_DNTV = "privacy.donottrackheader.value";
const PREF_TP = "privacy.trackingprotection.enabled";
const PREF_TPUI = "privacy.trackingprotection.ui.enabled";

let prefs = [PREF_DNT, PREF_TP, PREF_TPUI];

function spinEventLoop() {
  return new Promise((resolve) => executeSoon(resolve));
};

// Spin event loop before checking so that polaris pref observer can set
// dependent prefs.
function* assertPref(pref, enabled) {
  yield spinEventLoop();
  let prefEnabled = Services.prefs.getBoolPref(pref);
  Assert.equal(prefEnabled, enabled, "Checking state of pref " + pref + ".");
};

function* testPrefs(test) {
  for (let pref of prefs) {
    yield test(pref);
  }
}

add_task(function* test_default_values() {
  Assert.ok(!Services.prefs.getBoolPref(POLARIS_ENABLED), POLARIS_ENABLED + " is disabled by default.");
  Assert.ok(!Services.prefs.getBoolPref(PREF_TPUI), PREF_TPUI + "is disabled by default.");
});

add_task(function* test_changing_pref_changes_tracking() {
  function* testPref(pref) {
    Services.prefs.setBoolPref(POLARIS_ENABLED, true);
    yield assertPref(pref, true);
    Services.prefs.setBoolPref(POLARIS_ENABLED, false);
    yield assertPref(pref, false);
    Services.prefs.setBoolPref(POLARIS_ENABLED, true);
    yield assertPref(pref, true);
  }
  yield testPrefs(testPref);
});

add_task(function* test_prefs_can_be_changed_individually() {
  function* testPref(pref) {
    Services.prefs.setBoolPref(POLARIS_ENABLED, true);
    yield assertPref(pref, true);
    Services.prefs.setBoolPref(pref, false);
    yield assertPref(pref, false);
    yield assertPref(POLARIS_ENABLED, true);
    Services.prefs.setBoolPref(POLARIS_ENABLED, false);
    yield assertPref(pref, false);
    Services.prefs.setBoolPref(pref, true);
    yield assertPref(pref, true);
    yield assertPref(POLARIS_ENABLED, false);
  }
  yield testPrefs(testPref);
});

add_task(function* test_donottrackheader_having_0_value() {
  Services.prefs.setIntPref(PREF_DNTV, 0);
  Services.prefs.setBoolPref(POLARIS_ENABLED, true);
  yield spinEventLoop();
  Assert.ok(Services.prefs.getIntPref(PREF_DNTV, true), 1, PREF_DNTV + " reset after setting polaris.");
});