honeymoose/OpenSearch
1
0
Fork 0
mirror of https://github.com/honeymoose/OpenSearch.git synced 2025-03-31 04:18:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

[TEST] make BulkTest more robust if test infra is slow

Browse Source
This commit is contained in:
Simon Willnauer 2014-04-08 11:11:56 +02:00
parent 0cebbf1624
commit fd8a6ac382
1 changed files with 15 additions and 6 deletions

21
src/test/java/org/elasticsearch/document/BulkTests.java

@ -569,21 +569,27 @@ public class BulkTests extends ElasticsearchIntegrationTest {
@Test @Test
public void testThatBulkProcessorCountIsCorrect() throws InterruptedException { public void testThatBulkProcessorCountIsCorrect() throws InterruptedException {
final BlockingQueue<BulkResponse> responseQueue = new SynchronousQueue(); final AtomicReference<BulkResponse> responseRef = new AtomicReference<>();
final AtomicReference<Throwable> failureRef = new AtomicReference<>();
final CountDownLatch latch = new CountDownLatch(1);
BulkProcessor.Listener listener = new BulkProcessor.Listener() { BulkProcessor.Listener listener = new BulkProcessor.Listener() {
@Override @Override
public void beforeBulk(long executionId, BulkRequest request) {} public void beforeBulk(long executionId, BulkRequest request) {}
@Override @Override
public void afterBulk(long executionId, BulkRequest request, BulkResponse response) { public void afterBulk(long executionId, BulkRequest request, BulkResponse response) {
responseQueue.add(response); responseRef.set(response);
latch.countDown();
} }
@Override @Override
public void afterBulk(long executionId, BulkRequest request, Throwable failure) {} public void afterBulk(long executionId, BulkRequest request, Throwable failure) {
failureRef.set(failure);
latch.countDown();
}
}; };
try (BulkProcessor processor = BulkProcessor.builder(client(), listener).setBulkActions(5) try (BulkProcessor processor = BulkProcessor.builder(client(), listener).setBulkActions(5)
.setConcurrentRequests(1).setName("foo").build()) { .setConcurrentRequests(1).setName("foo").build()) {
Map<String, Object> data = Maps.newHashMap(); Map<String, Object> data = Maps.newHashMap();
@ -595,8 +601,11 @@ public class BulkTests extends ElasticsearchIntegrationTest {
processor.add(new IndexRequest("test", "test", "4").source(data)); processor.add(new IndexRequest("test", "test", "4").source(data));
processor.add(new IndexRequest("test", "test", "5").source(data)); processor.add(new IndexRequest("test", "test", "5").source(data));
BulkResponse response = responseQueue.poll(5, TimeUnit.SECONDS); latch.await();
assertThat("Could not get a bulk response in 5 seconds", response, is(notNullValue())); BulkResponse response = responseRef.get();
Throwable error = failureRef.get();
assertThat(error, nullValue());
assertThat("Could not get a bulk response even after an explicit flush.", response, notNullValue());
assertThat(response.getItems().length, is(5)); assertThat(response.getItems().length, is(5));
} }
} }