forked from TanStack/query
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(solid-query): Revise
queryOptions
implementation (TanStack#7366)
- Loading branch information
Showing
3 changed files
with
179 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
147 changes: 147 additions & 0 deletions
147
packages/solid-query/src/__tests__/queryOptions.test-d.tsx
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,147 @@ | ||
import { describe, expect, expectTypeOf, it } from 'vitest' | ||
import { QueryClient, dataTagSymbol, skipToken } from '@tanstack/query-core' | ||
import { createQuery, queryOptions } from '../createQuery' | ||
|
||
describe('queryOptions', () => { | ||
it('should not allow excess properties', () => { | ||
queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
// @ts-expect-error this is a good error, because stallTime does not exist! | ||
stallTime: 1000, | ||
}) | ||
}) | ||
it('should infer types for callbacks', () => { | ||
queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
staleTime: 1000, | ||
select: (data) => { | ||
expectTypeOf(data).toEqualTypeOf<number>() | ||
}, | ||
}) | ||
}) | ||
it('should work when passed to createQuery', () => { | ||
const options = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
}) | ||
|
||
const { data } = createQuery(() => options) | ||
expectTypeOf(data).toEqualTypeOf<number | undefined>() | ||
}) | ||
it('should work when passed to fetchQuery', async () => { | ||
const options = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
}) | ||
|
||
const data = await new QueryClient().fetchQuery(options) | ||
expectTypeOf(data).toEqualTypeOf<number>() | ||
}) | ||
it('should tag the queryKey with the result type of the QueryFn', () => { | ||
expect(() => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
}) | ||
|
||
expectTypeOf(queryKey[dataTagSymbol]).toEqualTypeOf<number>() | ||
}) | ||
}) | ||
it('should tag the queryKey even if no promise is returned', () => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => 5, | ||
}) | ||
|
||
expectTypeOf(queryKey[dataTagSymbol]).toEqualTypeOf<number>() | ||
}) | ||
it('should tag the queryKey with unknown if there is no queryFn', () => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
}) | ||
|
||
expectTypeOf(queryKey[dataTagSymbol]).toEqualTypeOf<unknown>() | ||
}) | ||
it('should tag the queryKey with the result type of the QueryFn if select is used', () => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
select: (data) => data.toString(), | ||
}) | ||
|
||
expectTypeOf(queryKey[dataTagSymbol]).toEqualTypeOf<number>() | ||
}) | ||
it('should return the proper type when passed to getQueryData', () => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
}) | ||
|
||
const queryClient = new QueryClient() | ||
const data = queryClient.getQueryData(queryKey) | ||
expectTypeOf(data).toEqualTypeOf<number | undefined>() | ||
}) | ||
it('should return the proper type when passed to getQueryState', () => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
}) | ||
|
||
const queryClient = new QueryClient() | ||
const state = queryClient.getQueryState(queryKey) | ||
expectTypeOf(state?.data).toEqualTypeOf<number | undefined>() | ||
}) | ||
it('should properly type updaterFn when passed to setQueryData', () => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
}) | ||
|
||
const queryClient = new QueryClient() | ||
const data = queryClient.setQueryData(queryKey, (prev) => { | ||
expectTypeOf(prev).toEqualTypeOf<number | undefined>() | ||
return prev | ||
}) | ||
expectTypeOf(data).toEqualTypeOf<number | undefined>() | ||
}) | ||
it('should properly type value when passed to setQueryData', () => { | ||
const { queryKey } = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: () => Promise.resolve(5), | ||
}) | ||
|
||
const queryClient = new QueryClient() | ||
|
||
// @ts-expect-error value should be a number | ||
queryClient.setQueryData(queryKey, '5') | ||
// @ts-expect-error value should be a number | ||
queryClient.setQueryData(queryKey, () => '5') | ||
|
||
const data = queryClient.setQueryData(queryKey, 5) | ||
expectTypeOf(data).toEqualTypeOf<number | undefined>() | ||
}) | ||
|
||
it('should infer even if there is a conditional skipToken', () => { | ||
const options = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: Math.random() > 0.5 ? skipToken : () => Promise.resolve(5), | ||
}) | ||
|
||
const queryClient = new QueryClient() | ||
const data = queryClient.getQueryData(options.queryKey) | ||
expectTypeOf(data).toEqualTypeOf<number | undefined>() | ||
}) | ||
|
||
it('should infer to unknown if we disable a query with just a skipToken', () => { | ||
const options = queryOptions({ | ||
queryKey: ['key'], | ||
queryFn: skipToken, | ||
}) | ||
|
||
const queryClient = new QueryClient() | ||
const data = queryClient.getQueryData(options.queryKey) | ||
expectTypeOf(data).toEqualTypeOf<unknown>() | ||
}) | ||
}) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters