@@ -10,11 +10,13 @@ import { RouterProvider } from '../src/RouterProvider'
10
10
11
11
describe ( 'Transitioner' , ( ) => {
12
12
it ( 'should call router.load() when Transitioner mounts on the client' , async ( ) => {
13
+ const loader = vi . fn ( )
13
14
const rootRoute = createRootRoute ( )
14
15
const indexRoute = createRoute ( {
15
16
getParentRoute : ( ) => rootRoute ,
16
17
path : '/' ,
17
18
component : ( ) => < div > Index</ div > ,
19
+ loader,
18
20
} )
19
21
20
22
const routeTree = rootRoute . addChildren ( [ indexRoute ] )
@@ -26,43 +28,16 @@ describe('Transitioner', () => {
26
28
} )
27
29
28
30
// Mock router.load() to verify it gets called
29
- const loadSpy = vi . spyOn ( router , 'load' ) . mockResolvedValue ( undefined )
31
+ const loadSpy = vi . spyOn ( router , 'load' )
30
32
31
- render ( ( ) => < RouterProvider router = { router } /> )
32
-
33
- // Wait for the createRenderEffect to run and call router.load()
34
- await waitFor ( ( ) => {
35
- expect ( loadSpy ) . toHaveBeenCalledTimes ( 1 )
36
- } )
37
-
38
- loadSpy . mockRestore ( )
39
- } )
40
-
41
- it ( 'should not call router.load() when on the server' , async ( ) => {
42
- const rootRoute = createRootRoute ( )
43
- const indexRoute = createRoute ( {
44
- getParentRoute : ( ) => rootRoute ,
45
- path : '/' ,
46
- component : ( ) => < div > Index</ div > ,
47
- } )
48
-
49
- const routeTree = rootRoute . addChildren ( [ indexRoute ] )
50
- const router = createRouter ( {
51
- routeTree,
52
- history : createMemoryHistory ( {
53
- initialEntries : [ '/' ] ,
54
- } ) ,
55
- isServer : true ,
56
- } )
57
-
58
- // Mock router.load() to verify it gets called
59
- const loadSpy = vi . spyOn ( router , 'load' ) . mockResolvedValue ( undefined )
33
+ await router . load ( )
60
34
61
35
render ( ( ) => < RouterProvider router = { router } /> )
62
36
63
37
// Wait for the createRenderEffect to run and call router.load()
64
38
await waitFor ( ( ) => {
65
- expect ( loadSpy ) . toHaveBeenCalledTimes ( 0 )
39
+ expect ( loadSpy ) . toHaveBeenCalledTimes ( 2 )
40
+ expect ( loader ) . toHaveBeenCalledTimes ( 1 )
66
41
} )
67
42
68
43
loadSpy . mockRestore ( )
0 commit comments