toolkit/mozapps/extensions/test/xpcshell/test_cookies.js

   1 "use strict";
   2
   3 let server = createHttpServer({ hosts: ["example.com"] });
   4
   5 createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "45", "45");
   6
   7 Services.prefs.setBoolPref("extensions.getAddons.cache.enabled", true);
   8 Services.prefs.setBoolPref(PREF_EM_CHECK_UPDATE_SECURITY, false);
   9
  10 // Tests that cookies are not sent with background requests.
  11 add_task(async function test_cookies() {
  12   const ID = "bg-cookies@tests.mozilla.org";
  13
  14   // Add a new handler to the test web server for the given file path.
  15   // The handler appends the incoming requests to `results` and replies
  16   // with the provided body.
  17   function makeHandler(path, results, body) {
  18     server.registerPathHandler(path, (request, response) => {
  19       results.push(request);
  20       response.write(body);
  21     });
  22   }
  23
  24   let gets = [];
  25   makeHandler("/get", gets, JSON.stringify({ results: [] }));
  26   Services.prefs.setCharPref(PREF_GETADDONS_BYIDS, "http://example.com/get");
  27
  28   let updates = [];
  29   makeHandler(
  30     "/update",
  31     updates,
  32     JSON.stringify({
  33       addons: {
  34         [ID]: {
  35           updates: [
  36             {
  37               version: "2.0",
  38               update_link: "http://example.com/update.xpi",
  39               applications: {
  40                 gecko: {},
  41               },
  42             },
  43           ],
  44         },
  45       },
  46     })
  47   );
  48
  49   let xpiFetches = [];
  50   makeHandler("/update.xpi", xpiFetches, "");
  51
  52   const COOKIE = "test";
  53   // cookies.add() takes a time in seconds
  54   let expiration = Date.now() / 1000 + 60 * 60;
  55   Services.cookies.add(
  56     "example.com",
  57     "/",
  58     COOKIE,
  59     "testing",
  60     false,
  61     false,
  62     false,
  63     expiration,
  64     {},
  65     Ci.nsICookie.SAMESITE_NONE,
  66     Ci.nsICookie.SCHEME_HTTP
  67   );
  68
  69   await promiseStartupManager();
  70
  71   let addon = await promiseInstallWebExtension({
  72     manifest: {
  73       version: "1.0",
  74       browser_specific_settings: {
  75         gecko: {
  76           id: ID,
  77           update_url: "http://example.com/update",
  78         },
  79       },
  80     },
  81   });
  82
  83   equal(gets.length, 1, "Saw one addon metadata request");
  84   equal(gets[0].hasHeader("Cookie"), false, "Metadata request has no cookies");
  85
  86   await Promise.all([
  87     AddonTestUtils.promiseInstallEvent("onDownloadFailed"),
  88     AddonManagerPrivate.backgroundUpdateCheck(),
  89   ]);
  90
  91   equal(updates.length, 1, "Saw one update check request");
  92   equal(updates[0].hasHeader("Cookie"), false, "Update request has no cookies");
  93
  94   equal(xpiFetches.length, 1, "Saw one request for updated xpi");
  95   equal(
  96     xpiFetches[0].hasHeader("Cookie"),
  97     false,
  98     "Request for updated XPI has no cookies"
  99   );
 100
 101   await addon.uninstall();
 102 });