queryClient.test.tsx

import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
import { queryKey, sleep } from '@tanstack/query-test-utils'
import {
  CancelledError,
  MutationObserver,
  QueryClient,
  QueryObserver,
  dehydrate,
  focusManager,
  hydrate,
  onlineManager,
  skipToken,
} from '..'
import { mockOnlineManagerIsOnline } from './utils'
import type { QueryCache, QueryFunction, QueryObserverOptions } from '..'

describe('queryClient', () => {
  let queryClient: QueryClient
  let queryCache: QueryCache

  beforeEach(() => {
    vi.useFakeTimers()
    queryClient = new QueryClient()
    queryCache = queryClient.getQueryCache()
    queryClient.mount()
  })

  afterEach(() => {
    queryClient.clear()
    queryClient.unmount()
    vi.useRealTimers()
  })

  describe('defaultOptions', () => {
    it('should merge defaultOptions', () => {
      const key = queryKey()

      const queryFn = () => 'data'
      const testClient = new QueryClient({
        defaultOptions: { queries: { queryFn } },
      })

      expect(() => testClient.prefetchQuery({ queryKey: key })).not.toThrow()
    })

    it('should merge defaultOptions when query is added to cache', async () => {
      const key = queryKey()

      const testClient = new QueryClient({
        defaultOptions: {
          queries: { gcTime: Infinity },
        },
      })

      const fetchData = () => Promise.resolve('data')
      await testClient.prefetchQuery({ queryKey: key, queryFn: fetchData })
      const newQuery = testClient.getQueryCache().find({ queryKey: key })
      expect(newQuery?.options.gcTime).toBe(Infinity)
    })

    it('should get defaultOptions', () => {
      const queryFn = () => 'data'
      const defaultOptions = { queries: { queryFn } }
      const testClient = new QueryClient({
        defaultOptions,
      })
      expect(testClient.getDefaultOptions()).toMatchObject(defaultOptions)
    })
  })

  describe('setQueryDefaults', () => {
    it('should not trigger a fetch', () => {
      const key = queryKey()
      queryClient.setQueryDefaults(key, { queryFn: () => 'data' })
      const data = queryClient.getQueryData(key)
      expect(data).toBeUndefined()
    })

    it('should be able to override defaults', async () => {
      const key = queryKey()
      queryClient.setQueryDefaults(key, { queryFn: () => 'data' })
      const observer = new QueryObserver(queryClient, { queryKey: key })
      const { data } = await observer.refetch()
      expect(data).toBe('data')
    })

    it('should match the query key partially', async () => {
      const key = queryKey()
      queryClient.setQueryDefaults([key], { queryFn: () => 'data' })
      const observer = new QueryObserver(queryClient, {
        queryKey: [key, 'a'],
      })
      const { data } = await observer.refetch()
      expect(data).toBe('data')
    })

    it('should not match if the query key is a subset', async () => {
      const key = queryKey()
      queryClient.setQueryDefaults([key, 'a'], {
        queryFn: () => 'data',
      })
      const observer = new QueryObserver(queryClient, {
        queryKey: [key],
        retry: false,
        enabled: false,
      })
      const { status } = await observer.refetch()
      expect(status).toBe('error')
    })

    it('should also set defaults for observers', () => {
      const key = queryKey()
      queryClient.setQueryDefaults(key, {
        queryFn: () => 'data',
        enabled: false,
      })
      const observer = new QueryObserver(queryClient, {
        queryKey: [key],
      })
      expect(observer.getCurrentResult().status).toBe('pending')
      expect(observer.getCurrentResult().fetchStatus).toBe('idle')
    })

    it('should update existing query defaults', () => {
      const key = queryKey()
      const queryOptions1 = { queryFn: () => 'data' }
      const queryOptions2 = { retry: false }
      queryClient.setQueryDefaults(key, { ...queryOptions1 })
      queryClient.setQueryDefaults(key, { ...queryOptions2 })
      expect(queryClient.getQueryDefaults(key)).toMatchObject(queryOptions2)
    })

    it('should merge defaultOptions', () => {
      const key = queryKey()

      queryClient.setQueryDefaults([...key, 'todo'], { suspense: true })
      queryClient.setQueryDefaults([...key, 'todo', 'detail'], {
        staleTime: 5000,
      })

      expect(
        queryClient.getQueryDefaults([...key, 'todo', 'detail']),
      ).toMatchObject({ suspense: true, staleTime: 5000 })
    })
  })

  describe('defaultQueryOptions', () => {
    it('should default networkMode when persister is present', () => {
      expect(
        new QueryClient({
          defaultOptions: {
            queries: {
              persister: 'ignore' as any,
            },
          },
        }).defaultQueryOptions({ queryKey: queryKey() }).networkMode,
      ).toBe('offlineFirst')
    })

    it('should not default networkMode without persister', () => {
      expect(
        new QueryClient({
          defaultOptions: {
            queries: {
              staleTime: 1000,
            },
          },
        }).defaultQueryOptions({ queryKey: queryKey() }).networkMode,
      ).toBe(undefined)
    })

    it('should not default networkMode when already present', () => {
      expect(
        new QueryClient({
          defaultOptions: {
            queries: {
              persister: 'ignore' as any,
              networkMode: 'always',
            },
          },
        }).defaultQueryOptions({ queryKey: queryKey() }).networkMode,
      ).toBe('always')
    })
  })

  describe('setQueryData', () => {
    it('should not crash if query could not be found', () => {
      const key = queryKey()
      const user = { userId: 1 }
      expect(() => {
        queryClient.setQueryData([key, user], (prevUser?: typeof user) => ({
          ...prevUser!,
          name: 'James',
        }))
      }).not.toThrow()
    })

    it('should not crash when variable is null', () => {
      const key = queryKey()
      queryClient.setQueryData([key, { userId: null }], 'Old Data')
      expect(() => {
        queryClient.setQueryData([key, { userId: null }], 'New Data')
      }).not.toThrow()
    })

    it('should use default options', () => {
      const key = queryKey()
      const testClient = new QueryClient({
        defaultOptions: { queries: { queryKeyHashFn: () => 'someKey' } },
      })
      const testCache = testClient.getQueryCache()
      testClient.setQueryData(key, 'data')
      expect(testClient.getQueryData(key)).toBe('data')
      expect(testCache.find({ queryKey: key })).toBe(testCache.get('someKey'))
    })

    it('should create a new query if query was not found 1', () => {
      const key = queryKey()
      queryClient.setQueryData(key, 'bar')
      expect(queryClient.getQueryData(key)).toBe('bar')
    })

    it('should create a new query if query was not found 2', () => {
      const key = queryKey()
      queryClient.setQueryData(key, 'qux')
      expect(queryClient.getQueryData(key)).toBe('qux')
    })

    it('should not create a new query if query was not found and data is undefined', () => {
      const key = queryKey()
      expect(queryClient.getQueryCache().find({ queryKey: key })).toBe(
        undefined,
      )
      queryClient.setQueryData(key, undefined)
      expect(queryClient.getQueryCache().find({ queryKey: key })).toBe(
        undefined,
      )
    })

    it('should not create a new query if query was not found and updater returns undefined', () => {
      const key = queryKey()
      expect(queryClient.getQueryCache().find({ queryKey: key })).toBe(
        undefined,
      )
      queryClient.setQueryData(key, () => undefined)
      expect(queryClient.getQueryCache().find({ queryKey: key })).toBe(
        undefined,
      )
    })

    it('should not update query data if data is undefined', () => {
      const key = queryKey()
      queryClient.setQueryData(key, 'qux')
      queryClient.setQueryData(key, undefined)
      expect(queryClient.getQueryData(key)).toBe('qux')
    })

    it('should not update query data if updater returns undefined', () => {
      const key = queryKey()
      queryClient.setQueryData<string>(key, 'qux')
      queryClient.setQueryData<string>(key, () => undefined)
      expect(queryClient.getQueryData(key)).toBe('qux')
    })

    it('should accept an update function', () => {
      const key = queryKey()

      const updater = vi.fn((oldData) => `new data + ${oldData}`)

      queryClient.setQueryData(key, 'test data')
      queryClient.setQueryData(key, updater)

      expect(updater).toHaveBeenCalled()
      expect(queryCache.find({ queryKey: key })!.state.data).toEqual(
        'new data + test data',
      )
    })

    it('should set the new data without comparison if structuralSharing is set to false', () => {
      const key = queryKey()

      queryClient.setDefaultOptions({
        queries: {
          structuralSharing: false,
        },
      })

      const oldData = { value: true }
      const newData = { value: true }
      queryClient.setQueryData(key, oldData)
      queryClient.setQueryData(key, newData)

      expect(queryCache.find({ queryKey: key })!.state.data).toBe(newData)
    })

    it('should apply a custom structuralSharing function when provided', () => {
      const key = queryKey()

      const queryObserverOptions = {
        structuralSharing: (
          prevData: { value: Date } | undefined,
          newData: { value: Date },
        ) => {
          if (!prevData) {
            return newData
          }
          return newData.value.getTime() === prevData.value.getTime()
            ? prevData
            : newData
        },
      } as QueryObserverOptions

      queryClient.setDefaultOptions({ queries: queryObserverOptions })

      const oldData = { value: new Date(2022, 6, 19) }
      const newData = { value: new Date(2022, 6, 19) }
      queryClient.setQueryData(key, oldData)
      queryClient.setQueryData(key, newData)

      expect(queryCache.find({ queryKey: key })!.state.data).toBe(oldData)

      const distinctData = { value: new Date(2021, 11, 25) }
      queryClient.setQueryData(key, distinctData)

      expect(queryCache.find({ queryKey: key })!.state.data).toBe(distinctData)
    })

    it('should not set isFetching to false', async () => {
      const key = queryKey()
      queryClient.prefetchQuery({
        queryKey: key,
        queryFn: () => sleep(10).then(() => 23),
      })
      expect(queryClient.getQueryState(key)).toMatchObject({
        data: undefined,
        fetchStatus: 'fetching',
      })
      queryClient.setQueryData(key, 42)
      expect(queryClient.getQueryState(key)).toMatchObject({
        data: 42,
        fetchStatus: 'fetching',
      })
      await vi.advanceTimersByTimeAsync(10)
      expect(queryClient.getQueryState(key)).toMatchObject({
        data: 23,
        fetchStatus: 'idle',
      })
    })
  })

  describe('setQueriesData', () => {
    it('should update all existing, matching queries', () => {
      queryClient.setQueryData(['key', 1], 1)
      queryClient.setQueryData(['key', 2], 2)

      const result = queryClient.setQueriesData<number>(
        { queryKey: ['key'] },
        (old) => (old ? old + 5 : undefined),
      )

      expect(result).toEqual([
        [['key', 1], 6],
        [['key', 2], 7],
      ])
      expect(queryClient.getQueryData(['key', 1])).toBe(6)
      expect(queryClient.getQueryData(['key', 2])).toBe(7)
    })

    it('should accept queryFilters', () => {
      queryClient.setQueryData(['key', 1], 1)
      queryClient.setQueryData(['key', 2], 2)
      const query1 = queryCache.find({ queryKey: ['key', 1] })!

      const result = queryClient.setQueriesData<number>(
        { predicate: (query) => query === query1 },
        (old) => old! + 5,
      )

      expect(result).toEqual([[['key', 1], 6]])
      expect(queryClient.getQueryData(['key', 1])).toBe(6)
      expect(queryClient.getQueryData(['key', 2])).toBe(2)
    })

    it('should not update non existing queries', () => {
      const result = queryClient.setQueriesData<string>(
        { queryKey: ['key'] },
        'data',
      )

      expect(result).toEqual([])
      expect(queryClient.getQueryData(['key'])).toBe(undefined)
    })
  })

  describe('isFetching', () => {
    it('should return length of fetching queries', async () => {
      expect(queryClient.isFetching()).toBe(0)
      queryClient.prefetchQuery({
        queryKey: queryKey(),
        queryFn: () => sleep(10).then(() => 'data'),
      })
      expect(queryClient.isFetching()).toBe(1)
      queryClient.prefetchQuery({
        queryKey: queryKey(),
        queryFn: () => sleep(5).then(() => 'data'),
      })
      expect(queryClient.isFetching()).toBe(2)
      await vi.advanceTimersByTimeAsync(5)
      expect(queryClient.isFetching()).toEqual(1)
      await vi.advanceTimersByTimeAsync(5)
      expect(queryClient.isFetching()).toEqual(0)
    })
  })

  describe('isMutating', () => {
    it('should return length of mutating', async () => {
      expect(queryClient.isMutating()).toBe(0)
      new MutationObserver(queryClient, {
        mutationFn: () => sleep(10).then(() => 'data'),
      }).mutate()
      expect(queryClient.isMutating()).toBe(1)
      new MutationObserver(queryClient, {
        mutationFn: () => sleep(5).then(() => 'data'),
      }).mutate()
      expect(queryClient.isMutating()).toBe(2)
      await vi.advanceTimersByTimeAsync(5)
      expect(queryClient.isMutating()).toEqual(1)
      await vi.advanceTimersByTimeAsync(5)
      expect(queryClient.isMutating()).toEqual(0)
    })
  })

  describe('getQueryData', () => {
    it('should return the query data if the query is found', () => {
      const key = queryKey()
      queryClient.setQueryData([key, 'id'], 'bar')
      expect(queryClient.getQueryData([key, 'id'])).toBe('bar')
    })

    it('should return undefined if the query is not found', () => {
      const key = queryKey()
      expect(queryClient.getQueryData(key)).toBeUndefined()
    })

    it('should match exact by default', () => {
      const key = queryKey()
      queryClient.setQueryData([key, 'id'], 'bar')
      expect(queryClient.getQueryData([key])).toBeUndefined()
    })
  })

  describe('ensureQueryData', () => {
    it('should return the cached query data if the query is found', async () => {
      const key = queryKey()
      const queryFn = () => Promise.resolve('data')

      queryClient.setQueryData([key, 'id'], 'bar')

      await expect(
        queryClient.ensureQueryData({ queryKey: [key, 'id'], queryFn }),
      ).resolves.toEqual('bar')
    })

    it('should return the cached query data if the query is found and cached query data is falsy', async () => {
      const key = queryKey()
      const queryFn = () => Promise.resolve(0)

      queryClient.setQueryData([key, 'id'], null)

      await expect(
        queryClient.ensureQueryData({ queryKey: [key, 'id'], queryFn }),
      ).resolves.toEqual(null)
    })

    it('should call fetchQuery and return its results if the query is not found', async () => {
      const key = queryKey()
      const queryFn = () => Promise.resolve('data')

      await expect(
        queryClient.ensureQueryData({ queryKey: [key], queryFn }),
      ).resolves.toEqual('data')
    })

    it('should return the cached query data if the query is found and preFetchQuery in the background when revalidateIfStale is set', async () => {
      const TIMEOUT = 10
      const key = queryKey()
      queryClient.setQueryData([key, 'id'], 'old')

      const queryFn = () =>
        new Promise((resolve) => {
          setTimeout(() => resolve('new'), TIMEOUT)
        })

      await expect(
        queryClient.ensureQueryData({
          queryKey: [key, 'id'],
          queryFn,
          revalidateIfStale: true,
        }),
      ).resolves.toEqual('old')
      await vi.advanceTimersByTimeAsync(TIMEOUT + 10)
      await expect(
        queryClient.ensureQueryData({
          queryKey: [key, 'id'],
          queryFn,
          revalidateIfStale: true,
        }),
      ).resolves.toEqual('new')
    })

    it('should not fetch with initialDat', async () => {
      const key = queryKey()
      const queryFn = vi.fn().mockImplementation(() => Promise.resolve('data'))

      await expect(
        queryClient.ensureQueryData({
          queryKey: [key, 'id'],
          queryFn,
          initialData: 'initial',
        }),
      ).resolves.toEqual('initial')

      expect(queryFn).toHaveBeenCalledTimes(0)
    })
  })

  describe('ensureInfiniteQueryData', () => {
    it('should return the cached query data if the query is found', async () => {
      const key = queryKey()
      const queryFn = () => Promise.resolve('data')

      queryClient.setQueryData([key, 'id'], { pages: ['bar'], pageParams: [0] })

      await expect(
        queryClient.ensureInfiniteQueryData({
          queryKey: [key, 'id'],
          queryFn,
          initialPageParam: 1,
          getNextPageParam: () => undefined,
        }),
      ).resolves.toEqual({ pages: ['bar'], pageParams: [0] })
    })

    it('should fetch the query and return its results if the query is not found', async () => {
      const key = queryKey()
      const queryFn = () => Promise.resolve('data')

      await expect(
        queryClient.ensureInfiniteQueryData({
          queryKey: [key, 'id'],
          queryFn,
          initialPageParam: 1,
          getNextPageParam: () => undefined,
        }),
      ).resolves.toEqual({ pages: ['data'], pageParams: [1] })
    })

    it('should return the cached query data if the query is found and preFetchQuery in the background when revalidateIfStale is set', async () => {
      const TIMEOUT = 10
      const key = queryKey()
      queryClient.setQueryData([key, 'id'], { pages: ['old'], pageParams: [0] })

      const queryFn = () => sleep(TIMEOUT).then(() => 'new')

      await expect(
        queryClient.ensureInfiniteQueryData({
          queryKey: [key, 'id'],
          queryFn,
          initialPageParam: 1,
          getNextPageParam: () => undefined,
          revalidateIfStale: true,
        }),
      ).resolves.toEqual({ pages: ['old'], pageParams: [0] })
      await vi.advanceTimersByTimeAsync(TIMEOUT + 10)
      await expect(
        queryClient.ensureInfiniteQueryData({
          queryKey: [key, 'id'],
          queryFn,
          initialPageParam: 1,
          getNextPageParam: () => undefined,
          revalidateIfStale: true,
        }),
      ).resolves.toEqual({ pages: ['new'], pageParams: [0] })
    })
  })

  describe('getQueriesData', () => {
    it('should return the query data for all matched queries', () => {
      const key1 = queryKey()
      const key2 = queryKey()
      queryClient.setQueryData([key1, 1], 1)
      queryClient.setQueryData([key1, 2], 2)
      queryClient.setQueryData([key2, 2], 2)
      expect(queryClient.getQueriesData({ queryKey: [key1] })).toEqual([
        [[key1, 1], 1],
        [[key1, 2], 2],
      ])
    })

    it('should return empty array if queries are not found', () => {
      const key = queryKey()
      expect(queryClient.getQueriesData({ queryKey: key })).toEqual([])
    })

    it('should accept query filters', () => {
      queryClient.setQueryData(['key', 1], 1)
      queryClient.setQueryData(['key', 2], 2)
      const query1 = queryCache.find({ queryKey: ['key', 1] })!

      const result = queryClient.getQueriesData({
        predicate: (query) => query === query1,
      })

      expect(result).toEqual([[['key', 1], 1]])
    })
  })

  describe('fetchQuery', () => {
    it('should not type-error with strict query key', async () => {
      type StrictData = 'data'
      type StrictQueryKey = ['strict', ...ReturnType<typeof queryKey>]
      const key: StrictQueryKey = ['strict', ...queryKey()]

      const fetchFn: QueryFunction<StrictData, StrictQueryKey> = () =>
        Promise.resolve('data')

      await expect(
        queryClient.fetchQuery<StrictData, any, StrictData, StrictQueryKey>({
          queryKey: key,
          queryFn: fetchFn,
        }),
      ).resolves.toEqual('data')
    })

    // https://github.com/tannerlinsley/react-query/issues/652
    it('should not retry by default', async () => {
      const key = queryKey()

      await expect(
        queryClient.fetchQuery({
          queryKey: key,
          queryFn: (): Promise<unknown> => {
            throw new Error('error')
          },
        }),
      ).rejects.toEqual(new Error('error'))
    })

    it('should return the cached data on cache hit', async () => {
      const key = queryKey()

      const fetchFn = () => Promise.resolve('data')
      const first = await queryClient.fetchQuery({
        queryKey: key,
        queryFn: fetchFn,
      })
      const second = await queryClient.fetchQuery({
        queryKey: key,
        queryFn: fetchFn,
      })

      expect(second).toBe(first)
    })

    it('should read from cache with static staleTime even if invalidated', async () => {
      const key = queryKey()

      const fetchFn = vi.fn(() => Promise.resolve({ data: 'data' }))
      const first = await queryClient.fetchQuery({
        queryKey: key,
        queryFn: fetchFn,
        staleTime: 'static',
      })

      expect(first.data).toBe('data')
      expect(fetchFn).toHaveBeenCalledTimes(1)

      await queryClient.invalidateQueries({
        queryKey: key,
        refetchType: 'none',
      })

      const second = await queryClient.fetchQuery({
        queryKey: key,
        queryFn: fetchFn,
        staleTime: 'static',
      })

      expect(fetchFn).toHaveBeenCalledTimes(1)

      expect(second).toBe(first)
    })

    it('should be able to fetch when garbage collection time is set to 0 and then be removed', async () => {
      const key1 = queryKey()
      const promise = queryClient.fetchQuery({
        queryKey: key1,
        queryFn: () => sleep(10).then(() => 1),
        gcTime: 0,
      })
      await vi.advanceTimersByTimeAsync(10)
      await expect(promise).resolves.toEqual(1)
      await vi.advanceTimersByTimeAsync(1)
      expect(queryClient.getQueryData(key1)).toEqual(undefined)
    })

    it('should keep a query in cache if garbage collection time is Infinity', async () => {
      const key1 = queryKey()
      const promise = queryClient.fetchQuery({
        queryKey: key1,
        queryFn: () => sleep(10).then(() => 1),
        gcTime: Infinity,
      })
      await vi.advanceTimersByTimeAsync(10)
      const result2 = queryClient.getQueryData(key1)
      await expect(promise).resolves.toEqual(1)
      expect(result2).toEqual(1)
    })

    it('should not force fetch', async () => {
      const key = queryKey()

      queryClient.setQueryData(key, 'og')
      const fetchFn = () => Promise.resolve('new')
      const first = await queryClient.fetchQuery({
        queryKey: key,
        queryFn: fetchFn,
        initialData: 'initial',
        staleTime: 100,
      })
      expect(first).toBe('og')
    })

    it('should only fetch if the data is older then the given stale time', async () => {
      const key = queryKey()

      let count = 0
      const queryFn = () => ++count

      queryClient.setQueryData(key, count)
      const firstPromise = queryClient.fetchQuery({
        queryKey: key,
        queryFn,
        staleTime: 100,
      })
      await expect(firstPromise).resolves.toBe(0)
      await vi.advanceTimersByTimeAsync(10)
      const secondPromise = queryClient.fetchQuery({
        queryKey: key,
        queryFn,
        staleTime: 10,
      })
      await expect(secondPromise).resolves.toBe(1)
      const thirdPromise = queryClient.fetchQuery({
        queryKey: key,
        queryFn,
        staleTime: 10,
      })
      await expect(thirdPromise).resolves.toBe(1)
      await vi.advanceTimersByTimeAsync(10)
      const fourthPromise = queryClient.fetchQuery({
        queryKey: key,
        queryFn,
        staleTime: 10,
      })
      await expect(fourthPromise).resolves.toBe(2)
    })

    it('should allow new meta', async () => {
      const key = queryKey()

      const first = await queryClient.fetchQuery({
        queryKey: key,
        queryFn: ({ meta }) => Promise.resolve(meta),
        meta: {
          foo: true,
        },
      })
      expect(first).toStrictEqual({ foo: true })

      const second = await queryClient.fetchQuery({
        queryKey: key,
        queryFn: ({ meta }) => Promise.resolve(meta),
        meta: {
          foo: false,
        },
      })
      expect(second).toStrictEqual({ foo: false })
    })
  })

  describe('fetchInfiniteQuery', () => {
    it('should not type-error with strict query key', async () => {
      type StrictData = string
      type StrictQueryKey = ['strict', ...ReturnType<typeof queryKey>]
      const key: StrictQueryKey = ['strict', ...queryKey()]

      const data = {
        pages: ['data'],
        pageParams: [0],
      } as const

      const fetchFn: QueryFunction<StrictData, StrictQueryKey, number> = () =>
        Promise.resolve(data.pages[0])

      await expect(
        queryClient.fetchInfiniteQuery<
          StrictData,
          any,
          StrictData,
          StrictQueryKey,
          number
        >({ queryKey: key, queryFn: fetchFn, initialPageParam: 0 }),
      ).resolves.toEqual(data)
    })

    it('should return infinite query data', async () => {
      const key = queryKey()
      const result = await queryClient.fetchInfiniteQuery({
        queryKey: key,
        initialPageParam: 10,
        queryFn: ({ pageParam }) => Number(pageParam),
      })
      const result2 = queryClient.getQueryData(key)

      const expected = {
        pages: [10],
        pageParams: [10],
      }

      expect(result).toEqual(expected)
      expect(result2).toEqual(expected)
    })
  })

  describe('prefetchInfiniteQuery', () => {
    it('should not type-error with strict query key', async () => {
      type StrictData = 'data'
      type StrictQueryKey = ['strict', ...ReturnType<typeof queryKey>]
      const key: StrictQueryKey = ['strict', ...queryKey()]

      const fetchFn: QueryFunction<StrictData, StrictQueryKey, number> = () =>
        Promise.resolve('data')

      await queryClient.prefetchInfiniteQuery<
        StrictData,
        any,
        StrictData,
        StrictQueryKey,
        number
      >({ queryKey: key, queryFn: fetchFn, initialPageParam: 0 })

      const result = queryClient.getQueryData(key)

      expect(result).toEqual({
        pages: ['data'],
        pageParams: [0],
      })
    })

    it('should return infinite query data', async () => {
      const key = queryKey()

      await queryClient.prefetchInfiniteQuery({
        queryKey: key,
        queryFn: ({ pageParam }) => Number(pageParam),
        initialPageParam: 10,
      })

      const result = queryClient.getQueryData(key)

      expect(result).toEqual({
        pages: [10],
        pageParams: [10],
      })
    })

    it('should prefetch multiple pages', async () => {
      const key = queryKey()

      await queryClient.prefetchInfiniteQuery({
        queryKey: key,
        queryFn: ({ pageParam }) => String(pageParam),
        getNextPageParam: (_lastPage, _pages, lastPageParam) =>
          lastPageParam + 5,
        initialPageParam: 10,
        pages: 3,
      })

      const result = queryClient.getQueryData(key)

      expect(result).toEqual({
        pages: ['10', '15', '20'],
        pageParams: [10, 15, 20],
      })
    })

    it('should stop prefetching if getNextPageParam returns undefined', async () => {
      const key = queryKey()
      let count = 0

      await queryClient.prefetchInfiniteQuery({
        queryKey: key,
        queryFn: ({ pageParam }) => String(pageParam),
        getNextPageParam: (_lastPage, _pages, lastPageParam) => {
          count++
          return lastPageParam >= 20 ? undefined : lastPageParam + 5
        },
        initialPageParam: 10,
        pages: 5,
      })

      const result = queryClient.getQueryData(key)

      expect(result).toEqual({
        pages: ['10', '15', '20'],
        pageParams: [10, 15, 20],
      })

      // this check ensures we're exiting the fetch loop early
      expect(count).toBe(3)
    })
  })

  describe('prefetchQuery', () => {
    it('should not type-error with strict query key', async () => {
      type StrictData = 'data'
      type StrictQueryKey = ['strict', ...ReturnType<typeof queryKey>]
      const key: StrictQueryKey = ['strict', ...queryKey()]

      const fetchFn: QueryFunction<StrictData, StrictQueryKey> = () =>
        Promise.resolve('data')

      await queryClient.prefetchQuery<
        StrictData,
        any,
        StrictData,
        StrictQueryKey
      >({ queryKey: key, queryFn: fetchFn })

      const result = queryClient.getQueryData(key)

      expect(result).toEqual('data')
    })

    it('should return undefined when an error is thrown', async () => {
      const key = queryKey()

      const result = await queryClient.prefetchQuery({
        queryKey: key,
        queryFn: (): Promise<unknown> => {
          throw new Error('error')
        },
        retry: false,
      })

      expect(result).toBeUndefined()
    })

    it('should be garbage collected after gcTime if unused', async () => {
      const key = queryKey()

      await queryClient.prefetchQuery({
        queryKey: key,
        queryFn: () => 'data',
        gcTime: 10,
      })
      expect(queryCache.find({ queryKey: key })).toBeDefined()
      await vi.advanceTimersByTimeAsync(15)
      expect(queryCache.find({ queryKey: key })).not.toBeDefined()
    })
  })

  describe('removeQueries', () => {
    it('should not crash when exact is provided', async () => {
      const key = queryKey()

      const fetchFn = () => Promise.resolve('data')

      // check the query was added to the cache
      await queryClient.prefetchQuery({ queryKey: key, queryFn: fetchFn })
      expect(queryCache.find({ queryKey: key })).toBeTruthy()

      // check the error doesn't occur
      expect(() =>
        queryClient.removeQueries({ queryKey: key, exact: true }),
      ).not.toThrow()

      // check query was successful removed
      expect(queryCache.find({ queryKey: key })).toBeFalsy()
    })
  })

  describe('cancelQueries', () => {
    it('should revert queries to their previous state', async () => {
      const key1 = queryKey()
      queryClient.setQueryData(key1, 'data')

      const pending = queryClient.fetchQuery({
        queryKey: key1,
        queryFn: () => sleep(1000).then(() => 'data2'),
      })

      await vi.advanceTimersByTimeAsync(10)

      await queryClient.cancelQueries()

      // with previous data present, imperative fetch should resolve to that data after cancel
      await expect(pending).resolves.toBe('data')

      const state1 = queryClient.getQueryState(key1)
      expect(state1).toMatchObject({
        data: 'data',
        status: 'success',
      })
    })

    it('should not revert if revert option is set to false', async () => {
      const key1 = queryKey()
      await queryClient.fetchQuery({
        queryKey: key1,
        queryFn: () => 'data',
      })
      queryClient.prefetchQuery({
        queryKey: key1,
        queryFn: () => sleep(1000).then(() => 'data2'),
      })
      await vi.advanceTimersByTimeAsync(10)
      await queryClient.cancelQueries({ queryKey: key1 }, { revert: false })
      const state1 = queryClient.getQueryState(key1)
      expect(state1).toMatchObject({
        status: 'error',
      })
    })

    it('should throw CancelledError for imperative methods when initial fetch is cancelled', async () => {
      const key = queryKey()

      const promise = queryClient.fetchQuery({
        queryKey: key,
        queryFn: () => sleep(50).then(() => 25),
      })

      await vi.advanceTimersByTimeAsync(10)

      await queryClient.cancelQueries({ queryKey: key })

      // we have to reject here because we can't resolve with `undefined`
      // the alternative would be a never-ending promise
      await expect(promise).rejects.toBeInstanceOf(CancelledError)

      // however, the query was correctly reverted to pending state
      expect(queryClient.getQueryState(key)).toMatchObject({
        status: 'pending',
        fetchStatus: 'idle',
        data: undefined,
        error: null,
      })
    })
  })

  describe('refetchQueries', () => {
    it('should not refetch if all observers are disabled', async () => {
      const key = queryKey()
      const queryFn = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data')
      await queryClient.fetchQuery({ queryKey: key, queryFn })
      const observer1 = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn,
        enabled: false,
      })
      observer1.subscribe(() => undefined)
      await queryClient.refetchQueries()
      observer1.destroy()
      expect(queryFn).toHaveBeenCalledTimes(1)
    })
    it('should refetch if at least one observer is enabled', async () => {
      const key = queryKey()
      const queryFn = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data')
      await queryClient.fetchQuery({ queryKey: key, queryFn })
      const observer1 = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn,
        enabled: false,
      })
      const observer2 = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn,
        refetchOnMount: false,
      })
      observer1.subscribe(() => undefined)
      observer2.subscribe(() => undefined)
      await queryClient.refetchQueries()
      observer1.destroy()
      observer2.destroy()
      expect(queryFn).toHaveBeenCalledTimes(2)
    })
    it('should refetch all queries when no arguments are given', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer1 = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
        initialData: 'initial',
      })
      const observer2 = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
        initialData: 'initial',
      })
      observer1.subscribe(() => undefined)
      observer2.subscribe(() => undefined)
      await queryClient.refetchQueries()
      observer1.destroy()
      observer2.destroy()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(2)
    })

    it('should be able to refetch all fresh queries', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.refetchQueries({ type: 'active', stale: false })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should be able to refetch all stale queries', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      queryClient.invalidateQueries({ queryKey: key1 })
      await queryClient.refetchQueries({ stale: true })
      unsubscribe()
      // fetchQuery, observer mount, invalidation (cancels observer mount) and refetch
      expect(queryFn1).toHaveBeenCalledTimes(4)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should be able to refetch all stale and active queries', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      queryClient.invalidateQueries({ queryKey: key1 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.refetchQueries(
        { type: 'active', stale: true },
        { cancelRefetch: false },
      )
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should be able to refetch all active and inactive queries (queryClient.refetchQueries()', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.refetchQueries()
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(2)
    })

    it('should be able to refetch all active and inactive queries (queryClient.refetchQueries({ type: "all" }))', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.refetchQueries({ type: 'all' })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(2)
    })

    it('should be able to refetch only active queries', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.refetchQueries({ type: 'active' })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should be able to refetch only inactive queries', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.refetchQueries({ type: 'inactive' })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(1)
      expect(queryFn2).toHaveBeenCalledTimes(2)
    })

    it('should throw an error if throwOnError option is set to true', async () => {
      const key1 = queryKey()
      const queryFnError = () => Promise.reject<unknown>('error')
      try {
        await queryClient.fetchQuery({
          queryKey: key1,
          queryFn: queryFnError,
          retry: false,
        })
      } catch {}
      let error: any
      try {
        await queryClient.refetchQueries(
          { queryKey: key1 },
          { throwOnError: true },
        )
      } catch (err) {
        error = err
      }
      expect(error).toEqual('error')
    })

    it('should resolve Promise immediately if query is paused', async () => {
      const key1 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      const onlineMock = mockOnlineManagerIsOnline(false)

      await queryClient.refetchQueries({ queryKey: key1 })

      // if we reach this point, the test succeeds because the Promise was resolved immediately
      expect(queryFn1).toHaveBeenCalledTimes(1)
      onlineMock.mockRestore()
    })

    it('should refetch if query we are offline but query networkMode is always', async () => {
      const key1 = queryKey()
      queryClient.setQueryDefaults(key1, { networkMode: 'always' })
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      const onlineMock = mockOnlineManagerIsOnline(false)

      await queryClient.refetchQueries({ queryKey: key1 })

      // initial fetch + refetch (even though we are offline)
      expect(queryFn1).toHaveBeenCalledTimes(2)
      onlineMock.mockRestore()
    })

    it('should not refetch static queries', async () => {
      const key = queryKey()
      const queryFn = vi.fn(() => 'data1')
      await queryClient.fetchQuery({ queryKey: key, queryFn: queryFn })

      expect(queryFn).toHaveBeenCalledTimes(1)

      const observer = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn,
        staleTime: 'static',
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.refetchQueries()

      expect(queryFn).toHaveBeenCalledTimes(1)
      unsubscribe()
    })
  })

  describe('invalidateQueries', () => {
    it('should refetch active queries by default', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      queryClient.invalidateQueries({ queryKey: key1 })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should not refetch inactive queries by default', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        enabled: false,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      queryClient.invalidateQueries({ queryKey: key1 })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(1)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should not refetch active queries when "refetch" is "none"', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      queryClient.invalidateQueries({
        queryKey: key1,
        refetchType: 'none',
      })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(1)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should refetch inactive queries when "refetch" is "inactive"', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
        refetchOnMount: false,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      unsubscribe()

      await queryClient.invalidateQueries({
        queryKey: key1,
        refetchType: 'inactive',
      })
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(1)
    })

    it('should refetch active and inactive queries when "refetch" is "all"', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      await queryClient.fetchQuery({ queryKey: key1, queryFn: queryFn1 })
      await queryClient.fetchQuery({ queryKey: key2, queryFn: queryFn2 })
      const observer = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        staleTime: Infinity,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      queryClient.invalidateQueries({
        refetchType: 'all',
      })
      unsubscribe()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(2)
    })

    it('should not refetch disabled inactive queries even if "refetchType" is "all', async () => {
      const queryFn = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const observer = new QueryObserver(queryClient, {
        queryKey: queryKey(),
        queryFn: queryFn,
        staleTime: Infinity,
        enabled: false,
      })
      const unsubscribe = observer.subscribe(() => undefined)
      unsubscribe()
      await queryClient.invalidateQueries({
        refetchType: 'all',
      })
      expect(queryFn).toHaveBeenCalledTimes(0)
    })

    it('should not refetch inactive queries that have a skipToken queryFn even if "refetchType" is "all', async () => {
      const key = queryKey()
      const observer = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn: skipToken,
        staleTime: Infinity,
      })

      queryClient.setQueryData(key, 'data1')

      const unsubscribe = observer.subscribe(() => undefined)
      unsubscribe()

      expect(queryClient.getQueryState(key)?.dataUpdateCount).toBe(1)

      await queryClient.invalidateQueries({
        refetchType: 'all',
      })

      expect(queryClient.getQueryState(key)?.dataUpdateCount).toBe(1)
    })

    it('should cancel ongoing fetches if cancelRefetch option is set (default value)', async () => {
      const key = queryKey()
      const abortFn = vi.fn()
      let fetchCount = 0
      const observer = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn: ({ signal }) =>
          new Promise((resolve) => {
            fetchCount++
            setTimeout(() => resolve(5), 10)
            signal.addEventListener('abort', abortFn)
          }),
        initialData: 1,
      })
      observer.subscribe(() => undefined)

      queryClient.refetchQueries()
      await vi.advanceTimersByTimeAsync(10)
      observer.destroy()
      expect(abortFn).toHaveBeenCalledTimes(1)
      expect(fetchCount).toBe(2)
    })

    it('should not cancel ongoing fetches if cancelRefetch option is set to false', async () => {
      const key = queryKey()
      const abortFn = vi.fn()
      let fetchCount = 0
      const observer = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn: ({ signal }) => {
          return new Promise((resolve) => {
            fetchCount++
            setTimeout(() => resolve(5), 10)
            signal.addEventListener('abort', abortFn)
          })
        },
        initialData: 1,
      })
      observer.subscribe(() => undefined)

      queryClient.refetchQueries(undefined, { cancelRefetch: false })
      await vi.advanceTimersByTimeAsync(10)
      observer.destroy()
      expect(abortFn).toHaveBeenCalledTimes(0)
      expect(fetchCount).toBe(1)
    })

    it('should not refetch static queries after invalidation', async () => {
      const key = queryKey()
      const queryFn = vi.fn(() => 'data1')
      await queryClient.fetchQuery({ queryKey: key, queryFn: queryFn })

      expect(queryFn).toHaveBeenCalledTimes(1)

      const observer = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn,
        staleTime: 'static',
      })
      const unsubscribe = observer.subscribe(() => undefined)
      await queryClient.invalidateQueries()

      expect(queryFn).toHaveBeenCalledTimes(1)
      unsubscribe()
    })

    it('should forward meta to the invalidate action in queryCache.subscribe', async () => {
      const key = queryKey()
      await queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data' })

      const events: Array<unknown> = []
      const unsubscribe = queryCache.subscribe((event) => {
        if (event.type === 'updated' && event.action.type === 'invalidate') {
          events.push(event.action.meta)
        }
      })

      await queryClient.invalidateQueries(
        { queryKey: key },
        { meta: { source: 'websocket', traceId: 'abc123' } },
      )

      unsubscribe()
      expect(events).toHaveLength(1)
      expect(events[0]).toEqual({ source: 'websocket', traceId: 'abc123' })
    })
  })

  describe('resetQueries', () => {
    it('should notify listeners when a query is reset', async () => {
      const key = queryKey()

      const callback = vi.fn()

      await queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data' })

      queryCache.subscribe(callback)

      queryClient.resetQueries({ queryKey: key })

      expect(callback).toHaveBeenCalled()
    })

    it('should reset query', async () => {
      const key = queryKey()

      await queryClient.prefetchQuery({ queryKey: key, queryFn: () => 'data' })

      let state = queryClient.getQueryState(key)
      expect(state?.data).toEqual('data')
      expect(state?.status).toEqual('success')

      queryClient.resetQueries({ queryKey: key })

      state = queryClient.getQueryState(key)

      expect(state).toBeTruthy()
      expect(state?.data).toBeUndefined()
      expect(state?.status).toEqual('pending')
      expect(state?.fetchStatus).toEqual('idle')
    })

    it('should reset query data to initial data if set', async () => {
      const key = queryKey()

      await queryClient.prefetchQuery({
        queryKey: key,
        queryFn: () => 'data',
        initialData: 'initial',
      })

      let state = queryClient.getQueryState(key)
      expect(state?.data).toEqual('data')

      queryClient.resetQueries({ queryKey: key })

      state = queryClient.getQueryState(key)

      expect(state).toBeTruthy()
      expect(state?.data).toEqual('initial')
    })

    it('should refetch all active queries', async () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const key3 = queryKey()
      const queryFn1 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data1')
      const queryFn2 = vi
        .fn<(...args: Array<unknown>) => string>()
        .mockReturnValue('data2')
      const observer1 = new QueryObserver(queryClient, {
        queryKey: key1,
        queryFn: queryFn1,
        enabled: true,
      })
      const observer2 = new QueryObserver(queryClient, {
        queryKey: key2,
        queryFn: queryFn2,
        enabled: false,
      })
      const observer3 = new QueryObserver(queryClient, {
        queryKey: key3,
        queryFn: skipToken,
      })
      let didSkipTokenRun = false
      observer1.subscribe(() => undefined)
      observer2.subscribe(() => undefined)
      observer3.subscribe(() => (didSkipTokenRun = true))
      await queryClient.resetQueries()
      observer3.destroy()
      observer2.destroy()
      observer1.destroy()
      expect(queryFn1).toHaveBeenCalledTimes(2)
      expect(queryFn2).toHaveBeenCalledTimes(0)
      expect(didSkipTokenRun).toBe(false)
    })
  })

  describe('focusManager and onlineManager', () => {
    afterEach(() => {
      onlineManager.setOnline(true)
      focusManager.setFocused(undefined)
    })
    it('should notify queryCache and mutationCache if focused', async () => {
      const testClient = new QueryClient()
      testClient.mount()

      const queryCacheOnFocusSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onFocus',
      )
      const queryCacheOnOnlineSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onOnline',
      )
      const mutationCacheResumePausedMutationsSpy = vi.spyOn(
        testClient.getMutationCache(),
        'resumePausedMutations',
      )

      focusManager.setFocused(false)
      expect(queryCacheOnFocusSpy).not.toHaveBeenCalled()
      expect(mutationCacheResumePausedMutationsSpy).not.toHaveBeenCalled()

      focusManager.setFocused(true)
      await vi.advanceTimersByTimeAsync(0)
      expect(queryCacheOnFocusSpy).toHaveBeenCalledTimes(1)
      expect(mutationCacheResumePausedMutationsSpy).toHaveBeenCalledTimes(1)

      expect(queryCacheOnOnlineSpy).not.toHaveBeenCalled()

      queryCacheOnFocusSpy.mockRestore()
      mutationCacheResumePausedMutationsSpy.mockRestore()
      queryCacheOnOnlineSpy.mockRestore()
    })

    it('should notify queryCache and mutationCache if online', async () => {
      const testClient = new QueryClient()
      testClient.mount()

      const queryCacheOnFocusSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onFocus',
      )
      const queryCacheOnOnlineSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onOnline',
      )
      const mutationCacheResumePausedMutationsSpy = vi.spyOn(
        testClient.getMutationCache(),
        'resumePausedMutations',
      )

      onlineManager.setOnline(false)
      expect(queryCacheOnOnlineSpy).not.toHaveBeenCalled()
      expect(mutationCacheResumePausedMutationsSpy).not.toHaveBeenCalled()

      onlineManager.setOnline(true)
      await vi.advanceTimersByTimeAsync(0)
      expect(queryCacheOnOnlineSpy).toHaveBeenCalledTimes(1)

      expect(mutationCacheResumePausedMutationsSpy).toHaveBeenCalledTimes(1)

      expect(queryCacheOnFocusSpy).not.toHaveBeenCalled()

      queryCacheOnFocusSpy.mockRestore()
      queryCacheOnOnlineSpy.mockRestore()
      mutationCacheResumePausedMutationsSpy.mockRestore()
    })

    it('should resume paused mutations when coming online', async () => {
      const consoleMock = vi.spyOn(console, 'error')
      consoleMock.mockImplementation(() => undefined)
      onlineManager.setOnline(false)

      const observer1 = new MutationObserver(queryClient, {
        mutationFn: () => Promise.resolve(1),
      })

      const observer2 = new MutationObserver(queryClient, {
        mutationFn: () => Promise.resolve(2),
      })
      void observer1.mutate()
      void observer2.mutate()

      expect(observer1.getCurrentResult().isPaused).toBeTruthy()
      expect(observer2.getCurrentResult().isPaused).toBeTruthy()

      onlineManager.setOnline(true)

      await vi.advanceTimersByTimeAsync(0)
      expect(observer1.getCurrentResult().status).toBe('success')
      expect(observer2.getCurrentResult().status).toBe('success')
    })

    it('should resume paused mutations in parallel', async () => {
      onlineManager.setOnline(false)

      const orders: Array<string> = []

      const observer1 = new MutationObserver(queryClient, {
        mutationFn: async () => {
          orders.push('1start')
          await sleep(50)
          orders.push('1end')
          return 1
        },
      })

      const observer2 = new MutationObserver(queryClient, {
        mutationFn: async () => {
          orders.push('2start')
          await sleep(20)
          orders.push('2end')
          return 2
        },
      })
      void observer1.mutate()
      void observer2.mutate()

      expect(observer1.getCurrentResult().isPaused).toBeTruthy()
      expect(observer2.getCurrentResult().isPaused).toBeTruthy()

      onlineManager.setOnline(true)

      await vi.advanceTimersByTimeAsync(50)
      expect(observer1.getCurrentResult().status).toBe('success')
      expect(observer2.getCurrentResult().status).toBe('success')

      expect(orders).toEqual(['1start', '2start', '2end', '1end'])
    })

    it('should resume paused mutations one after the other when in the same scope when invoked manually at the same time', async () => {
      const consoleMock = vi.spyOn(console, 'error')
      consoleMock.mockImplementation(() => undefined)
      onlineManager.setOnline(false)

      const orders: Array<string> = []

      const observer1 = new MutationObserver(queryClient, {
        scope: {
          id: 'scope',
        },
        mutationFn: async () => {
          orders.push('1start')
          await sleep(50)
          orders.push('1end')
          return 1
        },
      })

      const observer2 = new MutationObserver(queryClient, {
        scope: {
          id: 'scope',
        },
        mutationFn: async () => {
          orders.push('2start')
          await sleep(20)
          orders.push('2end')
          return 2
        },
      })
      void observer1.mutate()
      void observer2.mutate()

      expect(observer1.getCurrentResult().isPaused).toBeTruthy()
      expect(observer2.getCurrentResult().isPaused).toBeTruthy()

      onlineManager.setOnline(true)
      void queryClient.resumePausedMutations()

      await vi.advanceTimersByTimeAsync(70)
      expect(observer1.getCurrentResult().status).toBe('success')
      expect(observer2.getCurrentResult().status).toBe('success')

      expect(orders).toEqual(['1start', '1end', '2start', '2end'])
    })

    it('should resumePausedMutations when coming online after having called resumePausedMutations while offline', async () => {
      const consoleMock = vi.spyOn(console, 'error')
      consoleMock.mockImplementation(() => undefined)
      onlineManager.setOnline(false)

      const observer = new MutationObserver(queryClient, {
        mutationFn: () => Promise.resolve(1),
      })

      void observer.mutate()

      expect(observer.getCurrentResult().isPaused).toBeTruthy()

      await queryClient.resumePausedMutations()

      // still paused because we are still offline
      expect(observer.getCurrentResult().isPaused).toBeTruthy()

      onlineManager.setOnline(true)

      await vi.advanceTimersByTimeAsync(0)
      expect(observer.getCurrentResult().status).toBe('success')
    })

    it('should resumePausedMutations when coming online after having restored cache (and resumed) while offline', async () => {
      const consoleMock = vi.spyOn(console, 'error')
      consoleMock.mockImplementation(() => undefined)
      onlineManager.setOnline(false)

      const observer = new MutationObserver(queryClient, {
        mutationFn: () => Promise.resolve(1),
      })

      void observer.mutate()

      expect(observer.getCurrentResult().isPaused).toBeTruthy()

      const state = dehydrate(queryClient)

      const newQueryClient = new QueryClient({
        defaultOptions: {
          mutations: {
            mutationFn: () => Promise.resolve(1),
          },
        },
      })

      newQueryClient.mount()

      hydrate(newQueryClient, state)

      // still paused because we are still offline
      expect(
        newQueryClient.getMutationCache().getAll()[0]?.state.isPaused,
      ).toBeTruthy()

      await newQueryClient.resumePausedMutations()

      onlineManager.setOnline(true)

      await vi.advanceTimersByTimeAsync(0)
      expect(newQueryClient.getMutationCache().getAll()[0]?.state.status).toBe(
        'success',
      )

      newQueryClient.unmount()
    })

    it('should notify queryCache after resumePausedMutations has finished when coming online', async () => {
      const key = queryKey()

      let count = 0
      const results: Array<string> = []

      const queryObserver = new QueryObserver(queryClient, {
        queryKey: key,
        queryFn: async () => {
          count++
          results.push('data' + count)
          await sleep(10)
          return 'data' + count
        },
      })

      const unsubscribe = queryObserver.subscribe(() => undefined)

      await vi.advanceTimersByTimeAsync(10)
      expect(queryClient.getQueryData(key)).toBe('data1')

      onlineManager.setOnline(false)

      const observer = new MutationObserver(queryClient, {
        mutationFn: async () => {
          results.push('mutation1-start')
          await sleep(50)
          results.push('mutation1-end')
          return 1
        },
      })

      void observer.mutate()

      const observer2 = new MutationObserver(queryClient, {
        scope: {
          id: 'scope',
        },
        mutationFn: async () => {
          results.push('mutation2-start')
          await sleep(50)
          results.push('mutation2-end')
          return 2
        },
      })

      void observer2.mutate()

      const observer3 = new MutationObserver(queryClient, {
        scope: {
          id: 'scope',
        },
        mutationFn: async () => {
          results.push('mutation3-start')
          await sleep(50)
          results.push('mutation3-end')
          return 3
        },
      })

      void observer3.mutate()

      expect(observer.getCurrentResult().isPaused).toBeTruthy()
      expect(observer2.getCurrentResult().isPaused).toBeTruthy()
      expect(observer3.getCurrentResult().isPaused).toBeTruthy()
      onlineManager.setOnline(true)

      await vi.advanceTimersByTimeAsync(110)
      expect(queryClient.getQueryData(key)).toBe('data2')

      // refetch from coming online should happen after mutations have finished
      expect(results).toStrictEqual([
        'data1',
        'mutation1-start',
        'mutation2-start',
        'mutation1-end',
        'mutation2-end',
        'mutation3-start', // 3 starts after 2 because they are in the same scope
        'mutation3-end',
        'data2',
      ])

      unsubscribe()
    })

    it('should notify queryCache and mutationCache after multiple mounts and single unmount', async () => {
      const testClient = new QueryClient()
      testClient.mount()
      testClient.mount()
      testClient.unmount()

      const queryCacheOnFocusSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onFocus',
      )
      const queryCacheOnOnlineSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onOnline',
      )
      const mutationCacheResumePausedMutationsSpy = vi.spyOn(
        testClient.getMutationCache(),
        'resumePausedMutations',
      )

      onlineManager.setOnline(false)
      onlineManager.setOnline(true)
      await vi.advanceTimersByTimeAsync(0)
      expect(queryCacheOnOnlineSpy).toHaveBeenCalledTimes(1)
      expect(mutationCacheResumePausedMutationsSpy).toHaveBeenCalledTimes(1)

      focusManager.setFocused(true)
      await vi.advanceTimersByTimeAsync(0)
      expect(queryCacheOnFocusSpy).toHaveBeenCalledTimes(1)
      expect(mutationCacheResumePausedMutationsSpy).toHaveBeenCalledTimes(2)

      queryCacheOnFocusSpy.mockRestore()
      queryCacheOnOnlineSpy.mockRestore()
      mutationCacheResumePausedMutationsSpy.mockRestore()
      focusManager.setFocused(undefined)
      onlineManager.setOnline(true)
    })

    it('should not notify queryCache and mutationCache after multiple mounts/unmounts', () => {
      const testClient = new QueryClient()
      testClient.mount()
      testClient.mount()
      testClient.unmount()
      testClient.unmount()

      const queryCacheOnFocusSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onFocus',
      )
      const queryCacheOnOnlineSpy = vi.spyOn(
        testClient.getQueryCache(),
        'onOnline',
      )
      const mutationCacheResumePausedMutationsSpy = vi.spyOn(
        testClient.getMutationCache(),
        'resumePausedMutations',
      )

      onlineManager.setOnline(true)
      expect(queryCacheOnOnlineSpy).not.toHaveBeenCalled()
      expect(mutationCacheResumePausedMutationsSpy).not.toHaveBeenCalled()

      focusManager.setFocused(true)
      expect(queryCacheOnFocusSpy).not.toHaveBeenCalled()
      expect(mutationCacheResumePausedMutationsSpy).not.toHaveBeenCalled()

      queryCacheOnFocusSpy.mockRestore()
      queryCacheOnOnlineSpy.mockRestore()
      mutationCacheResumePausedMutationsSpy.mockRestore()
      focusManager.setFocused(undefined)
      onlineManager.setOnline(true)
    })
  })

  describe('setMutationDefaults', () => {
    it('should update existing mutation defaults', () => {
      const key = queryKey()
      const mutationOptions1 = { mutationFn: () => Promise.resolve('data') }
      const mutationOptions2 = { retry: false }
      queryClient.setMutationDefaults(key, mutationOptions1)
      queryClient.setMutationDefaults(key, mutationOptions2)
      expect(queryClient.getMutationDefaults(key)).toMatchObject(
        mutationOptions2,
      )
    })

    it('should return only matching defaults when multiple mutation defaults are set', () => {
      const key1 = queryKey()
      const key2 = queryKey()
      const mutationOptions1 = { retry: 1 }
      const mutationOptions2 = { retry: 2 }
      queryClient.setMutationDefaults(key1, mutationOptions1)
      queryClient.setMutationDefaults(key2, mutationOptions2)

      expect(queryClient.getMutationDefaults(key1)).toMatchObject(
        mutationOptions1,
      )
      expect(queryClient.getMutationDefaults(key2)).toMatchObject(
        mutationOptions2,
      )
    })
  })
})