honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Randomly use Bulk in AbstractIntegrationTest#indexRandom

This commit is contained in:
Simon Willnauer 2013-10-14 09:58:58 +02:00
parent 9dc59e29fa
commit 148a772ea0
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/test/java/org/elasticsearch/test/AbstractIntegrationTest.java
View File

@ -36,6 +36,8 @@ import org.elasticsearch.action.admin.indices.exists.indices.IndicesExistsRespon
import org.elasticsearch.action.admin.indices.flush.FlushResponse;
import org.elasticsearch.action.admin.indices.optimize.OptimizeResponse;
import org.elasticsearch.action.admin.indices.refresh.RefreshResponse;
import org.elasticsearch.action.bulk.BulkRequestBuilder;
import org.elasticsearch.action.bulk.BulkResponse;
import org.elasticsearch.action.get.GetResponse;
import org.elasticsearch.action.index.IndexRequestBuilder;
import org.elasticsearch.action.index.IndexResponse;
@ -510,7 +512,7 @@ public abstract class AbstractIntegrationTest extends ElasticsearchTestCase {
final CopyOnWriteArrayList<Tuple<IndexRequestBuilder, Throwable>> errors = new CopyOnWriteArrayList<Tuple<IndexRequestBuilder, Throwable>>();
List<CountDownLatch> latches = new ArrayList<CountDownLatch>();
if (frequently()) {
logger.info("Index [{}] docs async: [{}]", list.size(), true);
logger.info("Index [{}] docs async: [{}] bulk: [{}]", list.size(), true, false);
final CountDownLatch latch = new CountDownLatch(list.size());
latches.add(latch);
for (IndexRequestBuilder indexRequestBuilder : list) {
@ -526,8 +528,8 @@ public abstract class AbstractIntegrationTest extends ElasticsearchTestCase {
}
}
} else {
logger.info("Index [{}] docs async: [{}]", list.size(), false);
} else if (randomBoolean()) {
logger.info("Index [{}] docs async: [{}] bulk: [{}]", list.size(), false, false);
for (IndexRequestBuilder indexRequestBuilder : list) {
indexRequestBuilder.execute().actionGet();
if (rarely()) {
@ -540,6 +542,14 @@ public abstract class AbstractIntegrationTest extends ElasticsearchTestCase {
}
}
}
} else {
logger.info("Index [{}] docs async: [{}] bulk: [{}]", list.size(), false, true);
BulkRequestBuilder bulkBuilder = client().prepareBulk();
for (IndexRequestBuilder indexRequestBuilder : list) {
bulkBuilder.add(indexRequestBuilder);
}
BulkResponse actionGet = bulkBuilder.execute().actionGet();
assertThat(actionGet.hasFailures() ? actionGet.buildFailureMessage() : "", actionGet.hasFailures(), equalTo(false));
}
for (CountDownLatch countDownLatch : latches) {
countDownLatch.await();