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
105
106
107
108
109
|
"use strict";
/**
* This duplicates the test from netwerk/test/unit/test_captive_portal_service.js
* however using an extension to gather the captive portal information.
*/
const PREF_CAPTIVE_ENABLED = "network.captive-portal-service.enabled";
const PREF_CAPTIVE_TESTMODE = "network.captive-portal-service.testMode";
const PREF_CAPTIVE_MINTIME = "network.captive-portal-service.minInterval";
const PREF_CAPTIVE_ENDPOINT = "captivedetect.canonicalURL";
const PREF_DNS_NATIVE_IS_LOCALHOST = "network.dns.native-is-localhost";
const SUCCESS_STRING = "success\n";
let cpResponse = SUCCESS_STRING;
const httpserver = createHttpServer();
httpserver.registerPathHandler("/captive.txt", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
response.setHeader("Content-Type", "text/plain");
response.write(cpResponse);
});
registerCleanupFunction(() => {
Services.prefs.clearUserPref(PREF_CAPTIVE_ENABLED);
Services.prefs.clearUserPref(PREF_CAPTIVE_TESTMODE);
Services.prefs.clearUserPref(PREF_CAPTIVE_ENDPOINT);
Services.prefs.clearUserPref(PREF_CAPTIVE_MINTIME);
Services.prefs.clearUserPref(PREF_DNS_NATIVE_IS_LOCALHOST);
});
add_task(function setup() {
Services.prefs.setCharPref(
PREF_CAPTIVE_ENDPOINT,
`http://localhost:${httpserver.identity.primaryPort}/captive.txt`
);
Services.prefs.setBoolPref(PREF_CAPTIVE_TESTMODE, true);
Services.prefs.setIntPref(PREF_CAPTIVE_MINTIME, 0);
Services.prefs.setBoolPref(PREF_DNS_NATIVE_IS_LOCALHOST, true);
});
add_task(async function test_captivePortal_basic() {
let cps = Cc["@mozilla.org/network/captive-portal-service;1"].getService(
Ci.nsICaptivePortalService
);
let extension = ExtensionTestUtils.loadExtension({
manifest: {
permissions: ["captivePortal"],
},
isPrivileged: true,
async background() {
browser.captivePortal.onConnectivityAvailable.addListener(details => {
browser.test.log(
`onConnectivityAvailable received ${JSON.stringify(details)}`
);
browser.test.sendMessage("connectivity", details);
});
browser.captivePortal.onStateChanged.addListener(details => {
browser.test.log(`onStateChanged received ${JSON.stringify(details)}`);
browser.test.sendMessage("state", details);
});
browser.test.onMessage.addListener(async msg => {
if (msg == "getstate") {
browser.test.sendMessage(
"getstate",
await browser.captivePortal.getState()
);
}
});
browser.test.assertEq(
"unknown",
await browser.captivePortal.getState(),
"initial state unknown"
);
},
});
await extension.startup();
// The captive portal service is started by nsIOService when the pref becomes true, so we
// toggle the pref. We cannot set to false before the extension loads above.
Services.prefs.setBoolPref(PREF_CAPTIVE_ENABLED, false);
Services.prefs.setBoolPref(PREF_CAPTIVE_ENABLED, true);
let details = await extension.awaitMessage("connectivity");
equal(details.status, "clear", "initial connectivity");
extension.sendMessage("getstate");
details = await extension.awaitMessage("getstate");
equal(details, "not_captive", "initial state");
info("REFRESH to other");
cpResponse = "other";
cps.recheckCaptivePortal();
details = await extension.awaitMessage("state");
equal(details.state, "locked_portal", "state in portal");
info("REFRESH to success");
cpResponse = SUCCESS_STRING;
cps.recheckCaptivePortal();
details = await extension.awaitMessage("connectivity");
equal(details.status, "captive", "final connectivity");
details = await extension.awaitMessage("state");
equal(details.state, "unlocked_portal", "state after unlocking portal");
await extension.unload();
});
|
