fix tests

Change-Id: I7619ef3ca3921ecc92efd65dc16c86548d51af3f

Author: igorbernstein2

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/metrics/BigtableTracerCallableTest.java

@@ -403,7 +403,11 @@ public void mutateRow(MutateRowRequest request, StreamObserver<MutateRowResponse
 
  @Override
  public void mutateRows(MutateRowsRequest request, StreamObserver<MutateRowsResponse> observer) {
- observer.onNext(MutateRowsResponse.getDefaultInstance());
+ MutateRowsResponse.Builder builder = MutateRowsResponse.newBuilder();
+ for (int i = 0; i < request.getEntriesCount(); i++) {
+ builder.addEntries(MutateRowsResponse.Entry.newBuilder().setIndex(i));
+ }
+ observer.onNext(builder.build());
  observer.onCompleted();
  }
 

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/metrics/BuiltinMetricsTracerTest.java

@@ -648,7 +648,11 @@ public void mutateRows(
  Thread.sleep(SERVER_LATENCY);
  } catch (InterruptedException e) {
  }
- responseObserver.onNext(MutateRowsResponse.getDefaultInstance());
+ MutateRowsResponse.Builder builder = MutateRowsResponse.newBuilder();
+ for (int i = 0; i < request.getEntriesCount(); i++) {
+ builder.addEntriesBuilder().setIndex(i);
+ }
+ responseObserver.onNext(builder.build());
  responseObserver.onCompleted();
  }
 

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/metrics/MetricsTracerTest.java

@@ -422,10 +422,15 @@ public void testBatchMutateRowsThrottledTime() throws Exception {
  new Answer() {
  @Override
  public Object answer(InvocationOnMock invocation) {
+ MutateRowsRequest request = (MutateRowsRequest) invocation.getArguments()[0];
  @SuppressWarnings("unchecked")
  StreamObserver<MutateRowsResponse> observer =
  (StreamObserver<MutateRowsResponse>) invocation.getArguments()[1];
- observer.onNext(MutateRowsResponse.getDefaultInstance());
+ MutateRowsResponse.Builder builder = MutateRowsResponse.newBuilder();
+ for (int i = 0; i < request.getEntriesCount(); i++) {
+ builder.addEntriesBuilder().setIndex(i);
+ }
+ observer.onNext(builder.build());
  observer.onCompleted();
  return null;
  }

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/metrics/StatsHeadersCallableTest.java

@@ -223,7 +223,11 @@ public void mutateRows(MutateRowsRequest request, StreamObserver<MutateRowsRespo
  observer.onError(new StatusRuntimeException(Status.UNAVAILABLE));
  return;
  }
- observer.onNext(MutateRowsResponse.getDefaultInstance());
+ MutateRowsResponse.Builder builder = MutateRowsResponse.newBuilder();
+ for (int i = 0; i < request.getEntriesCount(); i++) {
+ builder.addEntriesBuilder().setIndex(i);
+ }
+ observer.onNext(builder.build());
  observer.onCompleted();
  }
 

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

@@ -107,7 +107,11 @@ public void mutateRows(
  MutateRowsRequest request, StreamObserver<MutateRowsResponse> responseObserver) {
  attemptCounter.incrementAndGet();
  if (expectations.isEmpty()) {
- responseObserver.onNext(MutateRowsResponse.getDefaultInstance());
+ MutateRowsResponse.Builder builder = MutateRowsResponse.newBuilder();
+ for (int i = 0; i < request.getEntriesCount(); i++) {
+ builder.addEntriesBuilder().setIndex(i);
+ }
+ responseObserver.onNext(builder.build());
  responseObserver.onCompleted();
  } else {
  Exception expectedRpc = expectations.poll();