diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /netwerk/test/unit/test_websocket_500k.js | |
parent | Initial commit. (diff) | |
download | firefox-esr-upstream.tar.xz firefox-esr-upstream.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | netwerk/test/unit/test_websocket_500k.js | 222 |
1 files changed, 222 insertions, 0 deletions
diff --git a/netwerk/test/unit/test_websocket_500k.js b/netwerk/test/unit/test_websocket_500k.js new file mode 100644 index 0000000000..f974283437 --- /dev/null +++ b/netwerk/test/unit/test_websocket_500k.js @@ -0,0 +1,222 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +"use strict"; + +/* import-globals-from head_cache.js */ +/* import-globals-from head_cookies.js */ +/* import-globals-from head_channels.js */ +/* import-globals-from head_servers.js */ + +XPCOMUtils.defineLazyModuleGetters(this, { + ObjectUtils: "resource://gre/modules/ObjectUtils.jsm", +}); + +add_setup(async function () { + Services.prefs.setCharPref("network.dns.localDomains", "foo.example.com"); + + let certdb = Cc["@mozilla.org/security/x509certdb;1"].getService( + Ci.nsIX509CertDB + ); + addCertFromFile(certdb, "http2-ca.pem", "CTu,u,u"); + addCertFromFile(certdb, "proxy-ca.pem", "CTu,u,u"); +}); + +registerCleanupFunction(async () => { + Services.prefs.clearUserPref("network.dns.localDomains"); +}); + +async function channelOpenPromise(url, msg) { + let conn = new WebSocketConnection(); + await conn.open(url); + conn.send(msg); + let res = await conn.receiveMessages(); + conn.close(); + let { status } = await conn.finished(); + return [status, res]; +} + +async function sendDataAndCheck(url) { + let data = "a".repeat(500000); + let [status, res] = await channelOpenPromise(url, data); + Assert.equal(status, Cr.NS_OK); + // Use "ObjectUtils.deepEqual" directly to avoid printing data. + Assert.ok(ObjectUtils.deepEqual(res, [data])); +} + +add_task(async function test_h2_websocket_500k() { + Services.prefs.setBoolPref("network.http.http2.websockets", true); + let wss = new NodeWebSocketHttp2Server(); + await wss.start(); + registerCleanupFunction(async () => wss.stop()); + + Assert.notEqual(wss.port(), null); + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + let url = `wss://foo.example.com:${wss.port()}`; + await sendDataAndCheck(url); +}); + +// h1.1 direct +add_task(async function test_h1_websocket_direct() { + let wss = new NodeWebSocketServer(); + await wss.start(); + registerCleanupFunction(async () => wss.stop()); + Assert.notEqual(wss.port(), null); + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + let url = `wss://localhost:${wss.port()}`; + await sendDataAndCheck(url); +}); + +// ws h1.1 with insecure h1.1 proxy +add_task(async function test_h1_ws_with_h1_insecure_proxy() { + Services.prefs.setBoolPref("network.http.http2.websockets", false); + let proxy = new NodeHTTPProxyServer(); + await proxy.start(); + + let wss = new NodeWebSocketServer(); + await wss.start(); + + registerCleanupFunction(async () => { + await wss.stop(); + await proxy.stop(); + }); + + Assert.notEqual(wss.port(), null); + + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + let url = `wss://localhost:${wss.port()}`; + await sendDataAndCheck(url); +}); + +// h1 server with secure h1.1 proxy +add_task(async function test_h1_ws_with_secure_h1_proxy() { + let proxy = new NodeHTTPSProxyServer(); + await proxy.start(); + + let wss = new NodeWebSocketServer(); + await wss.start(); + registerCleanupFunction(async () => { + await wss.stop(); + await proxy.stop(); + }); + + Assert.notEqual(wss.port(), null); + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + + let url = `wss://localhost:${wss.port()}`; + await sendDataAndCheck(url); + + await proxy.stop(); +}); + +// ws h1.1 with h2 proxy +add_task(async function test_h1_ws_with_h2_proxy() { + Services.prefs.setBoolPref("network.http.http2.websockets", false); + + let proxy = new NodeHTTP2ProxyServer(); + await proxy.start(); + + let wss = new NodeWebSocketServer(); + await wss.start(); + + registerCleanupFunction(async () => { + await wss.stop(); + await proxy.stop(); + }); + + Assert.notEqual(wss.port(), null); + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + + let url = `wss://localhost:${wss.port()}`; + await sendDataAndCheck(url); + + await proxy.stop(); +}); + +// ws h2 with insecure h1.1 proxy +add_task(async function test_h2_ws_with_h1_insecure_proxy() { + Services.prefs.setBoolPref("network.http.http2.websockets", true); + + let proxy = new NodeHTTPProxyServer(); + await proxy.start(); + + let wss = new NodeWebSocketHttp2Server(); + await wss.start(); + + registerCleanupFunction(async () => { + await wss.stop(); + await proxy.stop(); + }); + + Assert.notEqual(wss.port(), null); + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + + let url = `wss://localhost:${wss.port()}`; + await sendDataAndCheck(url); + + await proxy.stop(); +}); + +add_task(async function test_h2_ws_with_h1_secure_proxy() { + Services.prefs.setBoolPref("network.http.http2.websockets", true); + + let proxy = new NodeHTTPSProxyServer(); + await proxy.start(); + + let wss = new NodeWebSocketHttp2Server(); + await wss.start(); + + registerCleanupFunction(async () => { + await wss.stop(); + await proxy.stop(); + }); + + Assert.notEqual(wss.port(), null); + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + + let url = `wss://localhost:${wss.port()}`; + await sendDataAndCheck(url); + + await proxy.stop(); +}); + +// ws h2 with secure h2 proxy +add_task(async function test_h2_ws_with_h2_proxy() { + Services.prefs.setBoolPref("network.http.http2.websockets", true); + + let proxy = new NodeHTTP2ProxyServer(); + await proxy.start(); // start and register proxy "filter" + + let wss = new NodeWebSocketHttp2Server(); + await wss.start(); // init port + + registerCleanupFunction(async () => { + await wss.stop(); + await proxy.stop(); + }); + + Assert.notEqual(wss.port(), null); + await wss.registerMessageHandler((data, ws) => { + ws.send(data); + }); + + let url = `wss://localhost:${wss.port()}`; + await sendDataAndCheck(url); + + await proxy.stop(); +}); |