Skip to content

Commit b1269cd

Browse files
committed
Fix ESQL response object
It provides now more values: * `documents_found` * `values_loaded`
1 parent eabba85 commit b1269cd

File tree

1 file changed

+4
-10
lines changed

1 file changed

+4
-10
lines changed

src/test/java/fr/pilato/test/elasticsearch/hlclient/EsClientIT.java

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -972,22 +972,16 @@ void esql() throws IOException, SQLException {
972972
{
973973
// Using the Raw ES|QL API
974974
try (final BinaryResponse response = client.esql().query(q -> q.query(query)); InputStream is = response.content()) {
975-
// The response object is {
976-
// "took" : 4,
977-
// "is_partial" : false,
978-
// "columns" : [ {
979-
// "name" : "name",
980-
// "type" : "text"
981-
// } ],
982-
// "values" : [ [ "David" ] ]
983-
//}
975+
// The response object is {"took":173,"is_partial":false,"documents_found":1,"values_loaded":1,"columns":[{"name":"name","type":"text"}],"values":[["David"]]}
984976
final ObjectMapper mapper = new ObjectMapper();
985977
final JsonNode jsonNode = mapper.readTree(is);
986-
assertThat(jsonNode).isNotNull().hasSize(4);
978+
assertThat(jsonNode).isNotNull().hasSize(6);
987979
assertThat(jsonNode.get("columns")).isNotNull().hasSize(1).first().satisfies(column -> assertThat(column.get("name").asText()).isEqualTo("name"));
988980
assertThat(jsonNode.get("values")).isNotNull().hasSize(1).first().satisfies(value -> assertThat(value).hasSize(1).first().satisfies(singleValue -> assertThat(singleValue.asText()).isEqualTo("David")));
989981
assertThat(jsonNode.get("took").asInt()).isGreaterThan(0);
990982
assertThat(jsonNode.get("is_partial").asBoolean()).isFalse();
983+
assertThat(jsonNode.get("documents_found").asLong()).isEqualTo(1);
984+
assertThat(jsonNode.get("values_loaded").asLong()).isEqualTo(1);
991985
}
992986
}
993987

0 commit comments

Comments
 (0)