@tanstack/query-core 5.59.17 → 5.60.5

package/src/__tests__/queryCache.test.tsx

@@ -1,350 +0,0 @@
-import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest'
-import { waitFor } from '@testing-library/react'
-import { QueryCache, QueryClient, QueryObserver } from '..'
-import { createQueryClient, queryKey, sleep } from './utils'
-
-describe('queryCache', () => {
-  let queryClient: QueryClient
-  let queryCache: QueryCache
-
-  beforeEach(() => {
-    queryClient = createQueryClient()
-    queryCache = queryClient.getQueryCache()
-  })
-
-  afterEach(() => {
-    queryClient.clear()
-  })
-
-  describe('subscribe', () => {
-    test('should pass the correct query', async () => {
-      const key = queryKey()
-      const subscriber = vi.fn()
-      const unsubscribe = queryCache.subscribe(subscriber)
-      queryClient.setQueryData(key, 'foo')
-      const query = queryCache.find({ queryKey: key })
-      await sleep(1)
-      expect(subscriber).toHaveBeenCalledWith({ query, type: 'added' })
-      unsubscribe()
-    })
-
-    test('should notify listeners when new query is added', async () => {
-      const key = queryKey()
-      const callback = vi.fn()
-      queryCache.subscribe(callback)
-      queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data' })
-      await sleep(100)
-      expect(callback).toHaveBeenCalled()
-    })
-
-    test('should notify query cache when a query becomes stale', async () => {
-      const key = queryKey()
-      const events: Array<string> = []
-      const queries: Array<unknown> = []
-      const unsubscribe = queryCache.subscribe((event) => {
-        events.push(event.type)
-        queries.push(event.query)
-      })
-
-      const observer = new QueryObserver(queryClient, {
-        queryKey: key,
-        queryFn: () => 'data',
-        staleTime: 10,
-      })
-
-      const unsubScribeObserver = observer.subscribe(vi.fn())
-
-      await waitFor(() => {
-        expect(events.length).toBe(8)
-      })
-
-      expect(events).toEqual([
-        'added', // 1. Query added -> loading
-        'observerResultsUpdated', // 2. Observer result updated -> loading
-        'observerAdded', // 3. Observer added
-        'observerResultsUpdated', // 4. Observer result updated -> fetching
-        'updated', // 5. Query updated -> fetching
-        'observerResultsUpdated', // 6. Observer result updated -> success
-        'updated', // 7. Query updated -> success
-        'observerResultsUpdated', // 8. Observer result updated -> stale
-      ])
-
-      queries.forEach((query) => {
-        expect(query).toBeDefined()
-      })
-
-      unsubscribe()
-      unsubScribeObserver()
-    })
-
-    test('should include the queryCache and query when notifying listeners', async () => {
-      const key = queryKey()
-      const callback = vi.fn()
-      queryCache.subscribe(callback)
-      queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data' })
-      const query = queryCache.find({ queryKey: key })
-      await sleep(100)
-      expect(callback).toHaveBeenCalledWith({ query, type: 'added' })
-    })
-
-    test('should notify subscribers when new query with initialData is added', async () => {
-      const key = queryKey()
-      const callback = vi.fn()
-      queryCache.subscribe(callback)
-      queryClient.prefetchQuery({
-        queryKey: key,
-        queryFn: () => 'data',
-        initialData: 'initial',
-      })
-      await sleep(100)
-      expect(callback).toHaveBeenCalled()
-    })
-
-    test('should be able to limit cache size', async () => {
-      const testCache = new QueryCache()
-
-      const unsubscribe = testCache.subscribe((event) => {
-        if (event.type === 'added') {
-          if (testCache.getAll().length > 2) {
-            testCache
-              .findAll({
-                type: 'inactive',
-                predicate: (q) => q !== event.query,
-              })
-              .forEach((query) => {
-                testCache.remove(query)
-              })
-          }
-        }
-      })
-
-      const testClient = new QueryClient({ queryCache: testCache })
-
-      await testClient.prefetchQuery({
-        queryKey: ['key1'],
-        queryFn: () => 'data1',
-      })
-      expect(testCache.findAll().length).toBe(1)
-      await testClient.prefetchQuery({
-        queryKey: ['key2'],
-        queryFn: () => 'data2',
-      })
-      expect(testCache.findAll().length).toBe(2)
-      await testClient.prefetchQuery({
-        queryKey: ['key3'],
-        queryFn: () => 'data3',
-      })
-      expect(testCache.findAll().length).toBe(1)
-      expect(testCache.findAll()[0]!.state.data).toBe('data3')
-
-      unsubscribe()
-    })
-  })
-
-  describe('find', () => {
-    test('find should filter correctly', async () => {
-      const key = queryKey()
-      await queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data1' })
-      const query = queryCache.find({ queryKey: key })!
-      expect(query).toBeDefined()
-    })
-
-    test('find should filter correctly with exact set to false', async () => {
-      const key = queryKey()
-      await queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data1' })
-      const query = queryCache.find({ queryKey: key, exact: false })!
-      expect(query).toBeDefined()
-    })
-  })
-
-  describe('findAll', () => {
-    test('should filter correctly', async () => {
-      const key1 = queryKey()
-      const key2 = queryKey()
-      const keyFetching = queryKey()
-      await queryClient.prefetchQuery({
-        queryKey: key1,
-        queryFn: () => 'data1',
-      })
-      await queryClient.prefetchQuery({
-        queryKey: key2,
-        queryFn: () => 'data2',
-      })
-      await queryClient.prefetchQuery({
-        queryKey: [{ a: 'a', b: 'b' }],
-        queryFn: () => 'data3',
-      })
-      await queryClient.prefetchQuery({
-        queryKey: ['posts', 1],
-        queryFn: () => 'data4',
-      })
-      queryClient.invalidateQueries({ queryKey: key2 })
-      const query1 = queryCache.find({ queryKey: key1 })!
-      const query2 = queryCache.find({ queryKey: key2 })!
-      const query3 = queryCache.find({ queryKey: [{ a: 'a', b: 'b' }] })!
-      const query4 = queryCache.find({ queryKey: ['posts', 1] })!
-
-      expect(queryCache.findAll({ queryKey: key1 })).toEqual([query1])
-      // wrapping in an extra array doesn't yield the same results anymore since v4 because keys need to be an array
-      expect(queryCache.findAll({ queryKey: [key1] })).toEqual([])
-      expect(queryCache.findAll()).toEqual([query1, query2, query3, query4])
-      expect(queryCache.findAll({})).toEqual([query1, query2, query3, query4])
-      expect(queryCache.findAll({ queryKey: key1, type: 'inactive' })).toEqual([
-        query1,
-      ])
-      expect(queryCache.findAll({ queryKey: key1, type: 'active' })).toEqual([])
-      expect(queryCache.findAll({ queryKey: key1, stale: true })).toEqual([])
-      expect(queryCache.findAll({ queryKey: key1, stale: false })).toEqual([
-        query1,
-      ])
-      expect(
-        queryCache.findAll({ queryKey: key1, stale: false, type: 'active' }),
-      ).toEqual([])
-      expect(
-        queryCache.findAll({ queryKey: key1, stale: false, type: 'inactive' }),
-      ).toEqual([query1])
-      expect(
-        queryCache.findAll({
-          queryKey: key1,
-          stale: false,
-          type: 'inactive',
-          exact: true,
-        }),
-      ).toEqual([query1])
-
-      expect(queryCache.findAll({ queryKey: key2 })).toEqual([query2])
-      expect(queryCache.findAll({ queryKey: key2, stale: undefined })).toEqual([
-        query2,
-      ])
-      expect(queryCache.findAll({ queryKey: key2, stale: true })).toEqual([
-        query2,
-      ])
-      expect(queryCache.findAll({ queryKey: key2, stale: false })).toEqual([])
-      expect(queryCache.findAll({ queryKey: [{ b: 'b' }] })).toEqual([query3])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a' }], exact: false }),
-      ).toEqual([query3])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a' }], exact: true }),
-      ).toEqual([])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a', b: 'b' }], exact: true }),
-      ).toEqual([query3])
-      expect(queryCache.findAll({ queryKey: [{ a: 'a', b: 'b' }] })).toEqual([
-        query3,
-      ])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a', b: 'b', c: 'c' }] }),
-      ).toEqual([])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a' }], stale: false }),
-      ).toEqual([query3])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a' }], stale: true }),
-      ).toEqual([])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a' }], type: 'active' }),
-      ).toEqual([])
-      expect(
-        queryCache.findAll({ queryKey: [{ a: 'a' }], type: 'inactive' }),
-      ).toEqual([query3])
-      expect(
-        queryCache.findAll({ predicate: (query) => query === query3 }),
-      ).toEqual([query3])
-      expect(queryCache.findAll({ queryKey: ['posts'] })).toEqual([query4])
-
-      expect(queryCache.findAll({ fetchStatus: 'idle' })).toEqual([
-        query1,
-        query2,
-        query3,
-        query4,
-      ])
-      expect(
-        queryCache.findAll({ queryKey: key2, fetchStatus: undefined }),
-      ).toEqual([query2])
-
-      const promise = queryClient.prefetchQuery({
-        queryKey: keyFetching,
-        queryFn: async () => {
-          await sleep(20)
-          return 'dataFetching'
-        },
-      })
-      expect(queryCache.findAll({ fetchStatus: 'fetching' })).toEqual([
-        queryCache.find({ queryKey: keyFetching }),
-      ])
-      await promise
-      expect(queryCache.findAll({ fetchStatus: 'fetching' })).toEqual([])
-    })
-
-    test('should return all the queries when no filters are defined', async () => {
-      const key1 = queryKey()
-      const key2 = queryKey()
-      await queryClient.prefetchQuery({
-        queryKey: key1,
-        queryFn: () => 'data1',
-      })
-      await queryClient.prefetchQuery({
-        queryKey: key2,
-        queryFn: () => 'data2',
-      })
-      expect(queryCache.findAll().length).toBe(2)
-    })
-  })
-
-  describe('QueryCacheConfig error callbacks', () => {
-    test('should call onError and onSettled when a query errors', async () => {
-      const key = queryKey()
-      const onSuccess = vi.fn()
-      const onSettled = vi.fn()
-      const onError = vi.fn()
-      const testCache = new QueryCache({ onSuccess, onError, onSettled })
-      const testClient = createQueryClient({ queryCache: testCache })
-      await testClient.prefetchQuery({
-        queryKey: key,
-        queryFn: () => Promise.reject<unknown>('error'),
-      })
-      const query = testCache.find({ queryKey: key })
-      expect(onError).toHaveBeenCalledWith('error', query)
-      expect(onError).toHaveBeenCalledTimes(1)
-      expect(onSuccess).not.toHaveBeenCalled()
-      expect(onSettled).toHaveBeenCalledTimes(1)
-      expect(onSettled).toHaveBeenCalledWith(undefined, 'error', query)
-    })
-  })
-
-  describe('QueryCacheConfig success callbacks', () => {
-    test('should call onSuccess and onSettled when a query is successful', async () => {
-      const key = queryKey()
-      const onSuccess = vi.fn()
-      const onSettled = vi.fn()
-      const onError = vi.fn()
-      const testCache = new QueryCache({ onSuccess, onError, onSettled })
-      const testClient = createQueryClient({ queryCache: testCache })
-      await testClient.prefetchQuery({
-        queryKey: key,
-        queryFn: () => Promise.resolve({ data: 5 }),
-      })
-      const query = testCache.find({ queryKey: key })
-      expect(onSuccess).toHaveBeenCalledWith({ data: 5 }, query)
-      expect(onSuccess).toHaveBeenCalledTimes(1)
-      expect(onError).not.toHaveBeenCalled()
-      expect(onSettled).toHaveBeenCalledTimes(1)
-      expect(onSettled).toHaveBeenCalledWith({ data: 5 }, null, query)
-    })
-  })
-
-  describe('QueryCache.add', () => {
-    test('should not try to add a query already added to the cache', async () => {
-      const key = queryKey()
-
-      await queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data1' })
-
-      const query = queryCache.findAll()[0]!
-      const queryClone = Object.assign({}, query)
-
-      queryCache.add(queryClone)
-      expect(queryCache.getAll().length).toEqual(1)
-    })
-  })
-})

