@@ -870,25 +870,25 @@ describe('query', () => {
870870
871871 const key = queryKey ( )
872872
873- const queryFn = jest . fn <
874- Promise < unknown > ,
875- [ QueryFunctionContext < ReturnType < typeof queryKey > > ]
876- > ( )
877-
878- queryFn . mockImplementation ( ( { signal } ) => {
879- return new Promise ( ( resolve , reject ) => {
880- const abortListener = ( ) => {
881- clearTimeout ( timerId )
882- reject ( signal ! . reason )
883- }
884- signal ! . addEventListener ( 'abort' , abortListener )
885-
886- const timerId = setTimeout ( ( ) => {
887- signal ! . removeEventListener ( 'abort' , abortListener )
888- resolve ( mockedData . join ( ' - ' ) )
889- } , 50 )
873+ const queryFn = vi
874+ . fn <
875+ [ QueryFunctionContext < ReturnType < typeof queryKey > > ] ,
876+ Promise < unknown >
877+ > ( )
878+ . mockImplementation ( ( { signal } ) => {
879+ return new Promise ( ( resolve , reject ) => {
880+ const abortListener = ( ) => {
881+ clearTimeout ( timerId )
882+ reject ( signal . reason )
883+ }
884+ signal . addEventListener ( 'abort' , abortListener )
885+
886+ const timerId = setTimeout ( ( ) => {
887+ signal . removeEventListener ( 'abort' , abortListener )
888+ resolve ( mockedData . join ( ' - ' ) )
889+ } , 50 )
890+ } )
890891 } )
891- } )
892892
893893 const observer = new QueryObserver ( queryClient , {
894894 queryKey : key ,
@@ -899,9 +899,9 @@ describe('query', () => {
899899
900900 mockedData = [ 1 , 2 ] // update "server" state in the background
901901
902- queryClient . invalidateQueries ( key )
902+ queryClient . invalidateQueries ( { queryKey : key } )
903903 await sleep ( 1 )
904- queryClient . invalidateQueries ( key )
904+ queryClient . invalidateQueries ( { queryKey : key } )
905905 await sleep ( 1 )
906906 unsubscribe ( ) // unsubscribe to simulate unmount
907907
@@ -911,7 +911,7 @@ describe('query', () => {
911911 queryFn,
912912 } )
913913
914- const spy = jest . fn ( )
914+ const spy = vi . fn ( )
915915 newObserver . subscribe ( ( { data } ) => spy ( data ) )
916916 await sleep ( 60 ) // let it resolve
917917 expect ( spy ) . toHaveBeenCalledWith ( '1 - 2' )
0 commit comments