diff --git a/packages/middleware-sdk-sqs/src/receive-message.ts b/packages/middleware-sdk-sqs/src/receive-message.ts index a84801878c04..aeb4dad9ceb6 100644 --- a/packages/middleware-sdk-sqs/src/receive-message.ts +++ b/packages/middleware-sdk-sqs/src/receive-message.ts @@ -44,9 +44,7 @@ export function receiveMessageMiddleware(options: PreviouslyResolved): Initializ throw new Error("Invalid MD5 checksum on messages: " + messageIds.join(", ")); } - return next({ - ...args, - }); + return resp; }; } diff --git a/packages/middleware-sdk-sqs/src/receive-messages.spec.ts b/packages/middleware-sdk-sqs/src/receive-messages.spec.ts index 2fc7c1b95f71..ae4c6ef765d9 100644 --- a/packages/middleware-sdk-sqs/src/receive-messages.spec.ts +++ b/packages/middleware-sdk-sqs/src/receive-messages.spec.ts @@ -14,6 +14,22 @@ describe("receiveMessageMiddleware", () => { mockHashDigest.mockClear(); }); + it("should only call next once", async () => { + const next = jest.fn().mockReturnValue({ + output: { + Messages: [ + { Body: "foo", MD5OfBody: "00", MessageId: "fooMessage" }, + { Body: "bar", MD5OfBody: "00", MessageId: "barMessage" }, + ], + }, + }); + const handler = receiveMessageMiddleware({ + md5: MockHash, + })(next, {} as any); + await handler({ input: {} }); + expect(next).toBeCalledTimes(1); + }); + it("should do nothing if the checksums match", async () => { const next = jest.fn().mockReturnValue({ output: {