MutateRowsResponse.Entry.newBuilder() .setIndex(i++) .setStatus(com.google.rpc.Status.newBuilder().setCode(responseCode.value())) .build(); responseBuilder.addEntries(responseEntry);
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)) .build());
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)) .build()); 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(); }
@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(); }
@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); }