.setStatus(com.google.rpc.Status.newBuilder().setCode(responseCode.value())) .build(); responseBuilder.addEntries(responseEntry);
.addEntries( MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO)) .addEntries( MutateRowsResponse.Entry.newBuilder() .setIndex(1) .setStatus(TRANSIENT_ERROR_STATUS_PROTO)) .addEntries( MutateRowsResponse.Entry.newBuilder() .setIndex(2)
.addEntries( MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO)) .addEntries( MutateRowsResponse.Entry.newBuilder() .setIndex(1) .setStatus(TRANSIENT_ERROR_STATUS_PROTO)) .addEntries( MutateRowsResponse.Entry.newBuilder() .setIndex(2) Lists.newArrayList( MutateRowsResponse.newBuilder() .addEntries( MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO)) .build());
@Test public void singleEntrySuccessTest() throws Exception { MutateRowsRequest request = MutateRowsRequest.newBuilder().addEntries(Entry.getDefaultInstance()).build(); innerCallable.response.add( MutateRowsResponse.newBuilder() .addEntries( MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO)) .build()); MutateRowsAttemptCallable attemptCallable = new MutateRowsAttemptCallable(innerCallable, request, callContext, retryCodes); attemptCallable.setExternalFuture(parentFuture); attemptCallable.call(); // Attempt completed successfully and the useless response has been suppressed assertThat(parentFuture.attemptFuture.get()).isNull(); // innerCallable received the request assertThat(innerCallable.lastRequest).isEqualTo(request); }
@Test public void testNoRpcTimeout() { parentFuture.timedAttemptSettings = parentFuture.timedAttemptSettings.toBuilder().setRpcTimeout(Duration.ZERO).build(); MutateRowsRequest request = MutateRowsRequest.newBuilder().addEntries(Entry.getDefaultInstance()).build(); innerCallable.response.add( MutateRowsResponse.newBuilder() .addEntries( MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO)) .build()); MutateRowsAttemptCallable attemptCallable = new MutateRowsAttemptCallable(innerCallable, request, callContext, retryCodes); attemptCallable.setExternalFuture(parentFuture); attemptCallable.call(); assertThat(innerCallable.lastContext.getTimeout()).isNull(); }