MutateRowsResponse.Entry.newBuilder() .setIndex(i++) .setStatus(com.google.rpc.Status.newBuilder().setCode(responseCode.value()))
MutateRowsResponse.newBuilder() .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) .setStatus(PERMENANT_ERROR_STATUS_PROTO))
MutateRowsResponse.newBuilder() .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) .setStatus(PERMENANT_ERROR_STATUS_PROTO)) MutateRowsResponse.newBuilder() .addEntries( MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO)) .build()); attemptCallable.call();
private static Entry createEntry(int i, Status status) { return MutateRowsResponse.Entry.newBuilder().setIndex(i).setStatus(status).build(); }
public Builder newBuilderForType() { return newBuilder(); } public static Builder newBuilder() {
@java.lang.Override public Builder newBuilderForType() { return newBuilder(); }
@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(); }
@java.lang.Override public Builder newBuilderForType() { return newBuilder(); }