Skip to content

Commit 5227bc3

Browse files
committed
test(query-core): improve test for QueriesObserver early return notification
1 parent f860b8d commit 5227bc3

File tree

1 file changed

+16
-10
lines changed

1 file changed

+16
-10
lines changed

packages/query-core/src/__tests__/queriesObserver.test.tsx

Lines changed: 16 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -365,27 +365,33 @@ describe('queriesObserver', () => {
365365
const results: Array<Array<QueryObserverResult>> = []
366366
results.push(observer.getCurrentResult())
367367

368-
const unsubscribe = observer.subscribe((result) => {
368+
const onUpdate = vi.fn((result: Array<QueryObserverResult>) => {
369369
results.push(result)
370370
})
371+
const unsubscribe = observer.subscribe(onUpdate)
372+
const baseline = results.length
371373

372374
observer.setQueries([
373-
{ queryKey: key1, queryFn: queryFn1, staleTime: Infinity },
374-
{ queryKey: key2, queryFn: queryFn2, staleTime: Infinity },
375+
{
376+
queryKey: key1,
377+
queryFn: queryFn1,
378+
select: (d: any) => d + 100,
379+
},
380+
{
381+
queryKey: key2,
382+
queryFn: queryFn2,
383+
select: (d: any) => d + 100,
384+
},
375385
])
376386

377387
await vi.advanceTimersByTimeAsync(0)
378388

379389
unsubscribe()
380390

381-
expect(results.length).toBeGreaterThanOrEqual(2)
382-
expect(results[0]).toMatchObject([
383-
{ status: 'success', data: 1 },
384-
{ status: 'success', data: 2 },
385-
])
391+
expect(results.length).toBeGreaterThan(baseline)
386392
expect(results[results.length - 1]).toMatchObject([
387-
{ status: 'success', data: 1 },
388-
{ status: 'success', data: 2 },
393+
{ status: 'success', data: 101 },
394+
{ status: 'success', data: 102 },
389395
])
390396
})
391397
})

0 commit comments

Comments
 (0)