Bug 1544936 - Part 2: Add a test for the new sendQuery method, r=jdai

Differential Revision: https://phabricator.services.mozilla.com/D27810 --HG-- extra : moz-landing-system : lando
trman · Apr 17, 2019 · 1f9c0e0 · 1f9c0e0
1 parent f17e789
commit 1f9c0e0
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/dom/ipc/tests/browser_JSWindowActor.js b/dom/ipc/tests/browser_JSWindowActor.js
@@ -89,6 +89,21 @@ add_task(async function test_asyncMessage() {
     });
 });
 
+add_task(async function test_query() {
+  await BrowserTestUtils.withNewTab({gBrowser, url: URL},
+    async function(browser) {
+      ChromeUtils.registerWindowActor("Test", windowActorOptions);
+      let parent = browser.browsingContext.currentWindowGlobal;
+      let actorParent = parent.getActor("Test");
+      ok(actorParent, "JSWindowActorParent should have value.");
+
+      let {result} = await actorParent.sendQuery("asyncAdd", {a: 10, b: 20});
+      is(result, 30);
+
+      ChromeUtils.unregisterWindowActor("Test");
+    });
+});
+
 add_task(async function test_asyncMessage_without_both_side_actor() {
   await BrowserTestUtils.withNewTab({gBrowser, url: URL},
     async function(browser) {

diff --git a/toolkit/actors/TestChild.jsm b/toolkit/actors/TestChild.jsm
@@ -17,10 +17,17 @@ class TestChild extends JSWindowActorChild {
         aMessage.data.toChild = true;
         this.sendAsyncMessage("toParent", aMessage.data);
         break;
+      case "asyncAdd":
+        let {a, b} = aMessage.data;
+        return new Promise(resolve => {
+          resolve({ result: a + b });
+        });
       case "done":
         this.done(aMessage.data);
         break;
     }
+
+    return undefined;
   }
 
   handleEvent(aEvent) {