fix unit tests

blitz-js · Dec 29, 2023 · 4f238c4 · 4f238c4
1 parent ec4cfa7
commit 4f238c4
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/packages/blitz-next/src/use-params.test.tsx b/packages/blitz-next/src/use-params.test.tsx
@@ -44,18 +44,18 @@ const mockRouter: NextRouter = {
     emit: vi.fn(),
   },
   isFallback: false,
+  forward: vi.fn(),
 }
 
 export function renderHook(
   hook: RenderHook,
-  {wrapper, router, dehydratedState, ...options}: RenderHookOptions = {},
+  {router, dehydratedState, ...options}: RenderHookOptions = {},
 ) {
-  if (!wrapper) {
-    wrapper = ({children}) => (
+  const wrapper = ({children}: {children: React.ReactNode}) =>
+    (
       <RouterContext.Provider value={{...mockRouter, ...router}}>{children}</RouterContext.Provider>
-    )
-    wrapper.displayName = "wrapper"
-  }
+    ) as React.ReactElement
+  wrapper.displayName = "wrapper"
 
   return defaultRenderHook(hook, {wrapper, ...options})
 }
@@ -97,14 +97,14 @@ describe("useParams", () => {
   })
 
   it("works without parameter", () => {
-    vi.mock("next/router", () => {
+    vi.mock("next/compat/router", () => {
       return {
         useRouter: vi.fn(() => ({...mockRouter, query})),
       }
     })
 
     it("works with string", () => {
-      vi.mock("next/router", () => {
+      vi.mock("next/compat/router", () => {
         return {
           useRouter: vi.fn(() => ({...mockRouter, query})),
         }
@@ -121,7 +121,7 @@ describe("useParams", () => {
     })
 
     it("works with string", () => {
-      vi.mock("next/router", () => {
+      vi.mock("next/compat/router", () => {
         return {
           useRouter: vi.fn(() => ({...mockRouter, query})),
         }
@@ -138,7 +138,7 @@ describe("useParams", () => {
     })
 
     it("works with number", () => {
-      vi.mock("next/router", () => {
+      vi.mock("next/compat/router", () => {
         return {
           useRouter: vi.fn(() => ({...mockRouter, query})),
         }
@@ -155,7 +155,7 @@ describe("useParams", () => {
     })
 
     it("works with array", () => {
-      vi.mock("next/router", () => {
+      vi.mock("next/compat/router", () => {
         return {
           useRouter: vi.fn(() => ({...mockRouter, query})),
         }
@@ -180,7 +180,7 @@ describe("useParam", () => {
   })
 
   it("works without parameter", () => {
-    vi.mock("next/router", () => {
+    vi.mock("next/compat/router", () => {
       return {
         useRouter: vi.fn(() => ({...mockRouter, query})),
       }
@@ -201,7 +201,7 @@ describe("useParam", () => {
   })
 
   it("works with string", () => {
-    vi.mock("next/router", () => {
+    vi.mock("next/compat/router", () => {
       return {
         useRouter: vi.fn(() => ({...mockRouter, query})),
       }