chore: replace jest.fn with vi.fn

smithy-lang · May 10, 2024 · 536a13d · 536a13d
1 parent d2076eb
commit 536a13d
Show file tree

Hide file tree

Showing 67 changed files with 300 additions and 300 deletions.
diff --git a/packages/abort-controller/src/AbortSignal.spec.ts b/packages/abort-controller/src/AbortSignal.spec.ts
@@ -13,7 +13,7 @@ describe("AbortSignal", () => {
   it("should invoke the onabort handler when the signal is aborted", () => {
     const controller = new AbortController();
     const { signal } = controller;
-    const abortHandler = jest.fn();
+    const abortHandler = vi.fn();
     signal.onabort = abortHandler;
     expect(abortHandler.mock.calls.length).toBe(0);
     controller.abort();
@@ -23,7 +23,7 @@ describe("AbortSignal", () => {
   it("should not invoke the onabort handler multiple time", () => {
     const controller = new AbortController();
     const { signal } = controller;
-    const abortHandler = jest.fn();
+    const abortHandler = vi.fn();
     signal.onabort = abortHandler;
     expect(abortHandler.mock.calls.length).toBe(0);
     controller.abort();

diff --git a/packages/config-resolver/src/endpointsConfig/resolveCustomEndpointsConfig.spec.ts b/packages/config-resolver/src/endpointsConfig/resolveCustomEndpointsConfig.spec.ts
@@ -13,7 +13,7 @@ describe(resolveCustomEndpointsConfig.name, () => {
 
   const mockInput = {
     endpoint: mockEndpoint,
-    urlParser: jest.fn(() => mockEndpoint),
+    urlParser: vi.fn(() => mockEndpoint),
     useDualstackEndpoint: () => Promise.resolve(false),
   } as any;
 

diff --git a/packages/config-resolver/src/endpointsConfig/resolveEndpointsConfig.spec.ts b/packages/config-resolver/src/endpointsConfig/resolveEndpointsConfig.spec.ts
@@ -16,7 +16,7 @@ describe(resolveEndpointsConfig.name, () => {
 
   const mockInput = {
     endpoint: mockEndpoint,
-    urlParser: jest.fn(() => mockEndpoint),
+    urlParser: vi.fn(() => mockEndpoint),
     useDualstackEndpoint: () => Promise.resolve(false),
     useFipsEndpoint: () => Promise.resolve(false),
   } as any;

diff --git a/packages/config-resolver/src/endpointsConfig/utils/getEndpointFromRegion.spec.ts b/packages/config-resolver/src/endpointsConfig/utils/getEndpointFromRegion.spec.ts
@@ -1,11 +1,11 @@
 import { getEndpointFromRegion } from "./getEndpointFromRegion";
 
 describe(getEndpointFromRegion.name, () => {
-  const mockRegion = jest.fn();
-  const mockUrlParser = jest.fn();
-  const mockRegionInfoProvider = jest.fn();
-  const mockUseFipsEndpoint = jest.fn();
-  const mockUseDualstackEndpoint = jest.fn();
+  const mockRegion = vi.fn();
+  const mockUrlParser = vi.fn();
+  const mockRegionInfoProvider = vi.fn();
+  const mockUseFipsEndpoint = vi.fn();
+  const mockUseDualstackEndpoint = vi.fn();
 
   const mockInput = {
     region: mockRegion,

diff --git a/packages/config-resolver/src/regionConfig/resolveRegionConfig.spec.ts b/packages/config-resolver/src/regionConfig/resolveRegionConfig.spec.ts
@@ -49,8 +49,8 @@ describe("RegionConfig", () => {
     let mockUseFipsEndpoint;
 
     beforeEach(() => {
-      mockRegionProvider = jest.fn().mockResolvedValueOnce(Promise.resolve(mockRegion));
-      mockUseFipsEndpoint = jest.fn().mockResolvedValueOnce(Promise.resolve(false));
+      mockRegionProvider = vi.fn().mockResolvedValueOnce(Promise.resolve(mockRegion));
+      mockUseFipsEndpoint = vi.fn().mockResolvedValueOnce(Promise.resolve(false));
     });
 
     afterEach(() => {

diff --git a/packages/credential-provider-imds/src/remoteProvider/retry.spec.ts b/packages/credential-provider-imds/src/remoteProvider/retry.spec.ts
@@ -4,7 +4,7 @@ describe("retry", () => {
   const successMsg = "Success";
   const errorMsg = "Expected failure";
   const retries = 10;
-  const retryable = jest.fn().mockRejectedValue(errorMsg);
+  const retryable = vi.fn().mockRejectedValue(errorMsg);
 
   afterEach(() => {
     jest.clearAllMocks();

diff --git a/packages/credential-provider-imds/src/utils/getExtendedInstanceMetadataCredentials.spec.ts b/packages/credential-provider-imds/src/utils/getExtendedInstanceMetadataCredentials.spec.ts
@@ -9,7 +9,7 @@ describe("getExtendedInstanceMetadataCredentials()", () => {
     secretAccessKey: "secret",
   };
   const logger: Logger = {
-    warn: jest.fn(),
+    warn: vi.fn(),
   } as any;
 
   beforeEach(() => {

diff --git a/packages/credential-provider-imds/src/utils/staticStabilityProvider.spec.ts b/packages/credential-provider-imds/src/utils/staticStabilityProvider.spec.ts
@@ -35,7 +35,7 @@ describe("staticStabilityProvider", () => {
   });
 
   it("should refresh credentials if provider is functional", async () => {
-    const provider = jest.fn();
+    const provider = vi.fn();
     const stableProvider = staticStabilityProvider(provider);
     const repeat = 3;
     for (let i = 0; i < repeat; i++) {
@@ -46,7 +46,7 @@ describe("staticStabilityProvider", () => {
   });
 
   it("should throw if cannot load credentials at 1st load", async () => {
-    const provider = jest.fn().mockRejectedValue("Error");
+    const provider = vi.fn().mockRejectedValue("Error");
     try {
       await staticStabilityProvider(provider)();
       fail("This provider should throw");
@@ -58,7 +58,7 @@ describe("staticStabilityProvider", () => {
   });
 
   it("should extend expired credentials if refresh fails", async () => {
-    const provider = jest.fn().mockResolvedValueOnce(mockCreds).mockRejectedValue("Error");
+    const provider = vi.fn().mockResolvedValueOnce(mockCreds).mockRejectedValue("Error");
     const stableProvider = staticStabilityProvider(provider);
     expect(await stableProvider()).toEqual(mockCreds);
     const repeat = 3;
@@ -76,7 +76,7 @@ describe("staticStabilityProvider", () => {
 
   it("should extend expired credentials if loaded expired credentials", async () => {
     const ONE_HOUR_AGO = new Date(Date.now() - 60 * 60 * 1000);
-    const provider = jest.fn().mockResolvedValue({ ...mockCreds, expiration: ONE_HOUR_AGO });
+    const provider = vi.fn().mockResolvedValue({ ...mockCreds, expiration: ONE_HOUR_AGO });
     const stableProvider = staticStabilityProvider(provider);
     const repeat = 3;
     for (let i = 0; i < repeat; i++) {
@@ -88,8 +88,8 @@ describe("staticStabilityProvider", () => {
   });
 
   it("should allow custom logger to print warning messages", async () => {
-    const provider = jest.fn().mockResolvedValueOnce(mockCreds).mockRejectedValue("Error");
-    const logger = { warn: jest.fn() } as unknown as Logger;
+    const provider = vi.fn().mockResolvedValueOnce(mockCreds).mockRejectedValue("Error");
+    const logger = { warn: vi.fn() } as unknown as Logger;
     const stableProvider = staticStabilityProvider(provider, { logger });
     expect(await stableProvider()).toEqual(mockCreds); // load initial creds
     await stableProvider();

diff --git a/packages/eventstream-codec/src/MessageDecoderStream.spec.ts b/packages/eventstream-codec/src/MessageDecoderStream.spec.ts
@@ -15,11 +15,11 @@ describe("MessageDecoderStream", () => {
     };
 
     const messageDecoderMock = {
-      decode: jest.fn().mockReturnValueOnce(message1).mockReturnValueOnce(message2),
-      feed: jest.fn(),
-      endOfStream: jest.fn(),
-      getMessage: jest.fn(),
-      getAvailableMessages: jest.fn(),
+      decode: vi.fn().mockReturnValueOnce(message1).mockReturnValueOnce(message2),
+      feed: vi.fn(),
+      endOfStream: vi.fn(),
+      getMessage: vi.fn(),
+      getAvailableMessages: vi.fn(),
     };
 
     const inputStream = async function* () {

diff --git a/packages/eventstream-codec/src/MessageEncoderStream.spec.ts b/packages/eventstream-codec/src/MessageEncoderStream.spec.ts
@@ -13,7 +13,7 @@ describe("MessageEncoderStream", () => {
     };
 
     const messageEncoderMock = {
-      encode: jest.fn().mockReturnValueOnce(new Uint8Array(1)).mockReturnValueOnce(new Uint8Array(2)),
+      encode: vi.fn().mockReturnValueOnce(new Uint8Array(1)).mockReturnValueOnce(new Uint8Array(2)),
     };
 
     const inputStream = async function* () {
@@ -49,7 +49,7 @@ describe("MessageEncoderStream", () => {
     };
 
     const messageEncoderMock = {
-      encode: jest.fn().mockReturnValueOnce(new Uint8Array(1)).mockReturnValueOnce(new Uint8Array(2)),
+      encode: vi.fn().mockReturnValueOnce(new Uint8Array(1)).mockReturnValueOnce(new Uint8Array(2)),
     };
 
     const inputStream = async function* () {

diff --git a/packages/eventstream-codec/src/SmithyMessageEncoderStream.spec.ts b/packages/eventstream-codec/src/SmithyMessageEncoderStream.spec.ts
@@ -14,7 +14,7 @@ describe("SmithyMessageEncoderStream", () => {
       body: new Uint8Array(2),
     };
 
-    const serializer = jest.fn().mockReturnValueOnce(message1).mockReturnValueOnce(message2);
+    const serializer = vi.fn().mockReturnValueOnce(message1).mockReturnValueOnce(message2);
 
     const inputStream = async function* () {
       yield "first";

diff --git a/packages/eventstream-serde-config-resolver/src/EventStreamSerdeConfig.spec.ts b/packages/eventstream-serde-config-resolver/src/EventStreamSerdeConfig.spec.ts
@@ -1,7 +1,7 @@
 import { resolveEventStreamSerdeConfig } from "./EventStreamSerdeConfig";
 
 describe("resolveEventStreamSerdeConfig", () => {
-  const eventStreamSerdeProvider = jest.fn();
+  const eventStreamSerdeProvider = vi.fn();
 
   afterEach(() => {
     jest.clearAllMocks();