Merge a08a955 into f08e0c7

Author: milaGGL

firebase-firestore/src/androidTest/java/com/google/firebase/firestore/FirestoreTest.java

@@ -16,6 +16,7 @@
 
 import static com.google.firebase.firestore.AccessHelper.getAsyncQueue;
 import static com.google.firebase.firestore.testutil.IntegrationTestUtil.checkOnlineAndOfflineResultsMatch;
+import static com.google.firebase.firestore.testutil.IntegrationTestUtil.getLargestDocContent;
 import static com.google.firebase.firestore.testutil.IntegrationTestUtil.isRunningAgainstEmulator;
 import static com.google.firebase.firestore.testutil.IntegrationTestUtil.newTestSettings;
 import static com.google.firebase.firestore.testutil.IntegrationTestUtil.provider;
@@ -1988,4 +1989,121 @@ public void snapshotListenerSortsInvalidUnicodeStringsAsServer() {
 
     checkOnlineAndOfflineResultsMatch(colRef, orderedQuery, expectedDocIds.toArray(new String[0]));
   }
+
+  @Test
+  public void testCanCRUDAndQueryLargeDocuments() {
+    CollectionReference collRef = testCollection();
+    DocumentReference docRef = collRef.document();
+    Map<String, Object> data = getLargestDocContent();
+
+    // Set
+    waitFor(docRef.set(data));
+
+    // Get
+    DocumentSnapshot snapshot = waitFor(docRef.get());
+    assertEquals(data, snapshot.getData());
+
+    // Update
+    Map<String, Object> newData = getLargestDocContent();
+    waitFor(docRef.update(newData));
+    snapshot = waitFor(docRef.get());
+    assertEquals(newData, snapshot.getData());
+
+    // Query
+    QuerySnapshot querySnapshot = waitFor(collRef.get());
+    assertEquals(querySnapshot.size(), 1);
+    assertEquals(newData, querySnapshot.getDocuments().get(0).getData());
+
+    // Delete
+    waitFor(docRef.delete());
+    snapshot = waitFor(docRef.get());
+    assertFalse(snapshot.exists());
+  }
+
+  @Test
+  public void testCanCRUDLargeDocumentsInsideTransaction() {
+    CollectionReference collRef = testCollection();
+
+    DocumentReference docRef1 = collRef.document();
+    DocumentReference docRef2 = collRef.document();
+    DocumentReference docRef3 = collRef.document();
+    Map<String, Object> data = getLargestDocContent();
+    Map<String, Object> newData = getLargestDocContent();
+    waitFor(docRef1.set(data));
+    waitFor(docRef3.set(data));
+
+    waitFor(
+        collRef
+            .getFirestore()
+            .runTransaction(
+                transaction -> {
+                  // Get and update
+                  DocumentSnapshot snapshot = transaction.get(docRef1);
+                  assertEquals(data, snapshot.getData());
+                  transaction.update(docRef1, newData);
+
+                  // Set
+                  transaction.set(docRef2, data);
+
+                  // Delete
+                  transaction.delete(docRef3);
+                  return null;
+                }));
+
+    DocumentSnapshot snapshot = waitFor(docRef1.get());
+    assertEquals(newData, snapshot.getData());
+
+    snapshot = waitFor(docRef2.get());
+    assertEquals(data, snapshot.getData());
+
+    snapshot = waitFor(docRef3.get());
+    assertFalse(snapshot.exists());
+  }
+
+  @Test
+  public void listenToLargeQuerySnapshot() throws Exception {
+    CollectionReference collRef = testCollection();
+    DocumentReference docRef = collRef.document();
+    Map<String, Object> data = getLargestDocContent();
+    waitFor(docRef.set(data));
+
+    CountDownLatch latch = new CountDownLatch(1);
+    List<QuerySnapshot> querySnapshots = new ArrayList<>();
+    ListenerRegistration registration =
+        collRef.addSnapshotListener(
+            (value, error) -> {
+              querySnapshots.add(value);
+              latch.countDown();
+            });
+
+    latch.await();
+    registration.remove();
+
+    assertEquals(querySnapshots.size(), 1);
+    assertEquals(querySnapshots.get(0).getDocuments().size(), 1);
+    assertEquals(data, querySnapshots.get(0).getDocuments().get(0).getData());
+  }
+
+  @Test
+  public void listenToLargeDocumentSnapshot() throws Exception {
+    DocumentReference docRef = testDocument();
+    Map<String, Object> data = getLargestDocContent();
+    waitFor(docRef.set(data));
+
+    CountDownLatch latch = new CountDownLatch(1);
+    List<DocumentSnapshot> documentSnapshots = new ArrayList<>();
+
+    ListenerRegistration registration =
+        docRef.addSnapshotListener(
+            (value, error) -> {
+              documentSnapshots.add(value);
+              latch.countDown();
+            });
+
+    latch.await();
+    registration.remove();
+
+    assertEquals(documentSnapshots.size(), 1);
+    assertEquals(data, documentSnapshots.get(0).getData());
+  }
 }

firebase-firestore/src/androidTest/java/com/google/firebase/firestore/testutil/IntegrationTestUtil.java

@@ -30,6 +30,7 @@
 import com.google.firebase.FirebaseApp;
 import com.google.firebase.FirebaseOptions;
 import com.google.firebase.firestore.AccessHelper;
+import com.google.firebase.firestore.Blob;
 import com.google.firebase.firestore.BuildConfig;
 import com.google.firebase.firestore.CollectionReference;
 import com.google.firebase.firestore.DocumentReference;
@@ -52,9 +53,11 @@
 import com.google.firebase.firestore.util.Logger;
 import com.google.firebase.firestore.util.Logger.Level;
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Random;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Semaphore;
@@ -125,6 +128,10 @@ public enum TargetBackend {
 
   private static boolean backendPrimed = false;
 
+  private static final Random RANDOM = new Random();
+
+  private static final int MAX_BYTES_PER_FIELD_VALUE = 1048487;
+
   // FirebaseOptions needed to create a test FirebaseApp.
   private static final FirebaseOptions OPTIONS =
       new FirebaseOptions.Builder()
@@ -561,4 +568,17 @@ public static void checkOnlineAndOfflineResultsMatch(
       assertEquals(expectedDocIds, querySnapshotToIds(docsFromServer));
     }
   }
+
+  /**
+   * Returns a Blob with the size equal to the largest number of bytes allowed to
+   * be stored in a Firestore document.
+   */
+  public static Map<String, Object> getLargestDocContent() {
+    // Subtract 8 for '__name__', 20 for its value, and 4 for 'blob'.
+    int numBytesToUse = MAX_BYTES_PER_FIELD_VALUE - 8 - 20 - 4;
+    byte[] bytes = new byte[numBytesToUse];
+    // Fill the byte array with random values
+    RANDOM.nextBytes(bytes);
+    return Collections.singletonMap("blob", Blob.fromBytes(bytes));
+  }
 }