Update tests

Author: mayank1513

lib/src/client/core/core.test.tsx

@@ -1,9 +1,8 @@
 import { act, cleanup, fireEvent, render, renderHook } from "@testing-library/react";
 import { afterEach, beforeEach, describe, test } from "vitest";
-import { ServerTarget } from "../../server";
 import { Core } from "./core";
 import { useMode } from "../../hooks";
-import { COOKIE_KEY, DARK, LIGHT, SYSTEM } from "../../constants";
+import { COOKIE_KEY, DARK, LIGHT, MEDIA } from "../../constants";
 
 describe("theme-switcher", () => {
   afterEach(cleanup);
@@ -35,5 +34,13 @@ describe("theme-switcher", () => {
     expect(hook.result.current.mode).toBe(DARK);
   });
 
-  test.todo("test media change event -- not supported by fireEvent");
+  test("test media change event", async ({ expect }) => {
+    const hook = renderHook(() => useMode());
+    await act(() => {
+      // globalThis.window.media = LIGHT as ResolvedScheme;
+      // @ts-expect-error -- ok
+      matchMedia(MEDIA).onchange?.();
+    });
+    expect(hook.result.current.mode).toBe(DARK);
+  });
 });

lib/src/utils.ts

@@ -17,7 +17,8 @@ const DEFAULT_STORE_VAL: Store = {
 export const useStore = () =>
   useRGS<Store>("ndm", () =>
     typeof localStorage === "undefined"
-      ? DEFAULT_STORE_VAL
+      ? /* v8 ignore next */
+        DEFAULT_STORE_VAL
       : {
           m: (localStorage.getItem(COOKIE_KEY) ?? SYSTEM) as ColorSchemePreference,
           s: (matchMedia(MEDIA).matches ? DARK : LIGHT) as ResolvedScheme,

lib/vitest.setup.ts

@@ -7,9 +7,13 @@ Object.defineProperty(window, "matchMedia", {
   value: vi.fn().mockImplementation((query: string) => ({
     matches: query.includes(window.media),
     media: query,
-    onchange: () => mediaListeners.forEach(listener => listener()),
-    addEventListener: (listener: () => void) => mediaListeners.push(listener),
-    removeEventListener: vi.fn(),
+    onchange() {
+      this.matches = query.includes(window.media);
+      mediaListeners.forEach(listener => listener());
+    },
+    addEventListener: (_: string, listener: () => void) => mediaListeners.push(listener),
+    removeEventListener: (_: string, listener: () => void) =>
+      mediaListeners.splice(mediaListeners.indexOf(listener), 1),
     dispatchEvent: vi.fn(),
   })),
 });