From 9a5afb11f6438f94b4f20a6db015a40220f9ff56 Mon Sep 17 00:00:00 2001
From: Angel Mendez Cano <amendez1988@gmail.com>
Date: Fri, 2 Feb 2024 19:00:38 -0600
Subject: [PATCH] feat: update tests using vitest

- replace jest references with vitest
---
 test/Toast-test.tsx                     |  7 ++++---
 test/UrlParams-test.ts                  |  6 +++---
 test/__snapshots__/Toast-test.tsx.snap  |  2 +-
 test/environment.ts                     | 18 ------------------
 test/mocks/olmMock.ts                   |  4 +++-
 test/mocks/workerMock.ts                |  4 +++-
 test/room/checkForParallelCalls-test.ts |  8 ++++----
 test/utils.ts                           |  5 +++--
 8 files changed, 21 insertions(+), 33 deletions(-)
 delete mode 100644 test/environment.ts

diff --git a/test/Toast-test.tsx b/test/Toast-test.tsx
index 88cfbf4b..6b573523 100644
--- a/test/Toast-test.tsx
+++ b/test/Toast-test.tsx
@@ -14,6 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
+import { vi } from "vitest";
 import { screen, render } from "@testing-library/react";
 import { Toast } from "../src/Toast";
 import userEvent from "@testing-library/user-event";
@@ -35,7 +36,7 @@ test("Toast renders", () => {
 });
 
 test("Toast dismisses when clicked", async () => {
-  const onDismiss = jest.fn();
+  const onDismiss = vi.fn();
   render(
     <Toast open={true} onDismiss={onDismiss}>
       Hello world!
@@ -47,13 +48,13 @@ test("Toast dismisses when clicked", async () => {
 
 test("Toast dismisses itself after the specified timeout", async () => {
   withFakeTimers(() => {
-    const onDismiss = jest.fn();
+    const onDismiss = vi.fn();
     render(
       <Toast open={true} onDismiss={onDismiss} autoDismiss={2000}>
         Hello world!
       </Toast>,
     );
-    jest.advanceTimersByTime(2000);
+    vi.advanceTimersByTime(2000);
     expect(onDismiss).toHaveBeenCalled();
   });
 });
diff --git a/test/UrlParams-test.ts b/test/UrlParams-test.ts
index e0eb7c29..eb03d976 100644
--- a/test/UrlParams-test.ts
+++ b/test/UrlParams-test.ts
@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-import { mocked } from "jest-mock";
+import { vi } from "vitest";
 
 import { getRoomIdentifierFromUrl } from "../src/UrlParams";
 import { Config } from "../src/config/Config";
@@ -24,11 +24,11 @@ const ROOM_ID = "!d45f138fsd";
 const ORIGIN = "https://call.element.io";
 const HOMESERVER = "call.ems.host";
 
-jest.mock("../src/config/Config");
+vi.mock("../src/config/Config");
 
 describe("UrlParams", () => {
   beforeAll(() => {
-    mocked(Config.defaultServerName).mockReturnValue("call.ems.host");
+    vi.mocked(Config.defaultServerName).mockReturnValue("call.ems.host");
   });
 
   describe("handles URL with /room/", () => {
diff --git a/test/__snapshots__/Toast-test.tsx.snap b/test/__snapshots__/Toast-test.tsx.snap
index 64822136..7edfdcf7 100644
--- a/test/__snapshots__/Toast-test.tsx.snap
+++ b/test/__snapshots__/Toast-test.tsx.snap
@@ -1,4 +1,4 @@
-// Jest Snapshot v1, https://goo.gl/fbAQLP
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
 exports[`Toast renders 1`] = `
 <button
diff --git a/test/environment.ts b/test/environment.ts
deleted file mode 100644
index 5c29e1ea..00000000
--- a/test/environment.ts
+++ /dev/null
@@ -1,18 +0,0 @@
-import { TextEncoder } from "util";
-import JSDOMEnvironment_, {
-  TestEnvironment as TestEnvironment_,
-} from "jest-environment-jsdom";
-import { JestEnvironmentConfig, EnvironmentContext } from "@jest/environment";
-
-// This is a patched version of jsdom that adds TextEncoder, as a workaround for
-// https://github.com/jsdom/jsdom/issues/2524
-// Once that issue is resolved, this custom environment file can be deleted
-export default class JSDOMEnvironment extends JSDOMEnvironment_ {
-  constructor(config: JestEnvironmentConfig, context: EnvironmentContext) {
-    super(config, context);
-    this.global.TextEncoder ??= TextEncoder;
-  }
-}
-
-export const TestEnvironment =
-  TestEnvironment_ === JSDOMEnvironment_ ? JSDOMEnvironment : TestEnvironment_;
diff --git a/test/mocks/olmMock.ts b/test/mocks/olmMock.ts
index 2c18fde9..dfb9dc05 100644
--- a/test/mocks/olmMock.ts
+++ b/test/mocks/olmMock.ts
@@ -1 +1,3 @@
-module.exports = { loadOlm: jest.fn(async () => {}) };
+import { vi } from "vitest";
+
+module.exports = { loadOlm: vi.fn(async () => {}) };
diff --git a/test/mocks/workerMock.ts b/test/mocks/workerMock.ts
index 6ee58567..df75a840 100644
--- a/test/mocks/workerMock.ts
+++ b/test/mocks/workerMock.ts
@@ -1 +1,3 @@
-module.exports = jest.fn();
+import { vi } from "vitest";
+
+module.exports = vi.fn();
diff --git a/test/room/checkForParallelCalls-test.ts b/test/room/checkForParallelCalls-test.ts
index 49f5753d..605a2431 100644
--- a/test/room/checkForParallelCalls-test.ts
+++ b/test/room/checkForParallelCalls-test.ts
@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-import { Mocked, mocked } from "jest-mock";
+import { vi, Mocked } from "vitest";
 import { RoomState } from "matrix-js-sdk/src/models/room-state";
 import { PosthogAnalytics } from "../../src/analytics/PosthogAnalytics";
 import { checkForParallelCalls } from "../../src/room/checkForParallelCalls";
@@ -23,10 +23,10 @@ import { withFakeTimers } from "../utils";
 const withMockedPosthog = (
   continuation: (posthog: Mocked<PosthogAnalytics>) => void,
 ) => {
-  const posthog = mocked({
-    trackEvent: jest.fn(),
+  const posthog = vi.mocked({
+    trackEvent: vi.fn(),
   } as unknown as PosthogAnalytics);
-  const instanceSpy = jest
+  const instanceSpy = vi
     .spyOn(PosthogAnalytics, "instance", "get")
     .mockReturnValue(posthog);
   try {
diff --git a/test/utils.ts b/test/utils.ts
index 9fb282ef..f3d7874a 100644
--- a/test/utils.ts
+++ b/test/utils.ts
@@ -13,12 +13,13 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 */
+import { vi } from "vitest";
 
 export function withFakeTimers(continuation: () => void): void {
-  jest.useFakeTimers();
+  vi.useFakeTimers();
   try {
     continuation();
   } finally {
-    jest.useRealTimers();
+    vi.useRealTimers();
   }
 }