Fix another test

apollographql · Jan 15, 2025 · d9cfe5d · d9cfe5d
1 parent fb15c99
commit d9cfe5d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/src/__tests__/dataMasking.ts b/src/__tests__/dataMasking.ts
@@ -1892,7 +1892,15 @@ describe("client.watchQuery", () => {
 
     const updateQuery: Parameters<typeof observable.updateQuery>[0] = jest.fn(
       (previousResult) => {
-        return { user: { ...previousResult.user, name: "User (updated)" } };
+        return {
+          user: {
+            __typename: "User",
+            id: 1,
+            age: 30,
+            ...previousResult?.user,
+            name: "User (updated)",
+          },
+        };
       }
     );
 

diff --git a/src/react/hooks/__tests__/useBackgroundQuery.test.tsx b/src/react/hooks/__tests__/useBackgroundQuery.test.tsx
@@ -8380,7 +8380,7 @@ describe.skip("type tests", () => {
       subscribeToMore({
         document: subscription,
         updateQuery: (queryData, { subscriptionData }) => {
-          expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData>();
+          expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData | undefined>();
           expectTypeOf(queryData).not.toEqualTypeOf<MaskedVariablesCaseData>();
 
           expectTypeOf(
@@ -8412,7 +8412,7 @@ describe.skip("type tests", () => {
       subscribeToMore({
         document: subscription,
         updateQuery: (queryData, { subscriptionData }) => {
-          expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData>();
+          expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData | undefined>();
           expectTypeOf(queryData).not.toEqualTypeOf<MaskedVariablesCaseData>();
 
           expectTypeOf(

diff --git a/src/react/hooks/__tests__/useSuspenseQuery.test.tsx b/src/react/hooks/__tests__/useSuspenseQuery.test.tsx
@@ -12905,7 +12905,7 @@ describe("useSuspenseQuery", () => {
         subscribeToMore({
           document: subscription,
           updateQuery: (queryData, { subscriptionData }) => {
-            expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData>();
+            expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData | undefined>();
             expectTypeOf(
               queryData
             ).not.toEqualTypeOf<MaskedVariablesCaseData>();
@@ -12941,7 +12941,7 @@ describe("useSuspenseQuery", () => {
         subscribeToMore({
           document: subscription,
           updateQuery: (queryData, { subscriptionData }) => {
-            expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData>();
+            expectTypeOf(queryData).toEqualTypeOf<UnmaskedVariablesCaseData | undefined>();
             expectTypeOf(
               queryData
             ).not.toEqualTypeOf<MaskedVariablesCaseData>();