@@ -7,10 +7,14 @@ import { KEY_IS_NOT_STRING_ERROR, StatusEnum } from '../constants'
77const TEST_KEY = 'test'
88const PROMISE_TIMEOUT = 5
99const fakePromise = ( ) =>
10- new Promise ( resolve => { setTimeout ( ( ) => resolve ( true ) , PROMISE_TIMEOUT ) } )
10+ new Promise ( resolve => {
11+ setTimeout ( ( ) => resolve ( true ) , PROMISE_TIMEOUT )
12+ } )
1113
1214const fakeNullPromise = ( ) =>
13- new Promise ( resolve => { setTimeout ( ( ) => resolve ( null ) , PROMISE_TIMEOUT ) } )
15+ new Promise ( resolve => {
16+ setTimeout ( ( ) => resolve ( null ) , PROMISE_TIMEOUT )
17+ } )
1418
1519const wrapper = ( { children } : { children : ReactNode } ) => (
1620 < DataLoaderProvider > { children } </ DataLoaderProvider >
@@ -108,7 +112,11 @@ describe('DataLoaderProvider', () => {
108112 wrapper,
109113 } )
110114 act ( ( ) => {
111- result . current . addRequest ( TEST_KEY , {
115+ result . current . getOrAddRequest ( TEST_KEY , {
116+ enabled : true ,
117+ method,
118+ } )
119+ result . current . getOrAddRequest ( TEST_KEY , {
112120 enabled : true ,
113121 method,
114122 } )
@@ -125,6 +133,8 @@ describe('DataLoaderProvider', () => {
125133 expect ( result . current . getCachedData ( TEST_KEY ) ) . toBe ( undefined )
126134 expect ( result . current . getCachedData ( ) ) . toStrictEqual ( { test : undefined } )
127135 expect ( result . current . getRequest ( TEST_KEY ) ) . toBeDefined ( )
136+ const unknownReload = result . current . getReloads ( 'unknown' )
137+ expect ( unknownReload ) . toBeUndefined ( )
128138 await act ( async ( ) => {
129139 await reloads . test ( )
130140 } )
@@ -141,20 +151,15 @@ describe('DataLoaderProvider', () => {
141151 expect ( result . current . getCachedData ( TEST_KEY ) ) . toBeUndefined ( )
142152 act ( ( ) => result . current . clearAllCachedData ( ) )
143153 expect ( result . current . getCachedData ( ) ) . toStrictEqual ( { test : undefined } )
144- await waitFor ( ( ) =>
145- expect ( result . current . getRequest ( TEST_KEY ) ) . toBeUndefined ( ) ,
146- )
147154
148155 try {
149156 // @ts -expect-error Should throw an error
150157 result . current . clearCachedData ( 3 )
151158 } catch ( e ) {
152159 expect ( ( e as Error ) . message ) . toBe ( KEY_IS_NOT_STRING_ERROR )
153160 }
154- expect ( result . current . getReloads ( ) ) . not . toHaveProperty ( TEST_KEY )
155- expect ( result . current . getReloads ( TEST_KEY ) ) . not . toBeDefined ( )
156161 expect ( result . current . getCachedData ( TEST_KEY ) ) . toBe ( undefined )
157- expect ( result . current . getCachedData ( ) ) . not . toStrictEqual ( {
162+ expect ( result . current . getCachedData ( ) ) . toStrictEqual ( {
158163 test : undefined ,
159164 } )
160165 try {
@@ -184,7 +189,10 @@ describe('DataLoaderProvider', () => {
184189
185190 test ( 'should delay max concurrent request' , async ( ) => {
186191 const method = jest . fn (
187- ( ) => new Promise ( resolve => { setTimeout ( ( ) => resolve ( true ) , 100 ) } ) ,
192+ ( ) =>
193+ new Promise ( resolve => {
194+ setTimeout ( ( ) => resolve ( true ) , 100 )
195+ } ) ,
188196 )
189197 const { result, waitFor } = renderHook ( useDataLoaderContext , {
190198 wrapper : wrapperWith2ConcurrentRequests ,
0 commit comments