|
22 | 22 | import co.elastic.clients.elasticsearch._types.mapping.TypeMapping;
|
23 | 23 | import co.elastic.clients.elasticsearch.cluster.HealthRequest;
|
24 | 24 | import co.elastic.clients.elasticsearch.cluster.HealthResponse;
|
| 25 | +import co.elastic.clients.elasticsearch.core.CountRequest; |
| 26 | +import co.elastic.clients.elasticsearch.core.CountResponse; |
25 | 27 | import co.elastic.clients.elasticsearch.core.IndexRequest;
|
26 | 28 | import co.elastic.clients.elasticsearch.core.IndexResponse;
|
27 | 29 | import co.elastic.clients.elasticsearch.core.SearchRequest;
|
|
62 | 64 | * on port 9200 and an intercepting proxy on port 8080.
|
63 | 65 | *
|
64 | 66 | * @author Peter-Josef Meisch
|
| 67 | + * @author maryantocinn |
65 | 68 | */
|
66 | 69 | @Disabled
|
67 | 70 | @TestMethodOrder(MethodOrderer.OrderAnnotation.class)
|
@@ -352,6 +355,43 @@ private ResponseBody<EntityAsMap> searchImperative(SearchRequest searchRequest)
|
352 | 355 | private ResponseBody<EntityAsMap> searchReactive(SearchRequest searchRequest) {
|
353 | 356 | return Objects.requireNonNull(reactiveElasticsearchClient.search(searchRequest, EntityAsMap.class).block());
|
354 | 357 | }
|
| 358 | + |
| 359 | + // endregion |
| 360 | + // region count |
| 361 | + @Test |
| 362 | + @Order(40) |
| 363 | + void count() { |
| 364 | + |
| 365 | + CountRequest countRequest = new CountRequest.Builder().index(INDEX) |
| 366 | + .query(query -> query.match(matchQuery -> matchQuery.field("content").query(FieldValue.of("content1")))) |
| 367 | + .build(); |
| 368 | + |
| 369 | + CountResponse countResponse = null; |
| 370 | + |
| 371 | + try { |
| 372 | + countResponse = countImperative(countRequest); |
| 373 | + assertThat(countResponse).isNotNull(); |
| 374 | + assertThat(countResponse.count()).isEqualTo(1); |
| 375 | + } catch (IOException e) { |
| 376 | + LOGGER.error("error", e); |
| 377 | + } |
| 378 | + |
| 379 | + try { |
| 380 | + countResponse = countReactive(countRequest); |
| 381 | + assertThat(countResponse).isNotNull(); |
| 382 | + assertThat(countResponse.count()).isEqualTo(1); |
| 383 | + } catch (Exception e) { |
| 384 | + LOGGER.error("error", e); |
| 385 | + } |
| 386 | + } |
| 387 | + |
| 388 | + private CountResponse countImperative(CountRequest countRequest) throws IOException { |
| 389 | + return imperativeElasticsearchClient.count(countRequest); |
| 390 | + } |
| 391 | + |
| 392 | + private CountResponse countReactive(CountRequest countRequest) { |
| 393 | + return Objects.requireNonNull(reactiveElasticsearchClient.count(countRequest).block()); |
| 394 | + } |
355 | 395 | // endregion
|
356 | 396 |
|
357 | 397 | private ClientConfiguration clientConfiguration() {
|
|
0 commit comments