1 files changed, 61 insertions, 0 deletions
diff --git a/devtools/client/inspector/test/browser_inspector_highlighter-selector_02.js b/devtools/client/inspector/test/browser_inspector_highlighter-selector_02.js
new file mode 100644
index 000000000..85fcaeb1c
--- /dev/null
+++ b/devtools/client/inspector/test/browser_inspector_highlighter-selector_02.js
@@ -0,0 +1,61 @@
+/* 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
+}];
+
+requestLongerTimeout(5);
+
+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();
+});