From 4eb7a24e2fa76cefa492e2b7d03b0bd314058f31 Mon Sep 17 00:00:00 2001
From: Denver Coneybeare <dconeybe@google.com>
Date: Fri, 23 Sep 2022 11:54:31 -0400
Subject: [PATCH] test: Add
 ITQueryCountTest.countShouldReturnNumberOfDocumentsForPartitionQuery

---
 .../java/com/google/cloud/firestore/ITQueryCountTest.java | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/google-cloud-firestore/src/test/java/com/google/cloud/firestore/ITQueryCountTest.java b/google-cloud-firestore/src/test/java/com/google/cloud/firestore/ITQueryCountTest.java
index d8e9f51fae..c14b762b2e 100644
--- a/google-cloud-firestore/src/test/java/com/google/cloud/firestore/ITQueryCountTest.java
+++ b/google-cloud-firestore/src/test/java/com/google/cloud/firestore/ITQueryCountTest.java
@@ -136,6 +136,14 @@ public void countShouldReturnNumberOfDocumentsForCollectionGroups() throws Excep
     assertThat(snapshot.getCount()).isEqualTo(13);
   }
 
+  @Test
+  public void countShouldReturnNumberOfDocumentsForPartitionQuery() throws Exception {
+    CollectionReference collection = createCollectionWithDocuments(3).collection();
+    Query query = collection.select("abc");
+    AggregateQuerySnapshot snapshot = query.count().get().get();
+    assertThat(snapshot.getCount()).isEqualTo(3);
+  }
+
   @Test
   public void inFlightCountQueriesShouldCompleteSuccessfullyWhenFirestoreIsClosed()
       throws Exception {