1 files changed, 83 insertions, 0 deletions
diff --git a/devtools/client/shared/test/browser_flame-graph-01.js b/devtools/client/shared/test/browser_flame-graph-01.js
new file mode 100644
index 0000000000..b88afca99a
--- /dev/null
+++ b/devtools/client/shared/test/browser_flame-graph-01.js
@@ -0,0 +1,83 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+// Tests that flame graph widget works properly.
+
+const { FlameGraph } = require("devtools/client/shared/widgets/FlameGraph");
+
+add_task(async function() {
+  await addTab("about:blank");
+  await performTest();
+  gBrowser.removeCurrentTab();
+});
+
+async function performTest() {
+  const { host, doc } = await createHost();
+  doc.body.setAttribute(
+    "style",
+    "position: fixed; width: 100%; height: 100%; margin: 0;"
+  );
+
+  const graph = new FlameGraph(doc.body);
+
+  let readyEventEmitted;
+  graph.once("ready", () => {
+    readyEventEmitted = true;
+  });
+
+  await graph.ready();
+  ok(readyEventEmitted, "The 'ready' event should have been emitted");
+
+  testGraph(host, graph);
+
+  await graph.destroy();
+  host.destroy();
+}
+
+function testGraph(host, graph) {
+  ok(
+    graph._container.classList.contains("flame-graph-widget-container"),
+    "The correct graph container was created."
+  );
+  ok(
+    graph._canvas.classList.contains("flame-graph-widget-canvas"),
+    "The correct graph container was created."
+  );
+
+  const bounds = host.frame.getBoundingClientRect();
+
+  is(
+    graph.width,
+    Math.round(bounds.width * window.devicePixelRatio),
+    "The graph has the correct width."
+  );
+  is(
+    graph.height,
+    Math.round(bounds.height * window.devicePixelRatio),
+    "The graph has the correct height."
+  );
+
+  ok(
+    graph._selection.start === null,
+    "The graph's selection start value is initially null."
+  );
+  ok(
+    graph._selection.end === null,
+    "The graph's selection end value is initially null."
+  );
+
+  ok(
+    graph._selectionDragger.origin === null,
+    "The graph's dragger origin value is initially null."
+  );
+  ok(
+    graph._selectionDragger.anchor.start === null,
+    "The graph's dragger anchor start value is initially null."
+  );
+  ok(
+    graph._selectionDragger.anchor.end === null,
+    "The graph's dragger anchor end value is initially null."
+  );
+}