1 files changed, 103 insertions, 0 deletions
diff --git a/devtools/client/memory/test/xpcshell/test_utils-get-snapshot-totals.js b/devtools/client/memory/test/xpcshell/test_utils-get-snapshot-totals.js
new file mode 100644
index 0000000000..46879af77d
--- /dev/null
+++ b/devtools/client/memory/test/xpcshell/test_utils-get-snapshot-totals.js
@@ -0,0 +1,103 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+/**
+ * Tests that we use the correct snapshot aggregate value
+ * in `utils.getSnapshotTotals(snapshot)`
+ */
+
+const {
+  censusDisplays,
+  viewState,
+  censusState,
+} = require("resource://devtools/client/memory/constants.js");
+const {
+  getSnapshotTotals,
+} = require("resource://devtools/client/memory/utils.js");
+const {
+  takeSnapshotAndCensus,
+} = require("resource://devtools/client/memory/actions/snapshot.js");
+const {
+  setCensusDisplayAndRefresh,
+} = require("resource://devtools/client/memory/actions/census-display.js");
+const {
+  changeView,
+} = require("resource://devtools/client/memory/actions/view.js");
+
+add_task(async function() {
+  const front = new StubbedMemoryFront();
+  const heapWorker = new HeapAnalysesClient();
+  await front.attach();
+  const store = Store();
+  const { getState, dispatch } = store;
+
+  dispatch(changeView(viewState.CENSUS));
+
+  await dispatch(
+    setCensusDisplayAndRefresh(heapWorker, censusDisplays.allocationStack)
+  );
+
+  dispatch(takeSnapshotAndCensus(front, heapWorker));
+  await waitUntilCensusState(store, s => s.census, [censusState.SAVED]);
+
+  ok(
+    !getState().snapshots[0].census.display.inverted,
+    "Snapshot is not inverted"
+  );
+
+  const census = getState().snapshots[0].census;
+  let result = aggregate(census.report);
+  const totalBytes = result.bytes;
+  const totalCount = result.count;
+
+  ok(totalBytes > 0, "counted up bytes in the census");
+  ok(totalCount > 0, "counted up count in the census");
+
+  result = getSnapshotTotals(getState().snapshots[0].census);
+  equal(
+    totalBytes,
+    result.bytes,
+    "getSnapshotTotals reuslted in correct bytes"
+  );
+  equal(
+    totalCount,
+    result.count,
+    "getSnapshotTotals reuslted in correct count"
+  );
+
+  dispatch(
+    setCensusDisplayAndRefresh(
+      heapWorker,
+      censusDisplays.invertedAllocationStack
+    )
+  );
+
+  await waitUntilCensusState(store, s => s.census, [censusState.SAVING]);
+  await waitUntilCensusState(store, s => s.census, [censusState.SAVED]);
+  ok(getState().snapshots[0].census.display.inverted, "Snapshot is inverted");
+
+  result = getSnapshotTotals(getState().snapshots[0].census);
+  equal(
+    totalBytes,
+    result.bytes,
+    "getSnapshotTotals reuslted in correct bytes when inverted"
+  );
+  equal(
+    totalCount,
+    result.count,
+    "getSnapshotTotals reuslted in correct count when inverted"
+  );
+});
+
+function aggregate(report) {
+  let totalBytes = report.bytes;
+  let totalCount = report.count;
+  for (const child of report.children || []) {
+    const { bytes, count } = aggregate(child);
+    totalBytes += bytes;
+    totalCount += count;
+  }
+  return { bytes: totalBytes, count: totalCount };
+}