From b28cd03a4a5f95f429a1429a59319c8a9ac986f8 Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Sun, 9 Nov 2025 20:13:39 +0000
Subject: refactor: Allow runner functions to return results to onComplete

---
 packages/plugins/queue-restate/src/tests/queue.test.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'packages/plugins/queue-restate/src/tests')

diff --git a/packages/plugins/queue-restate/src/tests/queue.test.ts b/packages/plugins/queue-restate/src/tests/queue.test.ts
index e59d47cb..d716671b 100644
--- a/packages/plugins/queue-restate/src/tests/queue.test.ts
+++ b/packages/plugins/queue-restate/src/tests/queue.test.ts
@@ -90,8 +90,7 @@ describe("Restate Queue Provider", () => {
           const jobData = job.data;
           switch (jobData.type) {
             case "val":
-              testState.results.push(jobData.val);
-              break;
+              return jobData.val;
             case "err":
               throw new Error(jobData.err);
             case "stall":
@@ -108,8 +107,11 @@ describe("Restate Queue Provider", () => {
             testState.errors.push(jobData.err);
           }
         },
-        onComplete: async () => {
+        onComplete: async (_j, res) => {
           testState.inFlight--;
+          if (res) {
+            testState.results.push(res);
+          }
         },
       },
       {
-- 
cgit v1.2.3-70-g09d2