.containsAllOf(Code.ABORTED, Code.DEADLINE_EXCEEDED); assertThat(builder.bulkMutateRowsSettings().getRetrySettings()).isEqualTo(retrySettings); assertThat(builder.bulkMutateRowsSettings().getBatchingSettings()).isSameAs(batchingSettings); assertThat(builder.build().toBuilder().bulkMutateRowsSettings().getRetrySettings()) .isEqualTo(retrySettings); assertThat(builder.build().toBuilder().bulkMutateRowsSettings().getBatchingSettings()) .isSameAs(batchingSettings);
logBuilder.writeLogEntriesSettings().getBatchingSettings(); logBuilder .writeLogEntriesSettings()
@Test public void testEmptyBuilder() { @SuppressWarnings("unchecked") BatchingDescriptor<Integer, Integer> batchingDescriptor = Mockito.mock(BatchingDescriptor.class); BatchingCallSettings.Builder<Integer, Integer> builder = BatchingCallSettings.newBuilder(batchingDescriptor); Truth.assertThat(builder.getBatchingDescriptor()).isSameAs(batchingDescriptor); Truth.assertThat(builder.getBatchingSettings()).isNull(); Truth.assertThat(builder.getFlowController()).isNull(); Truth.assertThat(builder.getRetryableCodes().size()).isEqualTo(0); Truth.assertThat(builder.getRetrySettings()).isNotNull(); BatchingSettings batchingSettings = BatchingSettings.newBuilder().setElementCountThreshold(1L).build(); builder.setBatchingSettings(batchingSettings); BatchingCallSettings settings = builder.build(); Truth.assertThat(settings.getBatchingDescriptor()).isSameAs(batchingDescriptor); Truth.assertThat(settings.getBatchingSettings()).isSameAs(batchingSettings); Truth.assertThat(settings.getFlowController()).isNotNull(); Truth.assertThat(settings.getRetryableCodes().size()).isEqualTo(0); Truth.assertThat(settings.getRetrySettings()).isNotNull(); }
logBuilder.writeLogEntriesSettings().getBatchingSettings(); logBuilder .writeLogEntriesSettings()
@Test public void testBuilder() { @SuppressWarnings("unchecked") BatchingDescriptor<Integer, Integer> batchingDescriptor = Mockito.mock(BatchingDescriptor.class); BatchingCallSettings.Builder<Integer, Integer> builder = BatchingCallSettings.newBuilder(batchingDescriptor); BatchingSettings batchingSettings = BatchingSettings.newBuilder().setElementCountThreshold(1L).build(); FlowController flowController = Mockito.mock(FlowController.class); Set<StatusCode.Code> retryCodes = Sets.newHashSet(Code.UNAVAILABLE); RetrySettings retrySettings = RetrySettings.newBuilder().build(); builder.setBatchingSettings(batchingSettings); builder.setFlowController(flowController); builder.setRetryableCodes(retryCodes); builder.setRetrySettings(retrySettings); Truth.assertThat(builder.getBatchingDescriptor()).isSameAs(batchingDescriptor); Truth.assertThat(builder.getBatchingSettings()).isSameAs(batchingSettings); Truth.assertThat(builder.getFlowController()).isSameAs(flowController); Truth.assertThat(builder.getRetryableCodes().size()).isEqualTo(1); Truth.assertThat(builder.getRetrySettings()).isSameAs(retrySettings); BatchingCallSettings settings = builder.build(); Truth.assertThat(settings.getBatchingDescriptor()).isSameAs(batchingDescriptor); Truth.assertThat(settings.getBatchingSettings()).isSameAs(batchingSettings); Truth.assertThat(settings.getFlowController()).isSameAs(flowController); Truth.assertThat(settings.getRetryableCodes().size()).isEqualTo(1); Truth.assertThat(settings.getRetrySettings()).isSameAs(retrySettings); }
@Test public void testBuilderFromSettings() throws Exception { @SuppressWarnings("unchecked") BatchingDescriptor<Integer, Integer> batchingDescriptor = Mockito.mock(BatchingDescriptor.class); BatchingCallSettings.Builder<Integer, Integer> builder = BatchingCallSettings.newBuilder(batchingDescriptor); BatchingSettings batchingSettings = BatchingSettings.newBuilder().setElementCountThreshold(1L).build(); FlowController flowController = Mockito.mock(FlowController.class); Set<StatusCode.Code> retryCodes = Sets.newHashSet(Code.UNAVAILABLE); RetrySettings retrySettings = RetrySettings.newBuilder().build(); builder.setBatchingSettings(batchingSettings); builder.setFlowController(flowController); builder.setRetryableCodes(retryCodes); builder.setRetrySettings(retrySettings); BatchingCallSettings settings = builder.build(); BatchingCallSettings.Builder newBuilder = settings.toBuilder(); Truth.assertThat(newBuilder.getBatchingDescriptor()).isSameAs(batchingDescriptor); Truth.assertThat(newBuilder.getBatchingSettings()).isSameAs(batchingSettings); Truth.assertThat(newBuilder.getFlowController()).isSameAs(flowController); Truth.assertThat(newBuilder.getRetryableCodes().size()).isEqualTo(1); Truth.assertThat(newBuilder.getRetrySettings()).isSameAs(retrySettings); }
private static <RequestT, ResponseT> void assertIsReflectionEqual( BatchingCallSettings.Builder<RequestT, ResponseT> builderA, BatchingCallSettings.Builder<RequestT, ResponseT> builderB) { assertIsReflectionEqual( builderA, builderB, new String[] {"retrySettings", "batchingDescriptor", "batchingSettings", "flowController"}); assertIsReflectionEqual(builderA.getRetrySettings(), builderB.getRetrySettings()); assertIsReflectionEqual(builderA.getBatchingSettings(), builderB.getBatchingSettings()); // TODO compare other batching things (batchingDescriptor, flowController) } }
settings .bulkMutationsSettings() .getBatchingSettings() .toBuilder() .setDelayThreshold(FLUSH_PERIOD)
@Before public void setUp() throws Exception { server = new InProcessServer<>(new BigtableGrpc.BigtableImplBase() {}, NAME); server.start(); channelProvider = LocalChannelProvider.create(NAME); BigtableDataSettings.Builder settings = BigtableDataSettings.newBuilder() .setProjectId(PROJECT_ID) .setInstanceId(INSTANCE_ID) .setTransportChannelProvider(channelProvider) .setCredentialsProvider(NoCredentialsProvider.create()); // Force immediate flush settings .bulkMutationsSettings() .setBatchingSettings( settings .bulkMutationsSettings() .getBatchingSettings() .toBuilder() .setElementCountThreshold(1L) .build()); client = BigtableDataClient.create(settings.build()); }