---

gremi64 · sduskis · commit b24790e967d1 · 2019-02-08T10:11:29.000-05:00
yaml --- r: 20697 b: refs/heads/autosynth-iamcredentials c: 64f3b2b h: refs/heads/master i: 20695: f1bf883
diff --git a/[refs] b/[refs]
@@ -162,7 +162,7 @@ refs/heads/pubsub-ordering-keys: 858d4e986a0ba48e08f00d42f51cbdecb175f5d6
 refs/tags/v0.75.0: c3673089ae09a897c1b4cf7dfe167fe4f8ab32fb
 refs/tags/v0.76.0: 395b016826d3ddf9cb8b34919636df15a4dbd032
 refs/tags/v0.77.0: 28a85a77883ccf5d48f297fd0ef3b3dca6ce01f0
-refs/heads/autosynth-iamcredentials: bb2bb7cff5766d2070a8421d902e4af1c9c18bac
+refs/heads/autosynth-iamcredentials: 64f3b2be23eab15df420e5890f14409fa52c8411
 refs/heads/release-google-cloud-java-v0.78.0: fae5e980779cf0173a152636b278015b9f60ee55
 refs/heads/release-google-cloud-java-v0.81.0: 0352cd0dd11f4fd1fbd1ff16e7a96beaccc7b475
 refs/heads/release-google-cloud-java-v0.81.1-SNAPSHOT: 5a74ccb1f12506a3b67b65521881298fde20bd6f
diff --git a/branches/autosynth-iamcredentials/google-cloud-clients/google-cloud-firestore/src/main/java/com/google/cloud/firestore/UpdateBuilder.java b/branches/autosynth-iamcredentials/google-cloud-clients/google-cloud-firestore/src/main/java/com/google/cloud/firestore/UpdateBuilder.java
@@ -664,4 +664,9 @@ public List<WriteResult> apply(CommitResponse commitResponse) {
   boolean isEmpty() {
     return mutations.isEmpty();
   }
+
+  /** Get the number of mutations. */
+  public int getMutationsSize() {
+    return mutations.size();
+  }
 }
diff --git a/branches/autosynth-iamcredentials/google-cloud-clients/google-cloud-firestore/src/test/java/com/google/cloud/firestore/WriteBatchTest.java b/branches/autosynth-iamcredentials/google-cloud-clients/google-cloud-firestore/src/test/java/com/google/cloud/firestore/WriteBatchTest.java
@@ -91,6 +91,8 @@ public void updateDocument() throws Exception {
     batch.update(documentReference, updateTime, "foo", "bar");
     batch.update(documentReference, LocalFirestoreHelper.SINGLE_FIELD_MAP, updateTime);
 
+    assertEquals(4, batch.getMutationsSize());
+
     List<WriteResult> writeResults = batch.commit().get();
     List<Write> writes = new ArrayList<>();
 
@@ -126,6 +128,8 @@ public void setDocument() throws Exception {
     writes.add(set(LocalFirestoreHelper.SINGLE_FIELD_PROTO, Arrays.asList("foo")));
     writes.add(set(LocalFirestoreHelper.SINGLE_FIELD_PROTO, Arrays.asList("foo")));
 
+    assertEquals(4, batch.getMutationsSize());
+
     List<WriteResult> writeResults = batch.commit().get();
     for (int i = 0; i < writeResults.size(); ++i) {
       assertEquals(Timestamp.ofTimeSecondsAndNanos(i, i), writeResults.get(i).getUpdateTime());
@@ -144,6 +148,8 @@ public void omitWriteResultForDocumentTransforms() throws Exception {
 
     batch.set(documentReference, map("time", FieldValue.serverTimestamp()));
 
+    assertEquals(1, batch.getMutationsSize());
+
     List<WriteResult> writeResults = batch.commit().get();
     assertEquals(1, writeResults.size());
   }
@@ -159,6 +165,8 @@ public void createDocument() throws Exception {
         .create(documentReference, LocalFirestoreHelper.SINGLE_FIELD_MAP)
         .create(documentReference, LocalFirestoreHelper.SINGLE_FIELD_OBJECT);
 
+    assertEquals(2, batch.getMutationsSize());
+
     List<WriteResult> writeResults = batch.commit().get();
     List<Write> writes = new ArrayList<>();
 
@@ -188,6 +196,8 @@ public void deleteDocument() throws Exception {
     precondition.getUpdateTimeBuilder().setSeconds(1).setNanos(2);
     writes.add(delete(precondition.build()));
 
+    assertEquals(2, batch.getMutationsSize());
+
     List<WriteResult> writeResults = batch.commit().get();
 
     for (int i = 0; i < writeResults.size(); ++i) {

Original file line number	Diff line number	Diff line change
`@@ -664,4 +664,9 @@ public List<WriteResult> apply(CommitResponse commitResponse) {`
`664`	`664`	`boolean isEmpty() {`
`665`	`665`	`return mutations.isEmpty();`
`666`	`666`	`}`
	`667`	`+`
	`668`	`+ /** Get the number of mutations. */`
	`669`	`+ public int getMutationsSize() {`
	`670`	`+ return mutations.size();`
	`671`	`+ }`
`667`	`672`	`}`