1 files changed, 148 insertions, 0 deletions
diff --git a/devtools/client/webconsole/test/node/components/webconsole-wrapper.test.js b/devtools/client/webconsole/test/node/components/webconsole-wrapper.test.js
new file mode 100644
index 0000000000..1ea44b7388
--- /dev/null
+++ b/devtools/client/webconsole/test/node/components/webconsole-wrapper.test.js
@@ -0,0 +1,148 @@
+/* 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";
+
+// Test utils.
+const expect = require("expect");
+const {
+  stubPackets,
+} = require("resource://devtools/client/webconsole/test/node/fixtures/stubs/index.js");
+const {
+  clonePacket,
+  getMessageAt,
+  getPrivatePacket,
+  getWebConsoleUiMock,
+} = require("resource://devtools/client/webconsole/test/node/helpers.js");
+
+const WebConsoleWrapper = require("resource://devtools/client/webconsole/webconsole-wrapper.js");
+const {
+  messagesAdd,
+} = require("resource://devtools/client/webconsole/actions/messages.js");
+
+async function getWebConsoleWrapper() {
+  const hud = {
+    currentTarget: { client: {}, getFront: () => {} },
+    getMappedExpression: () => {},
+    commands: {
+      objectCommand: {
+        releaseObjects: async frontsToRelease => {},
+      },
+    },
+  };
+  const webConsoleUi = getWebConsoleUiMock(hud);
+
+  const wcow = new WebConsoleWrapper(null, webConsoleUi, null, null);
+  await wcow.init();
+  return wcow;
+}
+
+describe("WebConsoleWrapper", () => {
+  it("clears queues when dispatchMessagesClear is called", async () => {
+    const ncow = await getWebConsoleWrapper();
+    ncow.queuedMessageAdds.push({ fakePacket: "message", data: {} });
+    ncow.queuedMessageUpdates.push({ fakePacket: "message-update", data: {} });
+    ncow.queuedRequestUpdates.push({ fakePacket: "request-update", data: {} });
+
+    ncow.dispatchMessagesClear();
+
+    expect(ncow.queuedMessageAdds.length).toBe(0);
+    expect(ncow.queuedMessageUpdates.length).toBe(0);
+    expect(ncow.queuedRequestUpdates.length).toBe(0);
+  });
+
+  it("removes private packets from message queue on dispatchPrivateMessagesClear", async () => {
+    const ncow = await getWebConsoleWrapper();
+
+    const publicLog = stubPackets.get("console.log('mymap')");
+    ncow.queuedMessageAdds.push(
+      getPrivatePacket("console.trace()"),
+      publicLog,
+      getPrivatePacket("XHR POST request")
+    );
+
+    ncow.dispatchPrivateMessagesClear();
+    expect(ncow.queuedMessageAdds).toEqual([publicLog]);
+  });
+
+  it("removes private packets from network update queue on dispatchPrivateMessagesClear", async () => {
+    const ncow = await getWebConsoleWrapper();
+    const publicLog = stubPackets.get("console.log('mymap')");
+    ncow.queuedMessageAdds.push(
+      getPrivatePacket("console.trace()"),
+      publicLog,
+      getPrivatePacket("XHR POST request")
+    );
+
+    const postId = "pid1";
+    const getId = "gid1";
+
+    // Add messages in the store to make sure that update to private requests are
+    // removed from the queue.
+    ncow
+      .getStore()
+      .dispatch(
+        messagesAdd([
+          stubPackets.get("GET request"),
+          { ...getPrivatePacket("XHR GET request"), actor: getId },
+        ])
+      );
+
+    // Add packet to the message queue to make sure that update to private requests are
+    // removed from the queue.
+    ncow.queuedMessageAdds.push({
+      ...getPrivatePacket("XHR POST request"),
+      actor: postId,
+    });
+
+    const publicNetworkUpdate = stubPackets.get("GET request update");
+    ncow.queuedMessageUpdates.push(
+      publicNetworkUpdate,
+      {
+        ...getPrivatePacket("XHR GET request update"),
+        actor: getId,
+      },
+      {
+        ...getPrivatePacket("XHR POST request update"),
+        actor: postId,
+      }
+    );
+
+    ncow.dispatchPrivateMessagesClear();
+
+    expect(ncow.queuedMessageUpdates.length).toBe(1);
+    expect(ncow.queuedMessageUpdates).toEqual([publicNetworkUpdate]);
+  });
+
+  it("removes private packets from network request queue on dispatchPrivateMessagesClear", async () => {
+    const ncow = await getWebConsoleWrapper();
+
+    const packet1 = clonePacket(stubPackets.get("GET request"));
+    const packet2 = clonePacket(getPrivatePacket("XHR GET request"));
+    const packet3 = clonePacket(getPrivatePacket("XHR POST request"));
+
+    // We need to reassign the timeStamp of the packet to guarantee the order.
+    packet1.timeStamp = packet1.timeStamp + 1;
+    packet2.timeStamp = packet2.timeStamp + 2;
+    packet3.timeStamp = packet3.timeStamp + 3;
+
+    ncow.getStore().dispatch(messagesAdd([packet1, packet2, packet3]));
+
+    const state = ncow.getStore().getState();
+    const publicId = getMessageAt(state, 0).id;
+    const privateXhrGetId = getMessageAt(state, 1).id;
+    const privateXhrPostId = getMessageAt(state, 2).id;
+    ncow.queuedRequestUpdates.push(
+      { id: publicId },
+      { id: privateXhrGetId },
+      { id: privateXhrPostId }
+    );
+    // ncow.queuedRequestUpdates.push({fakePacket: "request-update"});
+
+    ncow.dispatchPrivateMessagesClear();
+
+    expect(ncow.queuedRequestUpdates.length).toBe(1);
+    expect(ncow.queuedRequestUpdates).toEqual([{ id: publicId }]);
+  });
+});