@Override public NoopFailoverPolicy build() { return new NoopFailoverPolicy(); }
@Override public NoopFailoverPolicy build() { return new NoopFailoverPolicy(); }
@Test public void deliveryAddsBatchItemToBatchEmitter() { // given TestHttpObjectFactory objectFactory = createTestObjectFactoryBuilder().build(); TestBatchEmitterFactory batchEmitterFactory = spy(new TestBatchEmitterFactory()); BatchEmitter emitter = batchEmitterFactory.createInstance(TEST_BATCH_SIZE, TEST_DELIVERY_INTERVAL, objectFactory, new NoopFailoverPolicy()); TestAsyncBatchDelivery delivery = spy(new TestAsyncBatchDelivery( TEST_BATCH_SIZE, TEST_DELIVERY_INTERVAL, objectFactory, new NoopFailoverPolicy(), null) { @Override protected BatchEmitterServiceProvider createBatchEmitterServiceProvider() { return batchEmitterFactory; } }); String testMessage = "test message"; // when delivery.add("testIndexName", testMessage); // then ArgumentCaptor<BulkEmitterTest.TestBatchItem> captor = ArgumentCaptor.forClass(BulkEmitterTest.TestBatchItem.class); verify(emitter, times(1)).add(captor.capture()); assertEquals(testMessage, captor.getValue().getData(null)); }
@Test public void lifecycleStartStartsBatchEmitter() { // given BatchEmitter batchEmitter = mock(BatchEmitter.class); BatchEmitterServiceProvider batchEmitterFactory = new TestBatchEmitterFactory() { @Override public BatchEmitter createInstance(int batchSize, int deliveryInterval, ClientObjectFactory clientObjectFactory, FailoverPolicy failoverPolicy) { return batchEmitter; } }; TestAsyncBatchDelivery batchDelivery = spy(new TestAsyncBatchDelivery( TEST_BATCH_SIZE, TEST_DELIVERY_INTERVAL, createTestObjectFactoryBuilder().build(), new NoopFailoverPolicy(), null) { @Override protected BatchEmitterServiceProvider createBatchEmitterServiceProvider() { return batchEmitterFactory; } }); // when batchDelivery.start(); // then verify(batchEmitter).start(); }
public static Builder createTestBatchDeliveryBuilder() { return spy(AsyncBatchDelivery.newBuilder() .withBatchSize(TEST_BATCH_SIZE) .withDeliveryInterval(TEST_DELIVERY_INTERVAL) .withClientObjectFactory(createTestObjectFactoryBuilder().build())) .withFailoverPolicy(new NoopFailoverPolicy()) .withIndexTemplate(new IndexTemplate(TEST_INDEX_TEMPLATE, TEST_PATH)); }