Expand example

Author: kibertoad

examples/sns-sqs/lib/01-publish-message.spec.ts

@@ -14,6 +14,14 @@ describe('Publish message', () => {
   })
 
   it('Publishes a message', async () => {
+    await publisherManager.publish(UserConsumer.SUBSCRIBED_TOPIC_NAME, {
+      type: 'user.created',
+      payload: {
+        id: '456',
+        name: 'Jane Doe',
+      },
+    })
+
     await publisherManager.publish(UserConsumer.SUBSCRIBED_TOPIC_NAME, {
       type: 'user.created',
       payload: {
@@ -24,6 +32,9 @@ describe('Publish message', () => {
 
     const receivedMessage = await userConsumer.handlerSpy.waitForMessage({
       type: 'user.created',
+      payload: {
+        name: 'John Doe',
+      },
     })
 
     expect(receivedMessage.message.payload).toMatchInlineSnapshot(`

examples/sns-sqs/lib/02-configurable-init.spec.ts

@@ -0,0 +1,60 @@
+import { SnsConsumerErrorResolver } from '@message-queue-toolkit/sns'
+import { beforeAll, describe, it } from 'vitest'
+import {
+  errorReporter,
+  logger,
+  snsClient,
+  sqsClient,
+  stsClient,
+  transactionObservabilityManager,
+} from './common/Dependencies.ts'
+import { publisherManager } from './common/TestPublisherManager.ts'
+import { UserConsumer } from './common/UserConsumer.js'
+
+// This test suite illustrates the importance of only initting the consumers you need
+// to prevent test execution time from increasing with every new consumer added
+describe('Consumer init', () => {
+  beforeAll(async () => {
+    await publisherManager.initRegisteredPublishers([UserConsumer.SUBSCRIBED_TOPIC_NAME])
+  })
+
+  it('Inits one consumer', async () => {
+    const userConsumer = new UserConsumer({
+      errorReporter,
+      logger,
+      transactionObservabilityManager,
+      consumerErrorResolver: new SnsConsumerErrorResolver(),
+      sqsClient,
+      snsClient,
+      stsClient,
+    })
+
+    await userConsumer.start()
+    await userConsumer.close()
+  })
+
+  it('Inits twenty consumers', async () => {
+    const consumers: UserConsumer[] = []
+    for (let i = 0; i < 20; i++) {
+      consumers.push(
+        new UserConsumer({
+          errorReporter,
+          logger,
+          transactionObservabilityManager,
+          consumerErrorResolver: new SnsConsumerErrorResolver(),
+          sqsClient,
+          snsClient,
+          stsClient,
+        }),
+      )
+    }
+
+    for (const userConsumer of consumers) {
+      await userConsumer.start()
+    }
+
+    for (const userConsumer of consumers) {
+      await userConsumer.close()
+    }
+  })
+})

examples/sns-sqs/lib/common/UserConsumer.ts

@@ -12,6 +12,8 @@ type SupportedMessages = ConsumerMessageSchema<
 // biome-ignore lint/complexity/noBannedTypes: to be expanded later
 type ExecutionContext = {}
 
+const isTest = true
+
 export class UserConsumer extends AbstractSnsSqsConsumer<SupportedMessages, ExecutionContext> {
   public static readonly CONSUMED_QUEUE_NAME = 'user-my_service'
   public static readonly SUBSCRIBED_TOPIC_NAME = 'user'
@@ -32,6 +34,9 @@ export class UserConsumer extends AbstractSnsSqsConsumer<SupportedMessages, Exec
             QueueName: UserConsumer.CONSUMED_QUEUE_NAME,
           },
         },
+        deletionConfig: {
+          deleteIfExists: isTest,
+        },
         locatorConfig: {
           // Topic is created by a publisher, consumer relies on it already existing.
           // Note that in order for this to work correctly you need to ensure that