gecko-dev/netwerk/test/unit/test_localhost_offline.js

76 строки
2.1 KiB
JavaScript

"use strict";
const { HttpServer } = ChromeUtils.import("resource://testing-common/httpd.js");
var httpServer = null;
const body = "Hello";
function makeChan(url) {
let chan = NetUtil.newChannel({
uri: url,
loadUsingSystemPrincipal: true,
}).QueryInterface(Ci.nsIHttpChannel);
chan.loadFlags |= Ci.nsIRequest.LOAD_BYPASS_CACHE;
chan.loadFlags |= Ci.nsIRequest.INHIBIT_CACHING;
return chan;
}
function channelOpenPromise(chan, flags) {
return new Promise(resolve => {
function finish(req, buffer) {
resolve([req, buffer]);
}
chan.asyncOpen(new ChannelListener(finish, null, flags));
});
}
function makeURL(host) {
return `http://${host}:${httpServer.identity.primaryPort}/`;
}
add_task(async function test_localhost_offline() {
Services.io.offline = true;
Services.prefs.setBoolPref("network.disable-localhost-when-offline", false);
let chan = makeChan(makeURL("127.0.0.1"));
let [, resp] = await channelOpenPromise(chan);
Assert.equal(resp, body, "Should get correct response");
chan = makeChan(makeURL("localhost"));
[, resp] = await channelOpenPromise(chan);
Assert.equal(resp, body, "Should get response");
Services.prefs.setBoolPref("network.disable-localhost-when-offline", true);
chan = makeChan(makeURL("127.0.0.1"));
let [req] = await channelOpenPromise(
chan,
CL_ALLOW_UNKNOWN_CL | CL_EXPECT_FAILURE
);
req.QueryInterface(Ci.nsIHttpChannel);
Assert.equal(req.status, Cr.NS_ERROR_OFFLINE);
chan = makeChan(makeURL("localhost"));
[req] = await channelOpenPromise(
chan,
CL_ALLOW_UNKNOWN_CL | CL_EXPECT_FAILURE
);
req.QueryInterface(Ci.nsIHttpChannel);
Assert.equal(req.status, Cr.NS_ERROR_OFFLINE);
Services.prefs.clearUserPref("network.disable-localhost-when-offline");
Services.io.offline = false;
});
function run_test() {
httpServer = new HttpServer();
httpServer.registerPathHandler("/", (request, response) => {
response.seizePower();
response.write("HTTP/1.1 200 OK\r\n");
response.write("Content-Length: " + body.length + "\r\n");
response.write("\r\n");
response.write(body);
response.finish();
});
httpServer.start(-1);
run_next_test();
}