chore: migrated postDatedLambda tests passing

tstephen-nhs · tstephen-nhs · commit e20aa352ac6c · 2026-03-31T16:48:58.000Z
diff --git a/packages/postDatedLambda/.vscode/launch.json b/packages/postDatedLambda/.vscode/launch.json
@@ -1,35 +1,40 @@
 {
-  // Use IntelliSense to learn about possible attributes.
-  // Hover to view descriptions of existing attributes.
-  // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
   "version": "0.2.0",
   "configurations": [
-      {
-          "type": "node",
-          "name": "vscode-jest-tests.v2",
-          "request": "launch",
-          "args": [
-              "--runInBand",
-              "--watchAll=false",
-              "--testNamePattern",
-              "${jest.testNamePattern}",
-              "--runTestsByPath",
-              "${jest.testFile}",
-              "--config",
-              "${workspaceFolder}/jest.debug.config.ts"
-          ],
-          "cwd": "${workspaceFolder}",
-          "console": "integratedTerminal",
-          "internalConsoleOptions": "neverOpen",
-          "disableOptimisticBPs": true,
-          "program": "${workspaceFolder}/../../node_modules/.bin/jest",
-          "windows": {
-              "program": "${workspaceFolder}/node_modules/jest/bin/jest"
-          },
-          "env": {
-              "POWERTOOLS_DEV": true,
-              "NODE_OPTIONS": "--experimental-vm-modules"
-          }
-      }
+        {
+            "name": "vitest: all tests",
+            "type": "node",
+            "request": "launch",
+            "cwd": "${workspaceFolder}",
+            "runtimeExecutable": "${workspaceFolder}/../../node_modules/vitest/vitest.mjs",
+            "args": [
+                "run",
+                "--config",
+                "vitest.config.ts"
+            ],
+            "env": {
+                "POWERTOOLS_DEV": "true"
+            },
+            "console": "integratedTerminal",
+            "internalConsoleOptions": "neverOpen"
+        },
+        {
+            "name": "vitest: current file",
+            "type": "node",
+            "request": "launch",
+            "cwd": "${workspaceFolder}",
+            "runtimeExecutable": "${workspaceFolder}/../../node_modules/vitest/vitest.mjs",
+            "args": [
+                "run",
+                "${relativeFile}",
+                "--config",
+                "vitest.config.ts"
+            ],
+            "env": {
+                "POWERTOOLS_DEV": "true"
+            },
+            "console": "integratedTerminal",
+            "internalConsoleOptions": "neverOpen"
+        }
   ]
 }
diff --git a/packages/postDatedLambda/.vscode/settings.json b/packages/postDatedLambda/.vscode/settings.json
@@ -1,7 +1,3 @@
 {
-  "jest.jestCommandLine": "/workspaces/eps-prescription-status-update-api/node_modules/.bin/jest --no-cache",
-  "jest.nodeEnv": {
-    "POWERTOOLS_DEV": true,
-    "NODE_OPTIONS": "--experimental-vm-modules"
-  }
+  "testing.automaticallyOpenPeekView": "never"
 }
diff --git a/packages/postDatedLambda/package.json b/packages/postDatedLambda/package.json
@@ -7,7 +7,7 @@
   "license": "MIT",
   "type": "module",
   "scripts": {
-    "unit": "POWERTOOLS_DEV=true NODE_OPTIONS=--experimental-vm-modules jest --no-cache --coverage",
+    "unit": "POWERTOOLS_DEV=true NODE_OPTIONS=--max-old-space-size=8192 vitest run --config vitest.config.ts --coverage",
     "lint": "eslint  --max-warnings 0 --fix --config ../../eslint.config.mjs .",
     "compile": "tsc",
     "test": "npm run compile && npm run unit",
diff --git a/packages/postDatedLambda/tests/testBusinessLogic.test.ts b/packages/postDatedLambda/tests/testBusinessLogic.test.ts
@@ -3,8 +3,8 @@ import {
   describe,
   it,
   afterEach,
-  jest
-} from "@jest/globals"
+  vi
+} from "vitest"
 
 import {Logger} from "@aws-lambda-powertools/logger"
 
