@Override public TestHttpObjectFactory build() { if (serverUris == null) { throw new ConfigurationException("No serverUris provided for JestClientConfig"); } return new TestHttpObjectFactory(Arrays.asList(serverUris.split(";")), connTimeout, readTimeout, maxTotalConnection, defaultMaxTotalConnectionPerRoute, discoveryEnabled); }
@Override public Function<BulkEmitterTest.TestBatch, Boolean> createBatchListener(FailoverPolicy failoverPolicy) { return new Function<BulkEmitterTest.TestBatch, Boolean>() { private Function<BulkEmitterTest.TestBatch, Boolean> failureHandler = createFailureHandler(failoverPolicy); private TestClient client = createClient(); @Override public Boolean apply(BulkEmitterTest.TestBatch bulk) { TestResultHandler<Object> jestResultHandler = createResultHandler(bulk, failureHandler); client.executeAsync(bulk, jestResultHandler); return true; } }; }
@Override public Boolean apply(BulkEmitterTest.TestBatch bulk) { TestResultHandler<Object> jestResultHandler = createResultHandler(bulk, failureHandler); client.executeAsync(bulk, jestResultHandler); return true; }
@Test public void lifecycleStartDoesntSetUpIndexTemplateExecutionIfIndexTemplateIsNotConfigured() { // given TestHttpObjectFactory objectFactory = spy(createTestObjectFactoryBuilder().build()); BatchDelivery batchDelivery = createTestBatchDeliveryBuilder() .withClientObjectFactory(objectFactory) .withIndexTemplate(null) .build(); // when batchDelivery.start(); // then verify(objectFactory, never()).addOperation(any()); }
public static TestHttpObjectFactory.Builder createTestObjectFactoryBuilder() { TestHttpObjectFactory.Builder builder = TestHttpObjectFactory.newBuilder(); builder.withServerUris(TEST_SERVER_URIS); return builder; }
@Test public void lifecycleStartSetsUpIndexTemplateExecutionIfIndexTemplateIsConfigured() { // given TestHttpObjectFactory objectFactory = spy(createTestObjectFactoryBuilder().build()); IndexTemplate indexTemplate = mock(IndexTemplate.class); BatchDelivery batchDelivery = createTestBatchDeliveryBuilder() .withClientObjectFactory(objectFactory) .withIndexTemplate(indexTemplate) .build(); // when batchDelivery.start(); // then verify(objectFactory).addOperation(any()); }