tenfourfox/devtools/client/inspector/test/browser_inspector_highlighter-selector_02.js

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

"use strict";

// Test that the custom selector highlighter creates highlighters for nodes in
// the right frame.

const FRAME_SRC = "data:text/html;charset=utf-8," +
  "<div class=sub-level-node></div>";

const TEST_URL = "data:text/html;charset=utf-8," +
  "<div class=root-level-node></div>" +
  "<iframe src=\"" + FRAME_SRC + "\" />";

const TEST_DATA = [{
  selector: ".root-level-node",
  containerCount: 1
}, {
  selector: ".sub-level-node",
  containerCount: 0
}, {
  inIframe: true,
  selector: ".root-level-node",
  containerCount: 0
}, {
  inIframe: true,
  selector: ".sub-level-node",
  containerCount: 1
}];

add_task(function*() {
  let {inspector, testActor} = yield openInspectorForURL(TEST_URL);
  let front = inspector.inspector;
  let highlighter = yield front.getHighlighterByType("SelectorHighlighter");

  for (let {inIframe, selector, containerCount} of TEST_DATA) {
    info("Showing the highlighter on " + selector + ". Expecting " +
      containerCount + " highlighter containers");

    let contextNode;
    if (inIframe) {
      contextNode = yield getNodeFrontInFrame("body", "iframe", inspector);
    } else {
      contextNode = yield getNodeFront("body", inspector);
    }

    yield highlighter.show(contextNode, {selector});

    let nb = yield testActor.getSelectorHighlighterBoxNb(highlighter.actorID);
    ok(nb !== null, "The number of highlighters was retrieved");

    is(nb, containerCount, "The correct number of highlighers were created");
    yield highlighter.hide();
  }

  yield highlighter.finalize();
});