Add a waitFor to set (#663)

Brian Chen · web-flow · commit 4751124eb6e9 · 2019-07-25T12:01:55.000-07:00
diff --git a/firebase-firestore/src/androidTest/java/com/google/firebase/firestore/TransactionTest.java b/firebase-firestore/src/androidTest/java/com/google/firebase/firestore/TransactionTest.java
@@ -156,7 +156,7 @@ private void expectError(Code expected) {
     private void prepareDoc() {
       docRef = db.collection("tester-docref").document();
       if (fromExistingDoc) {
-        docRef.set(map("foo", "bar0"));
+        waitFor(docRef.set(map("foo", "bar0")));
         DocumentSnapshot docSnap = waitFor(docRef.get());
         assertTrue(docSnap.exists());
       }

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ private void expectError(Code expected) {`
`156`	`156`	`private void prepareDoc() {`
`157`	`157`	`docRef = db.collection("tester-docref").document();`
`158`	`158`	`if (fromExistingDoc) {`
`159`		`- docRef.set(map("foo", "bar0"));`
	`159`	`+ waitFor(docRef.set(map("foo", "bar0")));`
`160`	`160`	`DocumentSnapshot docSnap = waitFor(docRef.get());`
`161`	`161`	`assertTrue(docSnap.exists());`
`162`	`162`	`}`