1 files changed, 65 insertions, 0 deletions
diff --git a/devtools/client/inspector/test/browser_inspector_highlighter-by-type.js b/devtools/client/inspector/test/browser_inspector_highlighter-by-type.js
new file mode 100644
index 0000000000..c9e67c8eca
--- /dev/null
+++ b/devtools/client/inspector/test/browser_inspector_highlighter-by-type.js
@@ -0,0 +1,65 @@
+/* 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";
+
+// Check that custom highlighters can be retrieved by type and that they expose
+// the expected API.
+
+const TEST_URL = "data:text/html;charset=utf-8,custom highlighters";
+
+add_task(async function () {
+  const { inspector } = await openInspectorForURL(TEST_URL);
+
+  await manyInstancesOfCustomHighlighters(inspector);
+  await showHideMethodsAreAvailable(inspector);
+  await unknownHighlighterTypeShouldntBeAccepted(inspector);
+});
+
+async function manyInstancesOfCustomHighlighters({ inspectorFront }) {
+  const h1 = await inspectorFront.getHighlighterByType("BoxModelHighlighter");
+  const h2 = await inspectorFront.getHighlighterByType("BoxModelHighlighter");
+  ok(h1 !== h2, "getHighlighterByType returns new instances every time (1)");
+
+  const h3 = await inspectorFront.getHighlighterByType(
+    "CssTransformHighlighter"
+  );
+  const h4 = await inspectorFront.getHighlighterByType(
+    "CssTransformHighlighter"
+  );
+  ok(h3 !== h4, "getHighlighterByType returns new instances every time (2)");
+  ok(
+    h3 !== h1 && h3 !== h2,
+    "getHighlighterByType returns new instances every time (3)"
+  );
+  ok(
+    h4 !== h1 && h4 !== h2,
+    "getHighlighterByType returns new instances every time (4)"
+  );
+
+  await h1.finalize();
+  await h2.finalize();
+  await h3.finalize();
+  await h4.finalize();
+}
+
+async function showHideMethodsAreAvailable({ inspectorFront }) {
+  const h1 = await inspectorFront.getHighlighterByType("BoxModelHighlighter");
+  const h2 = await inspectorFront.getHighlighterByType(
+    "CssTransformHighlighter"
+  );
+
+  ok("show" in h1, "Show method is present on the front API");
+  ok("show" in h2, "Show method is present on the front API");
+  ok("hide" in h1, "Hide method is present on the front API");
+  ok("hide" in h2, "Hide method is present on the front API");
+
+  await h1.finalize();
+  await h2.finalize();
+}
+
+async function unknownHighlighterTypeShouldntBeAccepted({ inspectorFront }) {
+  const h = await inspectorFront.getHighlighterByType("whatever");
+  ok(!h, "No highlighter was returned for the invalid type");
+}