test(lit-query): cover error brand and callback updater for infiniteQueryOptions

Author: lazerg

packages/lit-query/src/tests/type-inference.test.ts

@@ -1,4 +1,5 @@
 import {
+  dataTagErrorSymbol,
   dataTagSymbol,
   QueryClient,
   type DefinedQueryObserverResult,
@@ -217,6 +218,9 @@ describe('type inference', () => {
     expectTypeOf(infiniteQueryOpts.queryKey[dataTagSymbol]).toEqualTypeOf<
       InfiniteData<{ page: number }>
     >()
+    expectTypeOf(
+      infiniteQueryOpts.queryKey[dataTagErrorSymbol],
+    ).toEqualTypeOf<Error>()
     const cachedPages = client.getQueryData(infiniteQueryOpts.queryKey)
     expectTypeOf(cachedPages).toEqualTypeOf<
       InfiniteData<{ page: number }> | undefined
@@ -228,5 +232,17 @@ describe('type inference', () => {
     expectTypeOf(updatedPages).toEqualTypeOf<
       InfiniteData<{ page: number }> | undefined
     >()
+    const updatedPagesViaCallback = client.setQueryData(
+      infiniteQueryOpts.queryKey,
+      (previous) => {
+        expectTypeOf(previous).toEqualTypeOf<
+          InfiniteData<{ page: number }> | undefined
+        >()
+        return previous
+      },
+    )
+    expectTypeOf(updatedPagesViaCallback).toEqualTypeOf<
+      InfiniteData<{ page: number }> | undefined
+    >()
   })
 })