1 files changed, 107 insertions, 0 deletions
diff --git a/devtools/client/inspector/markup/test/browser_markup_flex_display_badge.js b/devtools/client/inspector/markup/test/browser_markup_flex_display_badge.js
new file mode 100644
index 0000000000..a40f50dcee
--- /dev/null
+++ b/devtools/client/inspector/markup/test/browser_markup_flex_display_badge.js
@@ -0,0 +1,107 @@
+/* Any copyright is dedicated to the Public Domain.
+ http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+// Tests that the flex display badge toggles on the flexbox highlighter.
+
+const TEST_URI = `
+  <style type="text/css">
+    #flex {
+      display: flex;
+    }
+  </style>
+  <div id="flex"></div>
+`;
+
+add_task(async function () {
+  await addTab("data:text/html;charset=utf-8," + encodeURIComponent(TEST_URI));
+  const { inspector } = await openLayoutView();
+  const { store } = inspector;
+  const HIGHLIGHTER_TYPE = inspector.highlighters.TYPES.FLEXBOX;
+  const {
+    getActiveHighlighter,
+    getNodeForActiveHighlighter,
+    waitForHighlighterTypeShown,
+    waitForHighlighterTypeHidden,
+  } = getHighlighterTestHelpers(inspector);
+
+  info("Check the flex display badge is shown and not active.");
+  await selectNode("#flex", inspector);
+
+  info("Wait until the flexbox store has been updated");
+  await waitUntilState(
+    store,
+    state =>
+      state.flexbox.flexContainer.nodeFront === inspector.selection.nodeFront
+  );
+
+  const flexContainer = await getContainerForSelector("#flex", inspector);
+  const flexDisplayBadge = flexContainer.elt.querySelector(
+    ".inspector-badge.interactive[data-display]"
+  );
+  ok(
+    !flexDisplayBadge.classList.contains("active"),
+    "flex display badge is not active."
+  );
+  ok(
+    flexDisplayBadge.classList.contains("interactive"),
+    "flex display badge is interactive."
+  );
+
+  info("Check the initial state of the flex highlighter.");
+  ok(
+    !getActiveHighlighter(HIGHLIGHTER_TYPE),
+    "No flexbox highlighter exists in the highlighters overlay."
+  );
+  ok(
+    !getNodeForActiveHighlighter(HIGHLIGHTER_TYPE),
+    "No flexbox highlighter is shown."
+  );
+
+  info("Toggling ON the flexbox highlighter from the flex display badge.");
+  const onHighlighterShown = waitForHighlighterTypeShown(HIGHLIGHTER_TYPE);
+  let onCheckboxChange = waitUntilState(
+    store,
+    state => state.flexbox.highlighted
+  );
+  flexDisplayBadge.click();
+  await onHighlighterShown;
+  await onCheckboxChange;
+
+  info(
+    "Check the flexbox highlighter is created and flex display badge state."
+  );
+  ok(
+    getActiveHighlighter(HIGHLIGHTER_TYPE),
+    "Flexbox highlighter is created in the highlighters overlay."
+  );
+  ok(
+    getNodeForActiveHighlighter(HIGHLIGHTER_TYPE),
+    "Flexbox highlighter is shown."
+  );
+  ok(
+    flexDisplayBadge.classList.contains("active"),
+    "flex display badge is active."
+  );
+  ok(
+    flexDisplayBadge.classList.contains("interactive"),
+    "flex display badge is interactive."
+  );
+
+  info("Toggling OFF the flexbox highlighter from the flex display badge.");
+  const onHighlighterHidden = waitForHighlighterTypeHidden(HIGHLIGHTER_TYPE);
+  onCheckboxChange = waitUntilState(store, state => !state.flexbox.highlighted);
+  flexDisplayBadge.click();
+  await onHighlighterHidden;
+  await onCheckboxChange;
+
+  ok(
+    !flexDisplayBadge.classList.contains("active"),
+    "flex display badge is not active."
+  );
+  ok(
+    flexDisplayBadge.classList.contains("interactive"),
+    "flex display badge is interactive."
+  );
+});