Update failing policies tests

apollographql · Feb 5, 2025 · 6519a65 · 6519a65
1 parent 7c6a9d7
commit 6519a65
Showing 1 changed file with 33 additions and 11 deletions.
diff --git a/src/cache/inmemory/__tests__/policies.ts b/src/cache/inmemory/__tests__/policies.ts
@@ -3647,14 +3647,16 @@ describe("type policies", function () {
 
       let result = await client.query({ query: firstQuery });
 
-      expect(result).toEqual({
+      expect(result).toEqualApolloQueryResult({
         loading: false,
         networkStatus: NetworkStatus.ready,
         data: {
           todos: {
             totalCount: 1292,
           },
         },
+        complete: true,
+        partial: false,
       });
 
       expect(cache.extract()).toEqual({
@@ -3678,7 +3680,7 @@ describe("type policies", function () {
         variables: secondVariables,
       });
 
-      expect(result).toEqual({
+      expect(result).toEqualApolloQueryResult({
         loading: false,
         networkStatus: NetworkStatus.ready,
         data: {
@@ -3688,12 +3690,14 @@ describe("type policies", function () {
             totalCount: 1292,
           },
         },
+        complete: true,
+        partial: false,
       });
 
       expect(cache.extract()).toMatchSnapshot();
 
       result = await client.query({ query: thirdQuery });
-      expect(result).toEqual({
+      expect(result).toEqualApolloQueryResult({
         loading: false,
         networkStatus: NetworkStatus.ready,
         data: {
@@ -3702,6 +3706,8 @@ describe("type policies", function () {
             extraMetaData: "extra",
           },
         },
+        complete: true,
+        partial: false,
       });
       expect(cache.extract()).toMatchSnapshot();
     });
@@ -4108,7 +4114,7 @@ describe("type policies", function () {
 
       const stream = new ObservableStream(observable);
 
-      await expect(stream).toEmitValue({
+      await expect(stream).toEmitApolloQueryResult({
         loading: false,
         networkStatus: NetworkStatus.ready,
         data: {
@@ -4118,6 +4124,8 @@ describe("type policies", function () {
             totalCount: 1292,
           },
         },
+        complete: true,
+        partial: false,
       });
       expect(cache.extract()).toMatchSnapshot();
 
@@ -4126,7 +4134,7 @@ describe("type policies", function () {
       {
         const result = await stream.takeNext();
 
-        expect(result).toEqual({
+        expect(result).toEqualApolloQueryResult({
           loading: false,
           networkStatus: NetworkStatus.ready,
           data: {
@@ -4142,6 +4150,8 @@ describe("type policies", function () {
               totalCount: 1292,
             },
           },
+          complete: true,
+          partial: false,
         });
         expect(cache.extract()).toMatchSnapshot();
       }
@@ -4153,7 +4163,7 @@ describe("type policies", function () {
 
         expect(result.data.search.edges.length).toBe(5);
 
-        expect(result).toEqual({
+        expect(result).toEqualApolloQueryResult({
           loading: false,
           networkStatus: NetworkStatus.ready,
           data: {
@@ -4169,6 +4179,8 @@ describe("type policies", function () {
               totalCount: 1292,
             },
           },
+          complete: true,
+          partial: false,
         });
 
         expect(cache.extract()).toMatchSnapshot();
@@ -4179,7 +4191,7 @@ describe("type policies", function () {
       {
         const result = await stream.takeNext();
 
-        expect(result).toEqual({
+        expect(result).toEqualApolloQueryResult({
           loading: false,
           networkStatus: NetworkStatus.ready,
           data: {
@@ -4195,6 +4207,8 @@ describe("type policies", function () {
               totalCount: 1292,
             },
           },
+          complete: true,
+          partial: false,
         });
 
         expect(result.data.search.edges).toEqual([
@@ -4212,7 +4226,7 @@ describe("type policies", function () {
 
         expect(result.data.search.edges.length).toBe(7);
 
-        expect(result).toEqual({
+        expect(result).toEqualApolloQueryResult({
           loading: false,
           networkStatus: NetworkStatus.ready,
           data: {
@@ -4228,6 +4242,8 @@ describe("type policies", function () {
               totalCount: 1292,
             },
           },
+          complete: true,
+          partial: false,
         });
 
         expect(cache.extract()).toMatchSnapshot();
@@ -4246,7 +4262,7 @@ describe("type policies", function () {
         });
         const snapshot = cache.extract();
 
-        expect(result).toEqual({
+        expect(result).toEqualApolloQueryResult({
           loading: false,
           networkStatus: NetworkStatus.ready,
           data: {
@@ -4256,6 +4272,8 @@ describe("type policies", function () {
               totalCount: 13531,
             },
           },
+          complete: true,
+          partial: false,
         });
 
         expect(snapshot).toMatchSnapshot();
@@ -4308,7 +4326,7 @@ describe("type policies", function () {
           },
         });
 
-        expect(result).toEqual({
+        expect(result).toEqualApolloQueryResult({
           loading: false,
           networkStatus: NetworkStatus.ready,
           data: {
@@ -4324,6 +4342,8 @@ describe("type policies", function () {
               totalCount: 1292,
             },
           },
+          complete: true,
+          partial: false,
         });
 
         expect(cache.extract()).toMatchSnapshot();
@@ -4341,7 +4361,7 @@ describe("type policies", function () {
         });
         const snapshot = cache.extract();
 
-        expect(result).toEqual({
+        expect(result).toEqualApolloQueryResult({
           loading: false,
           networkStatus: NetworkStatus.ready,
           data: {
@@ -4351,6 +4371,8 @@ describe("type policies", function () {
               totalCount: 13531,
             },
           },
+          complete: true,
+          partial: false,
         });
 
         expect(snapshot).toMatchSnapshot();