test: fix test

Author: bjohansebas

test/e2e/client.test.js

@@ -4,6 +4,7 @@ const webpack = require("webpack");
 const Server = require("../../lib/Server");
 const config = require("../fixtures/simple-config-other/webpack.config");
 const runBrowser = require("../helpers/run-browser");
+const sessionSubscribe = require("../helpers/session-subscribe");
 const port = require("../ports-map")["client-option"];
 
 describe("client option", () => {
@@ -119,14 +120,28 @@ describe("client option", () => {
           pageErrors.push(error);
         });
 
-      const response = await page.goto(
-        `http://localhost:${port}/foo/test/bar`,
-        {
-          waitUntil: "networkidle0",
-        },
-      );
+      const webSocketRequests = [];
 
-      expect(response.status()).toMatchSnapshot("response status");
+      const session = await page.target().createCDPSession();
+      session.on("Network.webSocketCreated", (test) => {
+        webSocketRequests.push(test);
+      });
+
+      await session.send("Target.setAutoAttach", {
+        autoAttach: true,
+        flatten: true,
+        waitForDebuggerOnStart: true,
+      });
+
+      sessionSubscribe(session);
+
+      await page.goto(`http://localhost:${port}/foo/test/bar`, {
+        waitUntil: "networkidle0",
+      });
+
+      const [webSocketRequest] = webSocketRequests;
+
+      expect(webSocketRequest.status()).toMatchSnapshot("response status");
 
       expect(consoleMessages.map((message) => message.text())).toMatchSnapshot(
         "console messages",