1 files changed, 78 insertions, 0 deletions
diff --git a/devtools/client/performance/test/xpcshell/test_waterfall-utils-collapse-03.js b/devtools/client/performance/test/xpcshell/test_waterfall-utils-collapse-03.js
new file mode 100644
index 0000000000..b9124c6dfd
--- /dev/null
+++ b/devtools/client/performance/test/xpcshell/test_waterfall-utils-collapse-03.js
@@ -0,0 +1,78 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+"use strict";
+
+/**
+ * Tests that the waterfall collapsing works when atleast two
+ * collapsible markers downward, and the following marker is outside of both ranges.
+ */
+
+add_task(function test() {
+  const WaterfallUtils = require("devtools/client/performance/modules/logic/waterfall-utils");
+
+  const rootMarkerNode = WaterfallUtils.createParentNode({ name: "(root)" });
+
+  WaterfallUtils.collapseMarkersIntoNode({
+    rootNode: rootMarkerNode,
+    markersList: gTestMarkers,
+  });
+
+  function compare(marker, expected) {
+    for (const prop in expected) {
+      if (prop === "submarkers") {
+        for (let i = 0; i < expected.submarkers.length; i++) {
+          compare(marker.submarkers[i], expected.submarkers[i]);
+        }
+      } else if (prop !== "uid") {
+        equal(marker[prop], expected[prop], `${expected.name} matches ${prop}`);
+      }
+    }
+  }
+
+  compare(rootMarkerNode, gExpectedOutput);
+});
+
+const gTestMarkers = [
+  { start: 2, end: 10, name: "DOMEvent" },
+  { start: 3, end: 9, name: "Javascript" },
+  { start: 4, end: 8, name: "GarbageCollection" },
+  { start: 11, end: 12, name: "Styles" },
+  { start: 13, end: 14, name: "Styles" },
+  { start: 15, end: 25, name: "DOMEvent" },
+  { start: 17, end: 24, name: "Javascript" },
+  { start: 18, end: 19, name: "GarbageCollection" },
+];
+
+const gExpectedOutput = {
+  name: "(root)",
+  submarkers: [
+    {
+      start: 2,
+      end: 10,
+      name: "DOMEvent",
+      submarkers: [
+        {
+          start: 3,
+          end: 9,
+          name: "Javascript",
+          submarkers: [{ start: 4, end: 8, name: "GarbageCollection" }],
+        },
+      ],
+    },
+    { start: 11, end: 12, name: "Styles" },
+    { start: 13, end: 14, name: "Styles" },
+    {
+      start: 15,
+      end: 25,
+      name: "DOMEvent",
+      submarkers: [
+        {
+          start: 17,
+          end: 24,
+          name: "Javascript",
+          submarkers: [{ start: 18, end: 19, name: "GarbageCollection" }],
+        },
+      ],
+    },
+  ],
+};