(feat) - allow passing operationContext to useQuery

src/hooks/useQuery.spec.ts

@@ -66,6 +66,27 @@ describe('useQuery', () => {
     });
   });
 
+  it('should set fetching to true and run effect on first mount', () => {
+    const context = { url: 'test' };
+    renderHook(
+      ({ query, variables }) => useQuery({ query, variables, context }),
+      { initialProps: { query: mockQuery, variables: mockVariables } }
+    );
+
+    expect(client.executeQuery).toBeCalledWith(
+      {
+        key: expect.any(Number),
+        query: expect.any(Object),
+        variables: mockVariables,
+      },
+      {
+        meta: { source: 'TestHook' },
+        requestPolicy: undefined,
+        url: 'test',
+      }
+    );
+  });
+
   it('should execute the subscription', async () => {
     const { waitForNextUpdate } = renderHook(
       ({ query, variables }) => useQuery({ query, variables }),

src/hooks/useQuery.ts

@@ -13,6 +13,7 @@ export interface UseQueryArgs<V> {
   query: string | DocumentNode;
   variables?: V;
   requestPolicy?: RequestPolicy;
+  context?: Partial<OperationContext>;
   pause?: boolean;
 }
 
@@ -55,6 +56,7 @@ export const useQuery = <T = any, V = object>(
       [unsubscribe.current] = pipe(
         client.executeQuery(request, {
           requestPolicy: args.requestPolicy,
+          ...args.context,
           ...opts,
           ...devtoolsContext,
         }),
@@ -63,7 +65,14 @@ export const useQuery = <T = any, V = object>(
         })
       );
     },
-    [args.requestPolicy, client, devtoolsContext, request, setState]
+    [
+      args.context,
+      args.requestPolicy,
+      client,
+      devtoolsContext,
+      request,
+      setState,
+    ]
   );
 
   useImmediateEffect(() => {