summaryrefslogtreecommitdiffstats
path: root/extensions/cookie/test/file_testloadflags.js
blob: 9cf4a15a6bded8dd18b2d0037ecad4237c2bf002 (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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
const SCRIPT_URL = SimpleTest.getTestFileURL('file_testloadflags_chromescript.js');

var gExpectedCookies;
var gExpectedHeaders;
var gExpectedLoads;

var gObs;
var gPopup;

var gHeaders = 0;
var gLoads = 0;

// setupTest() is run from 'onload='.
function setupTest(uri, domain, cookies, loads, headers) {
  info("setupTest uri: " + uri + " domain: " + domain + " cookies: " + cookies +
       " loads: " + loads + " headers: " + headers);

  SimpleTest.waitForExplicitFinish();

  var prefSet = new Promise(resolve => {
    SpecialPowers.pushPrefEnv({ set: [["network.cookie.cookieBehavior", 1]] }, resolve);
  });

  gExpectedCookies = cookies;
  gExpectedLoads = loads;
  gExpectedHeaders = headers;

  gScript = SpecialPowers.loadChromeScript(SCRIPT_URL);
  gScript.addMessageListener("info", ({ str }) => info(str));
  gScript.addMessageListener("ok", ({ c, m }) => ok(c, m));
  gScript.addMessageListener("observer:gotCookie", ({ cookie, uri }) => {
    isnot(cookie.indexOf("oh=hai"), -1,
          "cookie 'oh=hai' is in header for " + uri);
    ++gHeaders;
  });

  var scriptReady = new Promise(resolve => {
    gScript.addMessageListener("init:return", resolve);
    gScript.sendAsyncMessage("init", { domain });
  });

  // Listen for MessageEvents.
  window.addEventListener("message", messageReceiver, false);

  Promise.all([ prefSet, scriptReady ]).then(() => {
    // load a window which contains an iframe; each will attempt to set
    // cookies from their respective domains.
    gPopup = window.open(uri, 'hai', 'width=100,height=100');
  });
}

function finishTest()
{
  gScript.addMessageListener("shutdown:return", () => {
    gScript.destroy();
    SimpleTest.finish();
  });
  gScript.sendAsyncMessage("shutdown");
}

/** Receives MessageEvents to this window. */
// Count and check loads.
function messageReceiver(evt)
{
  ok(evt.data == "f_lf_i msg data img" || evt.data == "f_lf_i msg data page",
     "message data received from popup");
  if (evt.data == "f_lf_i msg data img") {
    info("message data received from popup for image");
  }
  if (evt.data == "f_lf_i msg data page") {
    info("message data received from popup for page");
  }
  if (evt.data != "f_lf_i msg data img" && evt.data != "f_lf_i msg data page") {
    info("got this message but don't know what it is " + evt.data);
    gPopup.close();
    window.removeEventListener("message", messageReceiver, false);

    finishTest();
    return;
  }

  // only run the test when all our children are done loading & setting cookies
  if (++gLoads == gExpectedLoads) {
    gPopup.close();
    window.removeEventListener("message", messageReceiver, false);

    runTest();
  }
}

// runTest() is run by messageReceiver().
// Check headers, and count and check cookies.
function runTest() {
  // set a cookie from a domain of "localhost"
  document.cookie = "o=noes";

  is(gHeaders, gExpectedHeaders, "number of observed request headers");
  gScript.addMessageListener("getCookieCount:return", ({ count }) => {
    is(count, gExpectedCookies, "total number of cookies");
    finishTest();
  });

  gScript.sendAsyncMessage("getCookieCount");
}