1 files changed, 80 insertions, 0 deletions
diff --git a/devtools/client/netmonitor/test/browser_net_ws-limit-payload.js b/devtools/client/netmonitor/test/browser_net_ws-limit-payload.js
new file mode 100644
index 0000000000..8a3082810a
--- /dev/null
+++ b/devtools/client/netmonitor/test/browser_net_ws-limit-payload.js
@@ -0,0 +1,80 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+/**
+ * Test that WS connection is established successfully and the truncated payload is correct.
+ */
+
+add_task(async function () {
+  // Set WS message payload limit to a lower value for testing
+  await pushPref("devtools.netmonitor.msg.messageDataLimit", 100);
+
+  const { tab, monitor } = await initNetMonitor(WS_PAGE_URL, {
+    requestCount: 1,
+  });
+  info("Starting test... ");
+
+  const { document, store, windowRequire } = monitor.panelWin;
+  const Actions = windowRequire("devtools/client/netmonitor/src/actions/index");
+
+  store.dispatch(Actions.batchEnable(false));
+
+  // Wait for WS connections to be established + send messages
+  const onNetworkEvents = waitForNetworkEvents(monitor, 1);
+  await SpecialPowers.spawn(tab.linkedBrowser, [], async () => {
+    await content.wrappedJSObject.openConnection(0);
+    content.wrappedJSObject.sendData(new Array(10 * 11).toString()); // > 100B payload
+  });
+  await onNetworkEvents;
+
+  const requests = document.querySelectorAll(".request-list-item");
+  is(requests.length, 1, "There should be one request");
+
+  // Wait for all sent/received messages to be displayed in DevTools
+  const wait = waitForDOM(
+    document,
+    "#messages-view .message-list-table .message-list-item",
+    2
+  );
+
+  // Select the first request
+  EventUtils.sendMouseEvent({ type: "mousedown" }, requests[0]);
+
+  // Click on the "Response" panel
+  clickOnSidebarTab(document, "response");
+  await wait;
+
+  // Get all messages present in the "Response" panel
+  const frames = document.querySelectorAll(
+    "#messages-view .message-list-table .message-list-item"
+  );
+
+  // Check expected results
+  is(frames.length, 2, "There should be two frames");
+
+  // Wait for next tick to do async stuff (The MessagePayload component uses the async function getMessagePayload)
+  await waitForTick();
+  EventUtils.sendMouseEvent({ type: "mousedown" }, frames[0]);
+
+  await waitForDOM(document, "#messages-view .truncated-data-message");
+
+  ok(
+    document.querySelector("#messages-view .truncated-data-message"),
+    "Truncated data header shown"
+  );
+  is(
+    document.querySelector("#messages-view .message-rawData-payload")
+      .textContent.length,
+    100,
+    "Payload size is kept to the limit"
+  );
+
+  // Close WS connection
+  await SpecialPowers.spawn(tab.linkedBrowser, [], async () => {
+    await content.wrappedJSObject.closeConnection();
+  });
+
+  await teardown(monitor);
+});