chore: update tests

Author: nicholasgriffintn

package-lock.json

@@ -15,17 +15,18 @@ describe('sqs-consumer integration surface', () => {
     const sqs = new SQSClient({ region: 'us-east-1' });
     const body = JSON.stringify({ hello: 'world' });
 
-    sqsMock.on(ReceiveMessageCommand).resolvesOnce({
-      Messages: [
-        {
-          MessageId: '1',
-          ReceiptHandle: 'abc',
-          Body: body
-        }
-      ]
-    });
-
-    sqsMock.on(ReceiveMessageCommand).resolves({ Messages: [] });
+    sqsMock
+      .on(ReceiveMessageCommand)
+      .resolvesOnce({
+        Messages: [
+          {
+            MessageId: '1',
+            ReceiptHandle: 'abc',
+            Body: body
+          }
+        ]
+      })
+      .resolves({ Messages: [] });
     sqsMock.on(DeleteMessageCommand).resolves({});
 
     const processedBodies: string[] = [];
@@ -35,6 +36,7 @@ describe('sqs-consumer integration surface', () => {
       sqs,
       handleMessage: async (message) => {
         processedBodies.push(message.Body ?? '');
+        return message;
       }
     });
 

packages/system-tests/src/consumer.test.ts

@@ -1,5 +1,5 @@
 import { Producer } from 'sqs-producer';
-import { SQSClient, SendMessageCommand } from '@aws-sdk/client-sqs';
+import { SQSClient, SendMessageBatchCommand } from '@aws-sdk/client-sqs';
 import { mockClient } from 'aws-sdk-client-mock';
 import { beforeEach, describe, expect, it } from 'vitest';
 
@@ -13,7 +13,7 @@ beforeEach(() => {
 describe('sqs-producer integration surface', () => {
   it('sends a message payload with attributes', async () => {
     const sqs = new SQSClient({ region: 'us-east-1' });
-    sqsMock.on(SendMessageCommand).resolves({});
+    sqsMock.on(SendMessageBatchCommand).resolves({});
 
     const producer = Producer.create({
       queueUrl,
@@ -31,12 +31,14 @@ describe('sqs-producer integration surface', () => {
       }
     });
 
-    const calls = sqsMock.commandCalls(SendMessageCommand);
+    const calls = sqsMock.commandCalls(SendMessageBatchCommand);
     expect(calls).toHaveLength(1);
     const [{ args }] = calls;
     expect(args[0]?.input?.QueueUrl).toBe(queueUrl);
-    expect(args[0]?.input?.MessageBody).toContain('"hello":"world"');
-    expect(args[0]?.input?.MessageAttributes?.source).toEqual({
+    expect(args[0]?.input?.Entries).toHaveLength(1);
+    const [entry] = args[0]?.input?.Entries ?? [];
+    expect(entry?.MessageBody).toContain('"hello":"world"');
+    expect(entry?.MessageAttributes?.source).toEqual({
       DataType: 'String',
       StringValue: 'system-tests'
     });