1 files changed, 87 insertions, 0 deletions
diff --git a/testing/specialpowers/content/AppTestDelegateParent.sys.mjs b/testing/specialpowers/content/AppTestDelegateParent.sys.mjs
new file mode 100644
index 0000000000..8264780a9a
--- /dev/null
+++ b/testing/specialpowers/content/AppTestDelegateParent.sys.mjs
@@ -0,0 +1,87 @@
+/* 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/. */
+
+import { XPCOMUtils } from "resource://gre/modules/XPCOMUtils.sys.mjs";
+
+const lazy = {};
+
+// Each app needs to implement this
+XPCOMUtils.defineLazyModuleGetters(lazy, {
+  AppUiTestDelegate: "resource://testing-common/AppUiTestDelegate.jsm",
+});
+
+export class AppTestDelegateParent extends JSWindowActorParent {
+  constructor() {
+    super();
+    this._tabs = new Map();
+  }
+
+  get browser() {
+    return this.browsingContext.top.embedderElement;
+  }
+
+  get window() {
+    return this.browser.ownerGlobal;
+  }
+
+  async receiveMessage(message) {
+    const { extensionId, url, waitForLoad, tabId } = message.data;
+    switch (message.name) {
+      case "DOMContentLoaded":
+      case "load": {
+        return this.browser?.dispatchEvent(
+          new CustomEvent(`AppTestDelegate:${message.name}`, {
+            detail: {
+              browsingContext: this.browsingContext,
+              ...message.data,
+            },
+          })
+        );
+      }
+      case "clickPageAction":
+        return lazy.AppUiTestDelegate.clickPageAction(this.window, extensionId);
+      case "clickBrowserAction":
+        return lazy.AppUiTestDelegate.clickBrowserAction(
+          this.window,
+          extensionId
+        );
+      case "closePageAction":
+        return lazy.AppUiTestDelegate.closePageAction(this.window, extensionId);
+      case "closeBrowserAction":
+        return lazy.AppUiTestDelegate.closeBrowserAction(
+          this.window,
+          extensionId
+        );
+      case "awaitExtensionPanel":
+        // The desktop delegate returns a <browser>, but that cannot be sent
+        // over IPC, so just ignore it. The promise resolves when the panel and
+        // its content is fully loaded.
+        await lazy.AppUiTestDelegate.awaitExtensionPanel(
+          this.window,
+          extensionId
+        );
+        return null;
+      case "openNewForegroundTab": {
+        // We cannot send the tab object across process so let's store it with
+        // a unique ID here.
+        const uuid = Services.uuid.generateUUID().toString();
+        const tab = await lazy.AppUiTestDelegate.openNewForegroundTab(
+          this.window,
+          url,
+          waitForLoad
+        );
+        this._tabs.set(uuid, tab);
+        return uuid;
+      }
+      case "removeTab": {
+        const tab = this._tabs.get(tabId);
+        this._tabs.delete(tabId);
+        return lazy.AppUiTestDelegate.removeTab(tab);
+      }
+
+      default:
+        throw new Error(`Unknown Test API: ${message.name}.`);
+    }
+  }
+}