diff options
| author | Mohamed Bassem <me@mbassem.com> | 2025-11-09 20:13:39 +0000 |
|---|---|---|
| committer | Mohamed Bassem <me@mbassem.com> | 2025-11-09 20:13:39 +0000 |
| commit | b28cd03a4a5f95f429a1429a59319c8a9ac986f8 (patch) | |
| tree | f8b6a50338675bd0073643b5ff31fe447795fad4 /packages/plugins/queue-restate/src/tests | |
| parent | 03161482b44bd67f6eafb3e3d51107811b638d4b (diff) | |
| download | karakeep-b28cd03a4a5f95f429a1429a59319c8a9ac986f8.tar.zst | |
refactor: Allow runner functions to return results to onComplete
Diffstat (limited to 'packages/plugins/queue-restate/src/tests')
| -rw-r--r-- | packages/plugins/queue-restate/src/tests/queue.test.ts | 8 |
1 files changed, 5 insertions, 3 deletions
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); + } }, }, { |