summaryrefslogtreecommitdiffstats
path: root/devtools/shared/security/tests/chrome/test_websocket-transport.html
blob: 8e4652fb1d50307a9acaf19e565dbb930a2728e4 (plain)
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
<!DOCTYPE HTML>
<html>
<head>
  <meta charset="utf-8">
  <title>Test the WebSocket debugger transport</title>
  <script src="chrome://mochikit/content/tests/SimpleTest/SimpleTest.js"></script>
  <link rel="stylesheet" type="text/css" href="chrome://mochikit/content/tests/SimpleTest/test.css">
</head>
<body>
<script>
"use strict";

window.onload = function() {
  const {require} = ChromeUtils.importESModule("resource://devtools/shared/loader/Loader.sys.mjs");
  // eslint-disable-next-line mozilla/reject-some-requires
  const {DevToolsClient} = require("devtools/client/devtools-client");
  const {DevToolsServer} = require("devtools/server/devtools-server");
  const { SocketListener } = require("devtools/shared/security/socket");

  Services.prefs.setBoolPref("devtools.debugger.remote-enabled", true);
  Services.prefs.setBoolPref("devtools.debugger.prompt-connection", false);

  SimpleTest.registerCleanupFunction(() => {
    Services.prefs.clearUserPref("devtools.debugger.remote-enabled");
    Services.prefs.clearUserPref("devtools.debugger.prompt-connection");
  });

  add_task(async function() {
    DevToolsServer.init();
    DevToolsServer.registerAllActors();

    is(DevToolsServer.listeningSockets, 0, "0 listening sockets");

    const socketOptions = {
      portOrPath: -1,
      webSocket: true,
    };
    const listener = new SocketListener(DevToolsServer, socketOptions);
    ok(listener, "Socket listener created");
    await listener.open();
    is(DevToolsServer.listeningSockets, 1, "1 listening socket");

    const transport = await DevToolsClient.socketConnect({
      host: "127.0.0.1",
      port: listener.port,
      webSocket: true,
    });
    ok(transport, "Client transport created");

    const client = new DevToolsClient(transport);
    const onUnexpectedClose = () => {
      ok(false, "Closed unexpectedly");
    };
    client.on("closed", onUnexpectedClose);

    await client.connect();

    // Send a message the server
    const reply = await client.mainRoot.getRoot();
    is(reply.from, "root", "Got expected response");

    client.off("closed", onUnexpectedClose);
    transport.close();
    listener.close();
    is(DevToolsServer.listeningSockets, 0, "0 listening sockets");

    DevToolsServer.destroy();
  });
};
</script>
</body>
</html>