Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 37 additions & 1 deletion packages/query-core/src/__tests__/queriesObserver.test.tsx
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest'
import { queryKey, sleep } from '@tanstack/query-test-utils'
import { QueriesObserver, QueryClient } from '..'
import { QueriesObserver, QueryClient, QueryObserver } from '..'
import type { QueryObserverResult } from '..'

describe('queriesObserver', () => {
Expand Down Expand Up @@ -510,4 +510,40 @@ describe('queriesObserver', () => {

expect(newCombined.keys).toEqual(['pending'])
})

test('should track properties on all observers when trackResult is called', () => {
const key1 = queryKey()
const key2 = queryKey()
const queryFn1 = () => 'data1'
const queryFn2 = () => 'data2'

const observer = new QueriesObserver(queryClient, [
{ queryKey: key1, queryFn: queryFn1 },
{ queryKey: key2, queryFn: queryFn2 },
])

const trackPropSpy = vi.spyOn(QueryObserver.prototype, 'trackProp')

const [, , trackResult] = observer.getOptimisticResult(
[
{ queryKey: key1, queryFn: queryFn1 },
{ queryKey: key2, queryFn: queryFn2 },
],
undefined,
)

const trackedResults = trackResult()

expect(trackedResults).toHaveLength(2)

// Accessing a property on the first result should trigger trackProp on all observers
void trackedResults[0]!.status

// 1 direct call from the accessed observer's proxy +
// 2 synchronized calls from onPropTracked callback (one per observer)
expect(trackPropSpy).toHaveBeenCalledWith('status')
expect(trackPropSpy).toHaveBeenCalledTimes(3)

trackPropSpy.mockRestore()
})
})
Loading