test_individuals_03.js

Enable keyboard shortcuts

/* Any copyright is dedicated to the Public Domain.

   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

// Test switching to the individuals view when we are in the diffing view.

const {

  censusState,

  diffingState,

  viewState,

  individualsState,

} = require("resource://devtools/client/memory/constants.js");

const {

  fetchIndividuals,

  takeSnapshotAndCensus,

} = require("resource://devtools/client/memory/actions/snapshot.js");

const {

  changeView,

  popViewAndRefresh,

} = require("resource://devtools/client/memory/actions/view.js");

const {

  selectSnapshotForDiffingAndRefresh,

} = require("resource://devtools/client/memory/actions/diffing.js");

const EXPECTED_INDIVIDUAL_STATES = [

  individualsState.COMPUTING_DOMINATOR_TREE,

  individualsState.FETCHING,

  individualsState.FETCHED,

];

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));

  // Take two snapshots and diff them from each other.

  dispatch(takeSnapshotAndCensus(front, heapWorker));

  dispatch(takeSnapshotAndCensus(front, heapWorker));

  await waitUntilCensusState(store, s => s.census, [

    censusState.SAVED,

    censusState.SAVED,

]);

  dispatch(changeView(viewState.DIFFING));

  dispatch(

    selectSnapshotForDiffingAndRefresh(heapWorker, getState().snapshots[0])

);

  dispatch(

    selectSnapshotForDiffingAndRefresh(heapWorker, getState().snapshots[1])

);

  await waitUntilState(store, state => {

    return state.diffing && state.diffing.state === diffingState.TOOK_DIFF;

});

  ok(getState().diffing.census);

  // Fetch individuals.

  const root = getState().diffing.census.report;

  ok(root, "Should have a census");

  const reportLeafIndex = findReportLeafIndex(root);

  ok(reportLeafIndex, "Should get a reportLeafIndex");

  const snapshotId = getState().diffing.secondSnapshotId;

  ok(snapshotId, "Should have a snapshot id");

  const breakdown = getState().censusDisplay.breakdown;

  ok(breakdown, "Should have a breakdown");

  dispatch(

    fetchIndividuals(heapWorker, snapshotId, breakdown, reportLeafIndex)

);

  for (const state of EXPECTED_INDIVIDUAL_STATES) {

    await waitUntilState(store, s => {

      return (

        s.view.state === viewState.INDIVIDUALS &&

        s.individuals &&

        s.individuals.state === state

);

});

    ok(true, `Reached state = ${state}`);

  ok(getState().individuals, "Should have individuals state");

  ok(getState().individuals.nodes, "Should have individuals nodes");

ok(

    !!getState().individuals.nodes.length,

    "Should have a positive number of nodes"

);

  // Pop the view back to the diffing.

  dispatch(popViewAndRefresh(heapWorker));

  await waitUntilState(store, state => {

    return state.diffing && state.diffing.state === diffingState.TOOK_DIFF;

});

ok(

    getState().diffing.census.report,

    "We have our census diff again after popping back to the last view"

);

  heapWorker.destroy();

  await front.detach();

});