1 files changed, 63 insertions, 0 deletions
diff --git a/devtools/client/netmonitor/test/browser_net_ws-limit-frames.js b/devtools/client/netmonitor/test/browser_net_ws-limit-frames.js
new file mode 100644
index 0000000000..1c9ee009cd
--- /dev/null
+++ b/devtools/client/netmonitor/test/browser_net_ws-limit-frames.js
@@ -0,0 +1,63 @@
+/* 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 message notification displays correctly.
+ */
+
+add_task(async function () {
+  // Set WS messages limit to a lower value for testing
+  await pushPref("devtools.netmonitor.msg.displayed-messages.limit", 30);
+
+  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(20);
+  });
+  await onNetworkEvents;
+
+  const requests = document.querySelectorAll(".request-list-item");
+  is(requests.length, 1, "There should be one request");
+
+  // Wait for truncated message notification to appear
+  const wait = waitForDOM(document, "#messages-view .truncated-message");
+
+  // 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, 30, "There should be thirty frames");
+  is(
+    document.querySelectorAll("#messages-view .truncated-message").length,
+    1,
+    "Truncated message notification is shown"
+  );
+
+  // Close WS connection
+  await SpecialPowers.spawn(tab.linkedBrowser, [], async () => {
+    await content.wrappedJSObject.closeConnection();
+  });
+
+  await teardown(monitor);
+});