test(bigtable): add unit test for omitted entry in multi-entry Mutate… (#14463)

…RowsResponse

Author: mutianf

java-bigtable/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/mutaterows/MutateRowsAttemptCallableTest.java

@@ -160,6 +160,39 @@ public void testNoRpcTimeout() {
     assertThat(innerCallable.lastContext.getTimeout()).isNull();
   }
 
+  @Test
+  public void partialOmissionMultiEntryTest() throws Exception {
+    MutateRowsRequest request =
+        MutateRowsRequest.newBuilder()
+            .addEntries(Entry.newBuilder().setRowKey(ByteString.copyFromUtf8("0-ok")))
+            .addEntries(Entry.newBuilder().setRowKey(ByteString.copyFromUtf8("1-omitted")))
+            .addEntries(Entry.newBuilder().setRowKey(ByteString.copyFromUtf8("2-ok")))
+            .build();
+    innerCallable.response.add(
+        MutateRowsResponse.newBuilder()
+            .addEntries(
+                MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO))
+            .addEntries(
+                MutateRowsResponse.Entry.newBuilder().setIndex(2).setStatus(OK_STATUS_PROTO))
+            .build());
+
+    MutateRowsAttemptCallable attemptCallable =
+        new MutateRowsAttemptCallable(
+            innerCallable, request, callContext, retryCodes, mockRetryAlgorithm);
+    attemptCallable.setExternalFuture(parentFuture);
+    attemptCallable.call();
+
+    MutateRowsAttemptResult result = parentFuture.attemptFuture.get();
+    assertThat(result.getFailedMutations()).hasSize(1);
+    FailedMutation failedMutation = result.getFailedMutations().get(0);
+    assertThat(failedMutation.getIndex()).isEqualTo(1);
+    assertThat(failedMutation.getError().getStatusCode().getCode()).isEqualTo(Code.INTERNAL);
+    assertThat(failedMutation.getError().isRetryable()).isFalse();
+    assertThat(failedMutation.getError())
+        .hasMessageThat()
+        .contains("Missing entry response for entry 1");
+  }
+
   @Test
   public void mixedTest() throws Exception {
     // Setup the request & response

java-bigtable/test-proxy/src/main/java/com/google/cloud/bigtable/testproxy/CbtTestProxy.java

@@ -321,10 +321,10 @@ public void bulkMutateRows(
     } catch (MutateRowsException e) {
       MutateRowsResult.Builder resultBuilder = MutateRowsResult.newBuilder();
       com.google.rpc.Status topLevelStatus =
-        com.google.rpc.Status.newBuilder()
-          .setCode(e.getStatusCode().getCode().ordinal())
-          .setMessage(e.getMessage())
-          .build();
+          com.google.rpc.Status.newBuilder()
+              .setCode(e.getStatusCode().getCode().ordinal())
+              .setMessage(e.getMessage())
+              .build();
       for (MutateRowsException.FailedMutation failed : e.getFailedMutations()) {
         resultBuilder
             .addEntriesBuilder()
@@ -335,15 +335,13 @@ public void bulkMutateRows(
                     .setMessage(failed.getError().getMessage())
                     .build());
 
-        topLevelStatus = com.google.rpc.Status.newBuilder()
-          .setCode(failed.getError().getStatusCode().getCode().ordinal())
-          .setMessage(failed.getError().getMessage())
-          .build();
+        topLevelStatus =
+            com.google.rpc.Status.newBuilder()
+                .setCode(failed.getError().getStatusCode().getCode().ordinal())
+                .setMessage(failed.getError().getMessage())
+                .build();
       }
-      responseObserver.onNext(
-          resultBuilder
-              .setStatus(topLevelStatus)
-              .build());
+      responseObserver.onNext(resultBuilder.setStatus(topLevelStatus).build());
       responseObserver.onCompleted();
       return;
     } catch (ApiException e) {