Fix an issue with methods interacting with unblocking mechanisms

packages/ddp-client/client/queueStubsHelpers.js

@@ -20,7 +20,7 @@ export const loadAsyncStubHelpers = () => {
 
     queue = queue.finally(() => {
       fn(resolve, reject);
-      return promise;
+      return promise.stubPromise;
     });
 
     promise.finally(() => {
@@ -95,13 +95,14 @@ export const loadAsyncStubHelpers = () => {
           const applyAsyncPromise = oldApplyAsync.apply(this, args);
           stubPromiseResolver(applyAsyncPromise.stubPromise);
           serverPromiseResolver(applyAsyncPromise.serverPromise);
+          applyAsyncPromise.stubPromise.finally(() => {
+            finished = true;
+          });
           applyAsyncPromise
             .then((result) => {
-              finished = true;
               resolve(result);
             })
             .catch((err) => {
-              finished = true;
               reject(err);
             });
         });

packages/ddp-client/test/async_stubs/client.js

@@ -320,3 +320,33 @@ Tinytest.addAsync(
     test.equal(serverEvents, ["server-only-sync", "publication"]);
   }
 );
+
+Tinytest.addAsync(
+  "method interaction with unblocking mechanism",
+  async function (test) {
+    await Meteor.callAsync("getAndResetEvents");
+
+    Meteor.callAsync("unblockedMethod", { delay: 200 }); // unblock + sleep for 200 milliseconds
+    Meteor.callAsync("blockingMethod"); // run straight + block
+
+    let serverEvents = await Meteor.callAsync("getAndResetEvents");
+
+    test.equal(
+      serverEvents,
+      ["unblock start", "blockingMethod"],
+      "should have started the unblock method and the block method straight"
+    );
+
+    return new Promise((resolve) =>
+      setTimeout(async () => {
+        serverEvents = await Meteor.callAsync("getAndResetEvents");
+        test.equal(
+          serverEvents,
+          ["unblock end"],
+          "should have ended the unblock method as sleep finished"
+        );
+        resolve();
+      }, 400)
+    );
+  }
+);

packages/ddp-client/test/async_stubs/server_setup.js

@@ -40,7 +40,16 @@ Meteor.methods({
     events.push('callAsyncStubFromAsyncStub');
 
     return 'server result';
-  }
+  },
+  async 'unblockedMethod'({ delay }) {
+    events.push('unblock start');
+    this.unblock();
+    await Meteor._sleepForMs(delay);
+    events.push('unblock end');
+  },
+  'blockingMethod'() {
+    events.push('blockingMethod');
+  },
 });
 
 Meteor.publish("simple-publication", function () {