diff --git a/server/tests/unit/balances/track/handle-track-queue-fallback.test.ts b/server/tests/unit/balances/track/handle-track-queue-fallback.test.ts index b87f60935..d7ac7a779 100644 --- a/server/tests/unit/balances/track/handle-track-queue-fallback.test.ts +++ b/server/tests/unit/balances/track/handle-track-queue-fallback.test.ts @@ -16,6 +16,8 @@ const mockState = { runTrackV3Calls: [] as Record[], v3Error: null as unknown, }; +const trackQueueUrl = + "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo"; mock.module("@/internal/balances/track/runTrackV2.js", () => ({ runTrackV2: async (args: Record) => { @@ -69,10 +71,9 @@ describe("track queue fallback", () => { mockState.runTrackV2Calls = []; mockState.runTrackV3Calls = []; mockState.v3Error = null; - process.env.TRACK_SQS_QUEUE_URL = - "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo"; + process.env.TRACK_SQS_QUEUE_URL = trackQueueUrl; - const sqsClient = getSqsClient(); + const sqsClient = getSqsClient({ queueUrl: trackQueueUrl }); mockState.originalSend = sqsClient.send.bind(sqsClient); sqsClient.send = (async (command: { input: Record }) => { if (mockState.queueError) { @@ -100,8 +101,7 @@ describe("track queue fallback", () => { expect(mockState.runTrackV2Calls).toHaveLength(0); expect(mockState.queueCommands).toHaveLength(1); expect(mockState.queueCommands[0]).toMatchObject({ - QueueUrl: - "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo", + QueueUrl: trackQueueUrl, }); expect(response).toEqual({ customer_id: "cus_123", @@ -149,7 +149,7 @@ describe("track queue fallback", () => { }); afterEach(() => { - const sqsClient = getSqsClient(); + const sqsClient = getSqsClient({ queueUrl: trackQueueUrl }); if (mockState.originalSend) { sqsClient.send = mockState.originalSend; } diff --git a/server/tests/unit/balances/track/queueTrack.test.ts b/server/tests/unit/balances/track/queueTrack.test.ts index 62ca8d912..cefee08b7 100644 --- a/server/tests/unit/balances/track/queueTrack.test.ts +++ b/server/tests/unit/balances/track/queueTrack.test.ts @@ -8,6 +8,8 @@ const mockState = { queueCommands: [] as Record[], originalSend: null as null | SQSClient["send"], }; +const trackQueueUrl = + "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo"; mock.module("@/internal/balances/track/utils/getQueuedTrackResponse.js", () => ({ getQueuedTrackResponse: () => ({ @@ -24,9 +26,8 @@ describe("queueTrack", () => { beforeEach(() => { mockState.queueCommands = []; - process.env.TRACK_SQS_QUEUE_URL = - "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo"; - const sqsClient = getSqsClient(); + process.env.TRACK_SQS_QUEUE_URL = trackQueueUrl; + const sqsClient = getSqsClient({ queueUrl: trackQueueUrl }); mockState.originalSend = sqsClient.send.bind(sqsClient); sqsClient.send = (async (command: { input: Record }) => { mockState.queueCommands.push(command.input); @@ -57,8 +58,7 @@ describe("queueTrack", () => { expect(mockState.queueCommands).toHaveLength(1); expect(mockState.queueCommands[0]).toMatchObject({ - QueueUrl: - "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo", + QueueUrl: trackQueueUrl, MessageGroupId: "org_123:sandbox:cus_123:ent_123", MessageDeduplicationId: "req_123", }); @@ -77,7 +77,7 @@ describe("queueTrack", () => { afterEach(() => { if (mockState.originalSend) { - const sqsClient = getSqsClient(); + const sqsClient = getSqsClient({ queueUrl: trackQueueUrl }); sqsClient.send = mockState.originalSend as typeof sqsClient.send; } process.env.TRACK_SQS_QUEUE_URL = originalTrackQueueUrl; diff --git a/server/tests/unit/queue/queue-utils.test.ts b/server/tests/unit/queue/queue-utils.test.ts index ced230c43..9febace95 100644 --- a/server/tests/unit/queue/queue-utils.test.ts +++ b/server/tests/unit/queue/queue-utils.test.ts @@ -6,6 +6,8 @@ const mockState = { commands: [] as Record[], originalSend: null as null | SQSClient["send"], }; +const overrideQueueUrl = + "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo"; import { JobName } from "@/queue/JobName.js"; import { getSqsClient } from "@/queue/initSqs.js"; @@ -17,7 +19,7 @@ describe("addTaskToQueue queue override", () => { beforeEach(() => { mockState.commands = []; - const sqsClient = getSqsClient(); + const sqsClient = getSqsClient({ queueUrl: overrideQueueUrl }); mockState.originalSend = sqsClient.send.bind(sqsClient); sqsClient.send = (async (command: { input: Record }) => { mockState.commands.push(command.input); @@ -29,7 +31,7 @@ describe("addTaskToQueue queue override", () => { afterEach(() => { if (mockState.originalSend) { - const sqsClient = getSqsClient(); + const sqsClient = getSqsClient({ queueUrl: overrideQueueUrl }); sqsClient.send = mockState.originalSend as typeof sqsClient.send; } process.env.SQS_QUEUE_URL_V2 = originalSqsQueueUrl; @@ -39,8 +41,7 @@ describe("addTaskToQueue queue override", () => { test("uses the provided SQS queueUrl override", async () => { await addTaskToQueue({ jobName: JobName.Track, - queueUrl: - "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo", + queueUrl: overrideQueueUrl, messageGroupId: "org:sandbox:cus_123", messageDeduplicationId: "idem_123", payload: { @@ -57,9 +58,7 @@ describe("addTaskToQueue queue override", () => { }); expect(mockState.commands).toHaveLength(1); - expect(mockState.commands[0]?.QueueUrl).toBe( - "https://sqs.eu-west-1.amazonaws.com/123456789012/track-dev.fifo", - ); + expect(mockState.commands[0]?.QueueUrl).toBe(overrideQueueUrl); expect(mockState.commands[0]?.MessageGroupId).toBe("org:sandbox:cus_123"); expect(mockState.commands[0]?.MessageDeduplicationId).toBe("idem_123"); });