|
@@ -21,20 +21,6 @@ package org.elasticsearch.client;
|
|
|
|
|
|
import com.fasterxml.jackson.core.JsonParseException;
|
|
|
|
|
|
-import org.elasticsearch.Build;
|
|
|
-import org.elasticsearch.ElasticsearchException;
|
|
|
-import org.elasticsearch.Version;
|
|
|
-import org.elasticsearch.action.ActionListener;
|
|
|
-import org.elasticsearch.action.ActionRequest;
|
|
|
-import org.elasticsearch.action.ActionRequestValidationException;
|
|
|
-import org.elasticsearch.action.main.MainRequest;
|
|
|
-import org.elasticsearch.action.main.MainResponse;
|
|
|
-import org.elasticsearch.action.search.ClearScrollRequest;
|
|
|
-import org.elasticsearch.action.search.ClearScrollResponse;
|
|
|
-import org.elasticsearch.action.search.SearchResponse;
|
|
|
-import org.elasticsearch.action.search.SearchResponseSections;
|
|
|
-import org.elasticsearch.action.search.SearchScrollRequest;
|
|
|
-import org.elasticsearch.action.search.ShardSearchFailure;
|
|
|
import org.apache.http.Header;
|
|
|
import org.apache.http.HttpEntity;
|
|
|
import org.apache.http.HttpHost;
|
|
@@ -49,6 +35,20 @@ import org.apache.http.message.BasicHttpResponse;
|
|
|
import org.apache.http.message.BasicRequestLine;
|
|
|
import org.apache.http.message.BasicStatusLine;
|
|
|
import org.apache.http.nio.entity.NStringEntity;
|
|
|
+import org.elasticsearch.Build;
|
|
|
+import org.elasticsearch.ElasticsearchException;
|
|
|
+import org.elasticsearch.Version;
|
|
|
+import org.elasticsearch.action.ActionListener;
|
|
|
+import org.elasticsearch.action.ActionRequest;
|
|
|
+import org.elasticsearch.action.ActionRequestValidationException;
|
|
|
+import org.elasticsearch.action.main.MainRequest;
|
|
|
+import org.elasticsearch.action.main.MainResponse;
|
|
|
+import org.elasticsearch.action.search.ClearScrollRequest;
|
|
|
+import org.elasticsearch.action.search.ClearScrollResponse;
|
|
|
+import org.elasticsearch.action.search.SearchResponse;
|
|
|
+import org.elasticsearch.action.search.SearchResponseSections;
|
|
|
+import org.elasticsearch.action.search.SearchScrollRequest;
|
|
|
+import org.elasticsearch.action.search.ShardSearchFailure;
|
|
|
import org.elasticsearch.cluster.ClusterName;
|
|
|
import org.elasticsearch.common.CheckedFunction;
|
|
|
import org.elasticsearch.common.xcontent.NamedXContentRegistry;
|
|
@@ -57,6 +57,10 @@ import org.elasticsearch.common.xcontent.XContentBuilder;
|
|
|
import org.elasticsearch.common.xcontent.XContentParser;
|
|
|
import org.elasticsearch.common.xcontent.cbor.CborXContent;
|
|
|
import org.elasticsearch.common.xcontent.smile.SmileXContent;
|
|
|
+import org.elasticsearch.index.rankeval.DiscountedCumulativeGain;
|
|
|
+import org.elasticsearch.index.rankeval.EvaluationMetric;
|
|
|
+import org.elasticsearch.index.rankeval.MeanReciprocalRank;
|
|
|
+import org.elasticsearch.index.rankeval.PrecisionAtK;
|
|
|
import org.elasticsearch.join.aggregations.ChildrenAggregationBuilder;
|
|
|
import org.elasticsearch.rest.RestStatus;
|
|
|
import org.elasticsearch.search.SearchHits;
|
|
@@ -648,7 +652,7 @@ public class RestHighLevelClientTests extends ESTestCase {
|
|
|
|
|
|
public void testProvidedNamedXContents() {
|
|
|
List<NamedXContentRegistry.Entry> namedXContents = RestHighLevelClient.getProvidedNamedXContents();
|
|
|
- assertEquals(2, namedXContents.size());
|
|
|
+ assertEquals(5, namedXContents.size());
|
|
|
Map<Class<?>, Integer> categories = new HashMap<>();
|
|
|
List<String> names = new ArrayList<>();
|
|
|
for (NamedXContentRegistry.Entry namedXContent : namedXContents) {
|
|
@@ -658,10 +662,14 @@ public class RestHighLevelClientTests extends ESTestCase {
|
|
|
categories.put(namedXContent.categoryClass, counter + 1);
|
|
|
}
|
|
|
}
|
|
|
- assertEquals(1, categories.size());
|
|
|
+ assertEquals(2, categories.size());
|
|
|
assertEquals(Integer.valueOf(2), categories.get(Aggregation.class));
|
|
|
assertTrue(names.contains(ChildrenAggregationBuilder.NAME));
|
|
|
assertTrue(names.contains(MatrixStatsAggregationBuilder.NAME));
|
|
|
+ assertEquals(Integer.valueOf(3), categories.get(EvaluationMetric.class));
|
|
|
+ assertTrue(names.contains(PrecisionAtK.NAME));
|
|
|
+ assertTrue(names.contains(DiscountedCumulativeGain.NAME));
|
|
|
+ assertTrue(names.contains(MeanReciprocalRank.NAME));
|
|
|
}
|
|
|
|
|
|
private static class TrackingActionListener implements ActionListener<Integer> {
|