honeymoose/OpenSearch
1
0
Fork 0
mirror of https://github.com/honeymoose/OpenSearch.git synced 2025-03-30 03:48:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

add async request unit test

Browse Source
This commit is contained in:
javanna 2016-07-22 19:01:56 +02:00 committed by Luca Cavanna
parent 4e8ee1f0ab
commit e6054a931e
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
client/rest/src/test/java/org/elasticsearch/client/RestClientIntegTests.java
View File

@ -48,11 +48,14 @@ import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import static org.elasticsearch.client.RestClientTestUtil.getAllStatusCodes; import static org.elasticsearch.client.RestClientTestUtil.getAllStatusCodes;
import static org.elasticsearch.client.RestClientTestUtil.getHttpMethods; import static org.elasticsearch.client.RestClientTestUtil.getHttpMethods;
import static org.elasticsearch.client.RestClientTestUtil.randomStatusCode; import static org.elasticsearch.client.RestClientTestUtil.randomStatusCode;
import static org.hamcrest.CoreMatchers.equalTo; import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.CoreMatchers.instanceOf;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat; import static org.junit.Assert.assertThat;
@ -215,4 +218,34 @@ public class RestClientIntegTests extends RestClientTestCase {
assertEquals(statusCode, esResponse.getStatusLine().getStatusCode()); assertEquals(statusCode, esResponse.getStatusLine().getStatusCode());
assertEquals(requestBody, EntityUtils.toString(esResponse.getEntity())); assertEquals(requestBody, EntityUtils.toString(esResponse.getEntity()));
} }
public void testAsyncRequests() throws Exception {
int numRequests = randomIntBetween(2, 10);
final CountDownLatch latch = new CountDownLatch(numRequests);
for (int i = 0; i < numRequests; i++) {
final String method = RestClientTestUtil.randomHttpMethod(getRandom());
final int statusCode = randomStatusCode(getRandom());
restClient.performRequest(method, "/" + statusCode, new ResponseListener() {
@Override
public void onSuccess(Response response) {
latch.countDown();
assertResponse(response);
}
@Override
public void onFailure(Exception exception) {
latch.countDown();
assertThat(exception, instanceOf(ResponseException.class));
ResponseException responseException = (ResponseException) exception;
assertResponse(responseException.getResponse());
}
private void assertResponse(Response response) {
assertEquals(method, response.getRequestLine().getMethod());
assertEquals(statusCode, response.getStatusLine().getStatusCode());
}
});
}
assertTrue(latch.await(5, TimeUnit.SECONDS));
}
} }