Update tests after restructuring queryManager.mutationStore.

apollographql · Oct 15, 2020 · d53928c · d53928c
1 parent ad4be6e
commit d53928c
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 19 deletions.
diff --git a/src/__tests__/optimistic.ts b/src/__tests__/optimistic.ts
@@ -147,6 +147,8 @@ describe('optimistic mutation results', () => {
           return null;
         },
       }),
+      // Enable client.queryManager.mutationStore tracking.
+      connectToDevTools: true,
     });
 
     const obsHandle = client.watchQuery({ query });
@@ -377,9 +379,9 @@ describe('optimistic mutation results', () => {
             );
 
             // @ts-ignore
-            const latestState = queryManager.mutationStore;
-            expect(latestState.get('1').loading).toBe(false);
-            expect(latestState.get('2').loading).toBe(true);
+            const latestState = queryManager.mutationStore!;
+            expect(latestState[1].loading).toBe(false);
+            expect(latestState[2].loading).toBe(true);
 
             return res;
           });
@@ -397,17 +399,17 @@ describe('optimistic mutation results', () => {
             );
 
             // @ts-ignore
-            const latestState = queryManager.mutationStore;
-            expect(latestState.get('1').loading).toBe(false);
-            expect(latestState.get('2').loading).toBe(false);
+            const latestState = queryManager.mutationStore!;
+            expect(latestState[1].loading).toBe(false);
+            expect(latestState[2].loading).toBe(false);
 
             return res;
           });
 
         // @ts-ignore
-        const mutationsState = queryManager.mutationStore;
-        expect(mutationsState.get('1').loading).toBe(true);
-        expect(mutationsState.get('2').loading).toBe(true);
+        const mutationsState = queryManager.mutationStore!;
+        expect(mutationsState[1].loading).toBe(true);
+        expect(mutationsState[2].loading).toBe(true);
 
         checkBothMutationsAreApplied(
           'Optimistically generated',
@@ -621,9 +623,9 @@ describe('optimistic mutation results', () => {
             );
 
             // @ts-ignore
-            const latestState = client.queryManager.mutationStore;
-            expect(latestState.get('1').loading).toBe(false);
-            expect(latestState.get('2').loading).toBe(true);
+            const latestState = client.queryManager.mutationStore!;
+            expect(latestState[1].loading).toBe(false);
+            expect(latestState[2].loading).toBe(true);
 
             return res;
           });
@@ -641,17 +643,17 @@ describe('optimistic mutation results', () => {
             );
 
             // @ts-ignore
-            const latestState = client.queryManager.mutationStore;
-            expect(latestState.get('1').loading).toBe(false);
-            expect(latestState.get('2').loading).toBe(false);
+            const latestState = client.queryManager.mutationStore!;
+            expect(latestState[1].loading).toBe(false);
+            expect(latestState[2].loading).toBe(false);
 
             return res;
           });
 
         // @ts-ignore
-        const mutationsState = client.queryManager.mutationStore;
-        expect(mutationsState.get('1').loading).toBe(true);
-        expect(mutationsState.get('2').loading).toBe(true);
+        const mutationsState = client.queryManager.mutationStore!;
+        expect(mutationsState[1].loading).toBe(true);
+        expect(mutationsState[2].loading).toBe(true);
 
         checkBothMutationsAreApplied(
           'Optimistically generated',

diff --git a/src/core/__tests__/QueryManager/index.ts b/src/core/__tests__/QueryManager/index.ts
@@ -74,6 +74,8 @@ describe('QueryManager', () => {
       cache: new InMemoryCache({ addTypename: false, ...config }),
       clientAwareness,
       queryDeduplication,
+      // Enable client.queryManager.mutationStore tracking.
+      onBroadcast() {},
     });
   };
 
@@ -3265,7 +3267,7 @@ describe('QueryManager', () => {
         queryManager.cache.extract(),
       ).toEqual({});
       expect(queryManager.getQueryStore()).toEqual({});
-      expect(queryManager.mutationStore.getStore()).toEqual({});
+      expect(queryManager.mutationStore).toEqual({});
 
       resolve();
     });