1 files changed, 79 insertions, 0 deletions
diff --git a/devtools/client/memory/test/xpcshell/test_dominator_trees_01.js b/devtools/client/memory/test/xpcshell/test_dominator_trees_01.js
new file mode 100644
index 0000000000..9b8bbb2533
--- /dev/null
+++ b/devtools/client/memory/test/xpcshell/test_dominator_trees_01.js
@@ -0,0 +1,79 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+// Test that we can compute and fetch the dominator tree for a snapshot.
+
+const {
+  dominatorTreeState,
+  treeMapState,
+} = require("resource://devtools/client/memory/constants.js");
+const {
+  takeSnapshotAndCensus,
+  computeAndFetchDominatorTree,
+} = require("resource://devtools/client/memory/actions/snapshot.js");
+
+add_task(async function () {
+  const front = new StubbedMemoryFront();
+  const heapWorker = new HeapAnalysesClient();
+  await front.attach();
+  const store = Store();
+  const { getState, dispatch } = store;
+
+  dispatch(takeSnapshotAndCensus(front, heapWorker));
+  await waitUntilCensusState(store, s => s.treeMap, [treeMapState.SAVED]);
+  ok(
+    !getState().snapshots[0].dominatorTree,
+    "There shouldn't be a dominator tree model yet since it is not computed " +
+      "until we switch to the dominators view."
+  );
+
+  // Change to the dominator tree view.
+  dispatch(
+    computeAndFetchDominatorTree(heapWorker, getState().snapshots[0].id)
+  );
+  ok(
+    getState().snapshots[0].dominatorTree,
+    "Should now have a dominator tree model for the selected snapshot"
+  );
+
+  // Wait for the dominator tree to start being computed.
+  await waitUntilState(
+    store,
+    state =>
+      state.snapshots[0].dominatorTree.state === dominatorTreeState.COMPUTING
+  );
+  ok(true, "The dominator tree started computing");
+  ok(
+    !getState().snapshots[0].dominatorTree.root,
+    "When the dominator tree is computing, we should not have its root"
+  );
+
+  // Wait for the dominator tree to finish computing and start being fetched.
+  await waitUntilState(
+    store,
+    state =>
+      state.snapshots[0].dominatorTree.state === dominatorTreeState.FETCHING
+  );
+  ok(true, "The dominator tree started fetching");
+  ok(
+    !getState().snapshots[0].dominatorTree.root,
+    "When the dominator tree is fetching, we should not have its root"
+  );
+
+  // Wait for the dominator tree to finish being fetched.
+  await waitUntilState(
+    store,
+    state =>
+      state.snapshots[0].dominatorTree.state === dominatorTreeState.LOADED
+  );
+  ok(true, "The dominator tree was fetched");
+  ok(
+    getState().snapshots[0].dominatorTree.root,
+    "When the dominator tree is loaded, we should have its root"
+  );
+
+  heapWorker.destroy();
+  await front.detach();
+});