Merge branch 'mila/count' into mila/count-non-lite-api-tests

Author: milaGGL

packages/firestore/test/integration/api/aggregation.test.ts

@@ -22,6 +22,7 @@ import {
   collectionGroup,
   countQuery,
   doc,
+  disableNetwork,
   getAggregateFromServerDirect,
   query,
   terminate,
@@ -119,4 +120,19 @@ apiDescribe('Aggregation query', (persistence: boolean) => {
       await terminate(firestore);
     });
   });
+
+  it('getAggregateFromServerDirect fails if user is offline', () => {
+    return withEmptyTestCollection(
+      persistence,
+      async (collection, firestore) => {
+        await disableNetwork(firestore);
+        const countQuery_ = countQuery(collection);
+        await expect(
+          getAggregateFromServerDirect(countQuery_)
+        ).to.be.eventually.rejectedWith(
+          'Failed to get aggregate result because the client is offline'
+        );
+      }
+    );
+  });
 });