netwerk/test/unit/test_bug669001.js

   1 "use strict";
   2
   3 const { HttpServer } = ChromeUtils.importESModule(
   4   "resource://testing-common/httpd.sys.mjs"
   5 );
   6
   7 var httpServer = null;
   8 var path = "/bug699001";
   9
  10 ChromeUtils.defineLazyGetter(this, "URI", function () {
  11   return "http://localhost:" + httpServer.identity.primaryPort + path;
  12 });
  13
  14 function make_channel(url) {
  15   return NetUtil.newChannel({ uri: url, loadUsingSystemPrincipal: true });
  16 }
  17
  18 var fetched;
  19
  20 // The test loads a resource that expires in one year, has an etag and varies only by User-Agent
  21 // First we load it, then check we load it only from the cache w/o even checking with the server
  22 // Then we modify our User-Agent and try it again
  23 // We have to get a new content (even though with the same etag) and again on next load only from
  24 // cache w/o accessing the server
  25 // Goal is to check we've updated User-Agent request header in cache after we've got 304 response
  26 // from the server
  27
  28 var tests = [
  29   {
  30     prepare() {},
  31     test() {
  32       Assert.ok(fetched);
  33     },
  34   },
  35   {
  36     prepare() {},
  37     test() {
  38       Assert.ok(!fetched);
  39     },
  40   },
  41   {
  42     prepare() {
  43       setUA("A different User Agent");
  44     },
  45     test() {
  46       Assert.ok(fetched);
  47     },
  48   },
  49   {
  50     prepare() {},
  51     test() {
  52       Assert.ok(!fetched);
  53     },
  54   },
  55   {
  56     prepare() {
  57       setUA("And another User Agent");
  58     },
  59     test() {
  60       Assert.ok(fetched);
  61     },
  62   },
  63   {
  64     prepare() {},
  65     test() {
  66       Assert.ok(!fetched);
  67     },
  68   },
  69 ];
  70
  71 function handler(metadata, response) {
  72   if (metadata.hasHeader("If-None-Match")) {
  73     response.setStatusLine(metadata.httpVersion, 304, "Not modified");
  74   } else {
  75     response.setStatusLine(metadata.httpVersion, 200, "OK");
  76     response.setHeader("Content-Type", "text/plain");
  77
  78     var body = "body";
  79     response.bodyOutputStream.write(body, body.length);
  80   }
  81
  82   fetched = true;
  83
  84   response.setHeader("Expires", getDateString(+1));
  85   response.setHeader("Cache-Control", "private");
  86   response.setHeader("Vary", "User-Agent");
  87   response.setHeader("ETag", "1234");
  88 }
  89
  90 function run_test() {
  91   httpServer = new HttpServer();
  92   httpServer.registerPathHandler(path, handler);
  93   httpServer.start(-1);
  94
  95   do_test_pending();
  96
  97   nextTest();
  98 }
  99
 100 function nextTest() {
 101   fetched = false;
 102   tests[0].prepare();
 103
 104   dump("Testing with User-Agent: " + getUA() + "\n");
 105   var chan = make_channel(URI);
 106
 107   // Give the old channel a chance to close the cache entry first.
 108   // XXX This is actually a race condition that might be considered a bug...
 109   executeSoon(function () {
 110     chan.asyncOpen(new ChannelListener(checkAndShiftTest, null));
 111   });
 112 }
 113
 114 function checkAndShiftTest(request, response) {
 115   tests[0].test(response);
 116
 117   tests.shift();
 118   if (!tests.length) {
 119     httpServer.stop(tearDown);
 120     return;
 121   }
 122
 123   nextTest();
 124 }
 125
 126 function tearDown() {
 127   setUA("");
 128   do_test_finished();
 129 }
 130
 131 // Helpers
 132
 133 function getUA() {
 134   var httphandler = Cc["@mozilla.org/network/protocol;1?name=http"].getService(
 135     Ci.nsIHttpProtocolHandler
 136   );
 137   return httphandler.userAgent;
 138 }
 139
 140 function setUA(value) {
 141   Services.prefs.setCharPref("general.useragent.override", value);
 142 }
 143
 144 function getDateString(yearDelta) {
 145   var months = [
 146     "Jan",
 147     "Feb",
 148     "Mar",
 149     "Apr",
 150     "May",
 151     "Jun",
 152     "Jul",
 153     "Aug",
 154     "Sep",
 155     "Oct",
 156     "Nov",
 157     "Dec",
 158   ];
 159   var days = ["Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"];
 160
 161   var d = new Date();
 162   return (
 163     days[d.getUTCDay()] +
 164     ", " +
 165     d.getUTCDate() +
 166     " " +
 167     months[d.getUTCMonth()] +
 168     " " +
 169     (d.getUTCFullYear() + yearDelta) +
 170     " " +
 171     d.getUTCHours() +
 172     ":" +
 173     d.getUTCMinutes() +
 174     ":" +
 175     d.getUTCSeconds() +
 176     " UTC"
 177   );
 178 }