package/src/__tests__/queryClient.test-d.tsx

@@ -1,156 +0,0 @@
-import { describe, expectTypeOf, it } from 'vitest'
-import { QueryClient } from '../queryClient'
-import type { DataTag, InfiniteData, QueryKey } from '../types'
-
-describe('getQueryData', () => {
-  it('should be typed if key is tagged', () => {
-    const queryKey = ['key'] as DataTag<Array<string>, number>
-    const queryClient = new QueryClient()
-    const data = queryClient.getQueryData(queryKey)
-
-    expectTypeOf(data).toEqualTypeOf<number | undefined>()
-  })
-
-  it('should infer unknown if key is not tagged', () => {
-    const queryKey = ['key'] as const
-    const queryClient = new QueryClient()
-    const data = queryClient.getQueryData(queryKey)
-
-    expectTypeOf(data).toEqualTypeOf<unknown>()
-  })
-
-  it('should infer passed generic if passed', () => {
-    const queryKey = ['key'] as const
-    const queryClient = new QueryClient()
-    const data = queryClient.getQueryData<number>(queryKey)
-
-    expectTypeOf(data).toEqualTypeOf<number | undefined>()
-  })
-
-  it('should only allow Arrays to be passed', () => {
-    const queryKey = 'key'
-    const queryClient = new QueryClient()
-    // @ts-expect-error TS2345: Argument of type 'string' is not assignable to parameter of type 'QueryKey'
-    return queryClient.getQueryData(queryKey)
-  })
-})
-
-describe('setQueryData', () => {
-  it('updater should be typed if key is tagged', () => {
-    const queryKey = ['key'] as DataTag<Array<string>, number>
-    const queryClient = new QueryClient()
-    const data = queryClient.setQueryData(queryKey, (prev) => {
-      expectTypeOf(prev).toEqualTypeOf<number | undefined>()
-      return prev
-    })
-
-    expectTypeOf(data).toEqualTypeOf<number | undefined>()
-  })
-
-  it('value should be typed if key is tagged', () => {
-    const queryKey = ['key'] as DataTag<Array<string>, number>
-    const queryClient = new QueryClient()
-
-    // @ts-expect-error value should be a number
-    queryClient.setQueryData(queryKey, '1')
-
-    // @ts-expect-error value should be a number
-    queryClient.setQueryData(queryKey, () => '1')
-
-    const data = queryClient.setQueryData(queryKey, 1)
-
-    expectTypeOf(data).toEqualTypeOf<number | undefined>()
-  })
-
-  it('should infer unknown for updater if key is not tagged', () => {
-    const queryKey = ['key'] as const
-    const queryClient = new QueryClient()
-    const data = queryClient.setQueryData(queryKey, (prev) => {
-      expectTypeOf(prev).toEqualTypeOf<unknown>()
-      return prev
-    })
-
-    expectTypeOf(data).toEqualTypeOf<unknown>()
-  })
-
-  it('should infer unknown for value if key is not tagged', () => {
-    const queryKey = ['key'] as const
-    const queryClient = new QueryClient()
-    const data = queryClient.setQueryData(queryKey, 'foo')
-
-    expectTypeOf(data).toEqualTypeOf<unknown>()
-  })
-
-  it('should infer passed generic if passed', () => {
-    const queryKey = ['key'] as const
-    const queryClient = new QueryClient()
-    const data = queryClient.setQueryData<string>(queryKey, (prev) => {
-      expectTypeOf(prev).toEqualTypeOf<string | undefined>()
-      return prev
-    })
-
-    expectTypeOf(data).toEqualTypeOf<string | undefined>()
-  })
-
-  it('should infer passed generic for value', () => {
-    const queryKey = ['key'] as const
-    const queryClient = new QueryClient()
-    const data = queryClient.setQueryData<string>(queryKey, 'foo')
-
-    expectTypeOf(data).toEqualTypeOf<string | undefined>()
-  })
-})
-
-describe('fetchInfiniteQuery', () => {
-  it('should allow passing pages', async () => {
-    const data = await new QueryClient().fetchInfiniteQuery({
-      queryKey: ['key'],
-      queryFn: () => Promise.resolve('string'),
-      getNextPageParam: () => 1,
-      initialPageParam: 1,
-      pages: 5,
-    })
-
-    expectTypeOf(data).toEqualTypeOf<InfiniteData<string, number>>()
-  })
-
-  it('should not allow passing getNextPageParam without pages', () => {
-    new QueryClient().fetchInfiniteQuery({
-      queryKey: ['key'],
-      queryFn: () => Promise.resolve('string'),
-      initialPageParam: 1,
-      getNextPageParam: () => 1,
-    })
-  })
-
-  it('should not allow passing pages without getNextPageParam', () => {
-    // @ts-expect-error Property 'getNextPageParam' is missing
-    return new QueryClient().fetchInfiniteQuery({
-      queryKey: ['key'],
-      queryFn: () => Promise.resolve('string'),
-      initialPageParam: 1,
-      pages: 5,
-    })
-  })
-})
-
-describe('defaultOptions', () => {
-  it('should have a typed QueryFunctionContext', () => {
-    new QueryClient({
-      defaultOptions: {
-        queries: {
-          queryFn: (context) => {
-            expectTypeOf(context).toEqualTypeOf<{
-              queryKey: QueryKey
-              meta: Record<string, unknown> | undefined
-              signal: AbortSignal
-              pageParam?: unknown
-              direction?: unknown
-            }>()
-            return Promise.resolve('data')
-          },
-        },
-      },
-    })
-  })
-})