extensions/cookie/test/file_testcommon.js

   1 SimpleTest.waitForExplicitFinish();
   2
   3 var gPopup = null;
   4
   5 var gExpectedCookies = 0;
   6 var gExpectedLoads = 0;
   7 var gLoads = 0;
   8
   9 function setupTest(uri, cookies, loads) {
  10   netscape.security.PrivilegeManager.enablePrivilege("UniversalXPConnect");
  11
  12   var prefs = Components.classes["@mozilla.org/preferences-service;1"]
  13                         .getService(Components.interfaces.nsIPrefBranch);
  14   prefs.setIntPref("network.cookie.cookieBehavior", 1);
  15
  16   var cs = Components.classes["@mozilla.org/cookiemanager;1"]
  17                       .getService(Components.interfaces.nsICookieManager2);
  18   cs.removeAll();
  19
  20   gExpectedCookies = cookies;
  21   gExpectedLoads = loads;
  22
  23   // load a window which contains an iframe; each will attempt to set
  24   // cookies from their respective domains.
  25   gPopup = window.open(uri, 'hai', 'width=100,height=100');
  26 }
  27
  28 /** Receives MessageEvents to this window. */
  29 function messageReceiver(evt)
  30 {
  31   ok(evt instanceof MessageEvent, "wrong event type");
  32
  33   if (evt.data == "message")
  34     gLoads++;
  35   else {
  36     ok(false, "wrong message");
  37     gPopup.close();
  38     SimpleTest.finish();
  39   }
  40
  41   // only run the test when all our children are done loading & setting cookies
  42   if (gLoads == gExpectedLoads)
  43     runTest();
  44 }
  45
  46 function runTest() {
  47   // set a cookie from a domain of "localhost"
  48   document.cookie = "oh=hai";
  49
  50   netscape.security.PrivilegeManager.enablePrivilege("UniversalXPConnect");
  51   var cs = Components.classes["@mozilla.org/cookiemanager;1"]
  52                      .getService(Components.interfaces.nsICookieManager);
  53   var list = cs.enumerator;
  54   var count = 0;
  55   while (list.hasMoreElements()) {
  56     count++;
  57     list.getNext();
  58   }
  59   is(count, gExpectedCookies, "incorrect number of cookies");
  60
  61   gPopup.close();
  62   cs.removeAll();
  63   SimpleTest.finish();
  64 }
  65
  66 window.addEventListener("message", messageReceiver, false);
  67