fix MockConnection error

Author: milaGGL

packages/firestore/src/api/aggregate.ts

@@ -34,7 +34,7 @@ export {
  * server. Returns an error if the network is not available.
  *
  * @param query - The `Query` to execute.
- * 
+ *
  * @returns A `Promise` that will be resolved with the results of the query.
  */
 export function getCountFromServer(

packages/firestore/src/remote/datastore.ts

@@ -250,10 +250,11 @@ export async function invokeRunAggregationQueryRpc(
   if (!datastoreImpl.connection.shouldResourcePathBeIncludedInRequest) {
     delete request.parent;
   }
+  const EXPECTED_RESPONSE_COUNT = 1;
   const response = await datastoreImpl.invokeStreamingRPC<
     ProtoRunAggregationQueryRequest,
     ProtoRunAggregationQueryResponse
-  >('RunAggregationQuery', parent!, request);
+  >('RunAggregationQuery', parent!, request, EXPECTED_RESPONSE_COUNT);
   return (
     response
       // Omit RunAggregationQueryResponse that only contain readTimes.

packages/firestore/test/unit/remote/datastore.test.ts

@@ -65,6 +65,8 @@ describe('Datastore', () => {
     ): Stream<Req, Resp> {
       throw new Error('MockConnection.openStream() must be replaced');
     }
+
+    shouldResourcePathBeIncludedInRequest: boolean = false;
   }
 
   class MockAuthCredentialsProvider extends EmptyAuthCredentialsProvider {

packages/firestore/test/unit/specs/spec_test_components.ts

@@ -253,6 +253,8 @@ export class MockConnection implements Connection {
 
   constructor(private queue: AsyncQueue) {}
 
+  shouldResourcePathBeIncludedInRequest: boolean = false;
+
   /**
    * Tracks the currently active watch targets as detected by the mock watch
    * stream, as a mapping from target ID to query Target.