1 files changed, 11 insertions, 2 deletions
diff --git a/toolkit/components/promiseworker/tests/xpcshell/data/worker.js b/toolkit/components/promiseworker/tests/xpcshell/data/worker.js
index 30087bdc4a..94b6dd17ad 100644
--- a/toolkit/components/promiseworker/tests/xpcshell/data/worker.js
+++ b/toolkit/components/promiseworker/tests/xpcshell/data/worker.js
@@ -12,8 +12,9 @@ importScripts("resource://gre/modules/workers/require.js");
 var PromiseWorker = require("resource://gre/modules/workers/PromiseWorker.js");
 
 var worker = new PromiseWorker.AbstractWorker();
-worker.dispatch = function (method, args = []) {
-  return Agent[method](...args);
+
+worker.dispatch = async function (method, args = []) {
+  return await Agent[method](...args);
 };
 worker.postMessage = function (...args) {
   self.postMessage(...args);
@@ -34,6 +35,14 @@ var Agent = {
     return args;
   },
 
+  async bounceWithExtraCalls(...args) {
+    let result = await worker.callMainThread("echo", [
+      "Posting something unrelated",
+    ]);
+    args.push(result.ok);
+    return args;
+  },
+
   throwError(msg, ...args) {
     throw new Error(msg);
   },