1 files changed, 83 insertions, 0 deletions
diff --git a/mobile/android/geckoview/src/androidTest/assets/web_extensions/test-support/TestSupportChild.jsm b/mobile/android/geckoview/src/androidTest/assets/web_extensions/test-support/TestSupportChild.jsm
new file mode 100644
index 0000000000..58c60a4e6c
--- /dev/null
+++ b/mobile/android/geckoview/src/androidTest/assets/web_extensions/test-support/TestSupportChild.jsm
@@ -0,0 +1,83 @@
+/* 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/. */
+
+const { GeckoViewActorChild } = ChromeUtils.importESModule(
+  "resource://gre/modules/GeckoViewActorChild.sys.mjs"
+);
+
+const EXPORTED_SYMBOLS = ["TestSupportChild"];
+
+class TestSupportChild extends GeckoViewActorChild {
+  receiveMessage(aMsg) {
+    debug`receiveMessage: ${aMsg.name}`;
+
+    switch (aMsg.name) {
+      case "FlushApzRepaints":
+        return new Promise(resolve => {
+          const repaintDone = () => {
+            debug`APZ flush done`;
+            Services.obs.removeObserver(repaintDone, "apz-repaints-flushed");
+            resolve();
+          };
+          Services.obs.addObserver(repaintDone, "apz-repaints-flushed");
+          if (this.contentWindow.windowUtils.flushApzRepaints()) {
+            debug`Flushed APZ repaints, waiting for callback...`;
+          } else {
+            debug`Flushing APZ repaints was a no-op, triggering callback directly...`;
+            repaintDone();
+          }
+        });
+      case "PromiseAllPaintsDone":
+        return new Promise(resolve => {
+          const window = this.contentWindow;
+          const utils = window.windowUtils;
+
+          function waitForPaints() {
+            // Wait until paint suppression has ended
+            if (utils.paintingSuppressed) {
+              dump`waiting for paint suppression to end...`;
+              window.setTimeout(waitForPaints, 0);
+              return;
+            }
+
+            if (utils.isMozAfterPaintPending) {
+              dump`waiting for paint...`;
+              window.addEventListener("MozAfterPaint", waitForPaints, {
+                once: true,
+              });
+              return;
+            }
+            resolve();
+          }
+          waitForPaints();
+        });
+      case "GetLinkColor": {
+        const { selector } = aMsg.data;
+        const element = this.document.querySelector(selector);
+        if (!element) {
+          throw new Error("No element for " + selector);
+        }
+        const color =
+          this.contentWindow.windowUtils.getVisitedDependentComputedStyle(
+            element,
+            "",
+            "color"
+          );
+        return color;
+      }
+      case "SetResolutionAndScaleTo": {
+        return new Promise(resolve => {
+          const window = this.contentWindow;
+          const { resolution } = aMsg.data;
+          window.visualViewport.addEventListener("resize", () => resolve(), {
+            once: true,
+          });
+          window.windowUtils.setResolutionAndScaleTo(resolution);
+        });
+      }
+    }
+    return null;
+  }
+}
+const { debug } = TestSupportChild.initLogging("GeckoViewTestSupport");