@@ -7,14 +7,17 @@ import { Upload } from '../routes/upload'
77vi . mock ( '@tanstack/react-router' , ( ) => ( {
88 createFileRoute : ( ) => ( config : { component : unknown } ) => config ,
99 useNavigate : ( ) => vi . fn ( ) ,
10+ useSearch : ( ) => ( { updateSlug : undefined } ) ,
1011} ) )
1112
1213const generateUploadUrl = vi . fn ( )
1314const publishVersion = vi . fn ( )
1415const fetchMock = vi . fn ( )
16+ const useQueryMock = vi . fn ( )
1517
1618vi . mock ( 'convex/react' , ( ) => ( {
1719 useConvexAuth : ( ) => ( { isAuthenticated : true } ) ,
20+ useQuery : ( ...args : unknown [ ] ) => useQueryMock ( ...args ) ,
1821 useMutation : ( ) => generateUploadUrl ,
1922 useAction : ( ) => publishVersion ,
2023} ) )
@@ -24,6 +27,11 @@ describe('Upload route', () => {
2427 generateUploadUrl . mockReset ( )
2528 publishVersion . mockReset ( )
2629 fetchMock . mockReset ( )
30+ useQueryMock . mockReset ( )
31+ useQueryMock . mockImplementation ( ( _fn : unknown , args : unknown ) => {
32+ if ( args === 'skip' ) return undefined
33+ return null
34+ } )
2735 fetchMock . mockResolvedValue ( {
2836 ok : true ,
2937 json : async ( ) => ( { storageId : 'storage-id' } ) ,
0 commit comments