@@ -21,7 +21,7 @@ async function loadBusinessLogic(
   envOverrides = {}
 ): Promise<BusinessLogicModule> {
   // Makes sure that the environment is set before import each time
-  jest.resetModules()
+  vi.resetModules()
   process.env = {...ORIGINAL_ENV, ...envOverrides}
   return import("../src/businessLogic")
 }
@@ -77,7 +77,7 @@ function createMessage(
 
 afterEach(() => {
   process.env = {...ORIGINAL_ENV}
-  jest.useRealTimers()
+  vi.useRealTimers()
 })
 
 describe("businessLogic", () => {
@@ -185,8 +185,8 @@ describe("businessLogic", () => {
     })
 
     it("should classify a message as immature when LastModified is in the future", async () => {
-      jest.useFakeTimers()
-      jest.setSystemTime(new Date("2026-01-01T12:00:00.000Z"))
+      vi.useFakeTimers()
+      vi.setSystemTime(new Date("2026-01-01T12:00:00.000Z"))
       const {determineAction} = await loadBusinessLogic()
       const logger = new Logger({serviceName: "post-dated-tests"})
       const message = createMessage({
@@ -203,8 +203,8 @@ describe("businessLogic", () => {
     })
 
     it("should classify a message as matured when LastModified is in the past", async () => {
-      jest.useFakeTimers()
-      jest.setSystemTime(new Date("2026-01-03T12:00:00.000Z"))
+      vi.useFakeTimers()
+      vi.setSystemTime(new Date("2026-01-03T12:00:00.000Z"))
       const {determineAction} = await loadBusinessLogic()
       const logger = new Logger({serviceName: "post-dated-tests"})
       const message = createMessage({
@@ -221,8 +221,8 @@ describe("businessLogic", () => {
     })
 
     it("should use the provided most recent record when determining maturity", async () => {
-      jest.useFakeTimers()
-      jest.setSystemTime(new Date("2026-01-05T12:00:00.000Z"))
+      vi.useFakeTimers()
+      vi.setSystemTime(new Date("2026-01-05T12:00:00.000Z"))
       const {determineAction} = await loadBusinessLogic()
       const logger = new Logger({serviceName: "post-dated-tests"})
       const message = createMessage({
diff --git a/packages/postDatedLambda/tests/testDatabaseClient.test.ts b/packages/postDatedLambda/tests/testDatabaseClient.test.ts
@@ -2,30 +2,27 @@ import {
   expect,
   describe,
   it,
-  jest
-} from "@jest/globals"
-
-import * as dynamo from "@aws-sdk/client-dynamodb"
+  vi,
+  beforeEach
+} from "vitest"
 
 import {Logger} from "@aws-lambda-powertools/logger"
 
 import {createMockPostModifiedDataItem} from "./testUtils"
 
-// Uses unstable jest method to enable mocking while using ESM. To be replaced in future.
-export function mockDynamoDBClient() {
-  const mockSend = jest.fn()
-  jest.unstable_mockModule("@aws-sdk/client-dynamodb", () => {
-    return {
-      ...dynamo,
-      DynamoDBClient: jest.fn().mockImplementation(() => ({
-        send: mockSend
-      }))
-    }
-  })
-  return {mockSend}
-}
-
-const {mockSend} = mockDynamoDBClient()
+const {mockSend} = vi.hoisted(() => ({
+  mockSend: vi.fn()
+}))
+
+vi.mock("@aws-sdk/client-dynamodb", async () => {
+  const dynamo = await vi.importActual<typeof import("@aws-sdk/client-dynamodb")>("@aws-sdk/client-dynamodb")
+  return {
+    ...dynamo,
+    DynamoDBClient: vi.fn().mockImplementation(() => ({
+      send: mockSend
+    }))
+  }
+})
 const {
   getRecentDataItemByPrescriptionID,
   enrichMessagesWithMostRecentDataItem
diff --git a/packages/postDatedLambda/tests/testMain.test.ts b/packages/postDatedLambda/tests/testMain.test.ts
@@ -1,22 +1,29 @@
 import {
-  jest,
+  vi,
+  expect,
   describe,
   it,
   beforeAll,
   afterEach
-} from "@jest/globals"
+} from "vitest"
 
-const mockReportQueueStatus = jest.fn()
-jest.unstable_mockModule(
+const {
+  mockReportQueueStatus,
+  mockProcessPostDatedQueue
+} = vi.hoisted(() => ({
+  mockReportQueueStatus: vi.fn(),
+  mockProcessPostDatedQueue: vi.fn()
+}))
+
+vi.mock(
   "../src/sqs",
   async () => ({
     __esModule: true,
     reportQueueStatus: mockReportQueueStatus
   })
 )
 
-const mockProcessPostDatedQueue = jest.fn()
-jest.unstable_mockModule(
+vi.mock(
   "../src/orchestration",
   async () => ({
     __esModule: true,
@@ -37,8 +44,8 @@ describe("Unit test for post-dated lambda handler", () => {
   afterEach(() => {
     process.env = {...ORIGINAL_ENV}
 
-    jest.clearAllMocks()
-    jest.restoreAllMocks()
+    vi.clearAllMocks()
+    vi.restoreAllMocks()
   })
 
   it("should run the lambda handler successfully", async () => {
diff --git a/packages/postDatedLambda/tests/testOrchestration.test.ts b/packages/postDatedLambda/tests/testOrchestration.test.ts
@@ -2,32 +2,46 @@ import {
   expect,
   describe,
   it,
-  jest
-} from "@jest/globals"
+  vi,
+  beforeEach,
+  afterEach
+} from "vitest"
 
 // Mock the imports from local modules
-const mockDetermineAction = jest.fn()
-const mockComputeTimeUntilMaturity = jest.fn().mockReturnValue(300)
-jest.unstable_mockModule("../src/businessLogic", () => {
+const {
+  mockDetermineAction,
+  mockComputeTimeUntilMaturity,
+  mockEnrichMessagesWithMostRecentDataItem,
+  mockReceivePostDatedSQSMessages,
+  mockReportQueueStatus,
+  mockForwardSQSMessageToNotificationQueue,
+  mockRemoveSQSMessage,
+  mockReturnMessageToQueue
+} = vi.hoisted(() => ({
+  mockDetermineAction: vi.fn(),
+  mockComputeTimeUntilMaturity: vi.fn().mockReturnValue(300),
+  mockEnrichMessagesWithMostRecentDataItem: vi.fn(),
+  mockReceivePostDatedSQSMessages: vi.fn(),
+  mockReportQueueStatus: vi.fn(),
+  mockForwardSQSMessageToNotificationQueue: vi.fn(),
+  mockRemoveSQSMessage: vi.fn(),
+  mockReturnMessageToQueue: vi.fn()
+}))
+
+vi.mock("../src/businessLogic", () => {
   return {
     determineAction: mockDetermineAction,
     computeTimeUntilMaturity: mockComputeTimeUntilMaturity
   }
 })
 
-const mockEnrichMessagesWithMostRecentDataItem = jest.fn()
-jest.unstable_mockModule("../src/databaseClient", () => {
+vi.mock("../src/databaseClient", () => {
   return {
     enrichMessagesWithMostRecentDataItem: mockEnrichMessagesWithMostRecentDataItem
   }
 })
 
-const mockReceivePostDatedSQSMessages = jest.fn()
-const mockReportQueueStatus = jest.fn()
-const mockForwardSQSMessageToNotificationQueue = jest.fn()
-const mockRemoveSQSMessage = jest.fn()
-const mockReturnMessageToQueue = jest.fn()
-jest.unstable_mockModule("../src/sqs", () => {
+vi.mock("../src/sqs", () => {
   return {
     receivePostDatedSQSMessages: mockReceivePostDatedSQSMessages,
     reportQueueStatus: mockReportQueueStatus,
@@ -66,7 +80,7 @@ function enrich(messages: Array<PostDatedSQSMessage>): Array<PostDatedSQSMessage
 describe("orchestration", () => {
   describe("processMessages", () => {
     beforeEach(() => {
-      jest.clearAllMocks()
+      vi.clearAllMocks()
       mockForwardSQSMessageToNotificationQueue.mockReturnValue(Promise.resolve("forwarded-id"))
       mockRemoveSQSMessage.mockReturnValue(Promise.resolve())
       mockReturnMessageToQueue.mockReturnValue(Promise.resolve())
@@ -125,12 +139,17 @@ describe("orchestration", () => {
 
   describe("processPostDatedQueue", () => {
     beforeEach(() => {
-      jest.clearAllMocks()
+      vi.clearAllMocks()
       mockForwardSQSMessageToNotificationQueue.mockReturnValue(Promise.resolve("forwarded-id"))
       mockRemoveSQSMessage.mockReturnValue(Promise.resolve())
       mockReturnMessageToQueue.mockReturnValue(Promise.resolve())
     })
 
+    afterEach(() => {
+      vi.useRealTimers()
+      vi.restoreAllMocks()
+    })
+
     it("should process the SQS queue correctly", async () => {
       const mockMessages: Array<PostDatedSQSMessage> = [
         {MessageId: "1", Body: "Message 1", prescriptionData: createMockPostModifiedDataItem({})},
@@ -157,7 +176,6 @@ describe("orchestration", () => {
     })
 
     it("Should stop processing if the max runtime is exceeded", async () => {
-      jest.useFakeTimers()
       const mockMessages: Array<PostDatedSQSMessage> = [
         {MessageId: "1", Body: "Message 1", prescriptionData: createMockPostModifiedDataItem({})},
         {MessageId: "2", Body: "Message 2", prescriptionData: createMockPostModifiedDataItem({})},
@@ -171,14 +189,13 @@ describe("orchestration", () => {
       mockEnrichMessagesWithMostRecentDataItem.mockReturnValue(enrich(mockMessages))
       const {MAX_QUEUE_RUNTIME} = await import("../src/orchestration")
       mockDetermineAction.mockReturnValue(PostDatedProcessingResult.FORWARD_TO_NOTIFICATIONS)
+      vi.spyOn(Date, "now")
+        .mockReturnValueOnce(0)
+        .mockReturnValueOnce(MAX_QUEUE_RUNTIME + 1000)
 
-      const promise = processPostDatedQueue(logger)
-      // Overrun by a second
-      jest.advanceTimersByTime(MAX_QUEUE_RUNTIME + 1000)
-      await promise
+      await processPostDatedQueue(logger)
 
       expect(mockReportQueueStatus).toHaveBeenCalled()
-      jest.useRealTimers()
     })
 
     it("should continue processing batches until message count drops below threshold", async () => {
diff --git a/packages/postDatedLambda/tests/testSqs.test.ts b/packages/postDatedLambda/tests/testSqs.test.ts
diff --git a/packages/postDatedLambda/vitest.config.ts b/packages/postDatedLambda/vitest.config.ts

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,3 @@`
`1`	`1`	`{`
`2`		`- "jest.jestCommandLine": "/workspaces/eps-prescription-status-update-api/node_modules/.bin/jest --no-cache",`
`3`		`- "jest.nodeEnv": {`
`4`		`- "POWERTOOLS_DEV": true,`
`5`		`- "NODE_OPTIONS": "--experimental-vm-modules"`
`6`		`- }`
	`2`	`+ "testing.automaticallyOpenPeekView": "never"`
`7`	`3`	`}`