chore: update for vitest 4 mocking

Author: tstephen-nhs

packages/nhsNotifyLambda/tests/testAuth.test.ts

@@ -68,7 +68,9 @@ describe("tokenExchange", () => {
       },
       sign: vi.fn().mockImplementation(() => Promise.resolve("signed.jwt.token"))
     }
-    mockSignJWTConstructor.mockImplementation(() => fakeJwtInstance)
+    mockSignJWTConstructor.mockImplementation(function() {
+      return fakeJwtInstance
+    })
 
     // Mock the HTTP call
     nock(`${host}`)
@@ -101,18 +103,20 @@ describe("tokenExchange", () => {
     }
 
     mockImportPKCS8.mockImplementation(() => Promise.resolve("imported"))
-    mockSignJWTConstructor.mockImplementation(() => ({
-      setProtectedHeader() {
-        return this
-      },
-      setIssuedAt() {
-        return this
-      },
-      setExpirationTime() {
-        return this
-      },
-      sign: vi.fn().mockImplementation(() => Promise.resolve("jwt-tkn"))
-    }))
+    mockSignJWTConstructor.mockImplementation(function() {
+      return ({
+        setProtectedHeader() {
+          return this
+        },
+        setIssuedAt() {
+          return this
+        },
+        setExpirationTime() {
+          return this
+        },
+        sign: vi.fn().mockImplementation(() => Promise.resolve("jwt-tkn"))
+      })
+    })
 
     nock(`${host}`)
       .post("/oauth2/token")
@@ -131,18 +135,20 @@ describe("tokenExchange", () => {
     }
 
     mockImportPKCS8.mockImplementation(() => Promise.resolve("imported"))
-    mockSignJWTConstructor.mockImplementation(() => ({
-      setProtectedHeader() {
-        return this
-      },
-      setIssuedAt() {
-        return this
-      },
-      setExpirationTime() {
-        return this
-      },
-      sign: vi.fn().mockImplementation(() => Promise.resolve("jwt-tkn"))
-    }))
+    mockSignJWTConstructor.mockImplementation(function() {
+      return ({
+        setProtectedHeader() {
+          return this
+        },
+        setIssuedAt() {
+          return this
+        },
+        setExpirationTime() {
+          return this
+        },
+        sign: vi.fn().mockImplementation(() => Promise.resolve("jwt-tkn"))
+      })
+    })
 
     nock(`${host}`)
       .post("/oauth2/token")

packages/nhsNotifyLambda/tests/testHelpers.ts

@@ -10,9 +10,9 @@ vi.mock("@aws-sdk/client-sqs", async (importOriginal) => {
   const sqs = await importOriginal<typeof import("@aws-sdk/client-sqs")>()
   return {
     ...sqs,
-    SQSClient: vi.fn().mockImplementation(() => ({
-      send: mockSend
-    }))
+    SQSClient: vi.fn(class {
+      send = mockSend
+    })
   }
 })
 

packages/nhsNotifyLambda/tests/testNhsNotifyLambda.test.ts

@@ -59,11 +59,11 @@ vi.mock(
   "@aws-lambda-powertools/logger",
   async () => ({
     __esModule: true,
-    Logger: vi.fn().mockImplementation(() => ({
-      info: mockInfo,
-      error: mockError,
-      warn: mockWarn
-    }))
+    Logger: vi.fn(class {
+      info = mockInfo
+      error = mockError
+      warn = mockWarn
+    })
   })
 )
 

packages/nhsNotifyLambda/tests/testUtils.test.ts

@@ -42,9 +42,9 @@ vi.mock(
   "@aws-lambda-powertools/parameters/ssm",
   async () => ({
     __esModule: true,
-    SSMProvider: vi.fn().mockImplementation(() => ({
-      getParametersByName: mockGetParametersByName
-    }))
+    SSMProvider: vi.fn(class {
+      getParametersByName = mockGetParametersByName
+    })
   })
 )
 

packages/postDatedLambda/tests/testDatabaseClient.test.ts

@@ -18,9 +18,9 @@ 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
-    }))
+    DynamoDBClient: vi.fn(class {
+      send = mockSend
+    })
   }
 })
 const {

packages/postDatedLambda/tests/testSqs.test.ts

@@ -20,9 +20,9 @@ vi.mock("@aws-sdk/client-sqs", async () => {
   const sqs = await vi.importActual<typeof import("@aws-sdk/client-sqs")>("@aws-sdk/client-sqs")
   return {
     ...sqs,
-    SQSClient: vi.fn().mockImplementation(() => ({
-      send: mockSend
-    }))
+    SQSClient: vi.fn(class {
+      send = mockSend
+    })
   }
 })
 

packages/updatePrescriptionStatus/tests/testDatabaseClient.test.ts

@@ -18,7 +18,9 @@ vi.mock(
     const mod = await importOriginal()
     return {
       ...mod,
-      DynamoDBClient: vi.fn().mockImplementation(() => ({send: mockSend}))
+      DynamoDBClient: vi.fn(class {
+        send = mockSend
+      })
     }
   }
 )

packages/updatePrescriptionStatus/tests/testHandler.test.ts

@@ -54,7 +54,9 @@ vi.mock("@aws-sdk/client-dynamodb", async (importOriginal) => {
   const mod = await importOriginal<typeof import("@aws-sdk/client-dynamodb")>()
   return {
     ...mod,
-    DynamoDBClient: vi.fn().mockImplementation(() => ({send: dynamoDBMockSend}))
+    DynamoDBClient: vi.fn(class {
+      send = dynamoDBMockSend
+    })
   }
 })
 

packages/updatePrescriptionStatus/tests/testPrescriptionIntercept.test.ts

@@ -34,7 +34,9 @@ vi.mock("@aws-sdk/client-dynamodb", async (importOriginal) => {
   const mod = await importOriginal<typeof import("@aws-sdk/client-dynamodb")>()
   return {
     ...mod,
-    DynamoDBClient: vi.fn().mockImplementation(() => ({send: mockSend}))
+    DynamoDBClient: vi.fn(class {
+      send = mockSend
+    })
   }
 })
 

packages/updatePrescriptionStatus/tests/testSqsClient.test.ts

@@ -30,7 +30,9 @@ vi.mock("@aws-sdk/client-sqs", async (importOriginal: () => Promise<typeof impor
   const mod = await importOriginal()
   return {
     ...mod,
-    SQSClient: vi.fn().mockImplementation(() => ({send: mockSend}))
+    SQSClient: vi.fn(class {
+      send = mockSend
+    })
   }
 })