1 files changed, 120 insertions, 0 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/searchers/test/ValidateMatchPhaseSearcherTestCase.java b/container-search/src/test/java/com/yahoo/search/searchers/test/ValidateMatchPhaseSearcherTestCase.java
new file mode 100644
index 00000000000..4f7654ba3c0
--- /dev/null
+++ b/container-search/src/test/java/com/yahoo/search/searchers/test/ValidateMatchPhaseSearcherTestCase.java
@@ -0,0 +1,120 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.search.searchers.test;
+
+import com.yahoo.component.chain.Chain;
+import com.yahoo.config.subscription.ConfigGetter;
+import com.yahoo.config.subscription.RawSource;
+import com.yahoo.language.Linguistics;
+import com.yahoo.language.simple.SimpleLinguistics;
+import com.yahoo.search.Searcher;
+import com.yahoo.search.rendering.RendererRegistry;
+import com.yahoo.search.searchchain.Execution;
+import com.yahoo.search.searchers.ValidateMatchPhaseSearcher;
+import com.yahoo.search.Query;
+import com.yahoo.search.Result;
+import com.yahoo.vespa.config.search.AttributesConfig;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.junit.Assert.*;
+
+/**
+ * @author balder
+ */
+public class ValidateMatchPhaseSearcherTestCase {
+
+    private ValidateMatchPhaseSearcher searcher;
+
+    public ValidateMatchPhaseSearcherTestCase() {
+        searcher = new ValidateMatchPhaseSearcher(
+                ConfigGetter.getConfig(AttributesConfig.class,
+                                       "raw:",
+                                       new RawSource("attribute[4]\n" +
+                                                     "attribute[0].name                ok\n" +
+                                                     "attribute[0].datatype            INT32\n" +
+                                                     "attribute[0].collectiontype      SINGLE\n" +
+                                                     "attribute[0].fastsearch          true\n" +
+                                                     "attribute[1].name                not_fast\n" +
+                                                     "attribute[1].datatype            INT32\n" +
+                                                     "attribute[1].collectiontype      SINGLE\n" +
+                                                     "attribute[1].fastsearch          false\n" +
+                                                     "attribute[2].name                not_numeric\n" +
+                                                     "attribute[2].datatype            STRING\n" +
+                                                     "attribute[2].collectiontype      SINGLE\n" +
+                                                     "attribute[2].fastsearch          true\n" +
+                                                     "attribute[3].name                not_single\n" +
+                                                     "attribute[3].datatype            INT32\n" +
+                                                     "attribute[3].collectiontype      ARRAY\n" +
+                                                     "attribute[3].fastsearch          true"
+        )));
+    }
+
+    private static String getErrorMatch(String attribute) {
+        return "4: Invalid query parameter: The attribute '" +
+                attribute +
+                "' is not available for match-phase. It must be a single value numeric attribute with fast-search.";
+    }
+
+    private static String getErrorDiversity(String attribute) {
+        return "4: Invalid query parameter: The attribute '" +
+                attribute +
+                "' is not available for match-phase diversification. It must be a single value numeric or string attribute.";
+    }
+
+    @Test
+    public void testMatchPhaseAttribute() {
+        assertEquals("", search(""));
+        assertEquals("", match("ok"));
+        assertEquals(getErrorMatch("not_numeric"), match("not_numeric"));
+        assertEquals(getErrorMatch("not_single"), match("not_single"));
+        assertEquals(getErrorMatch("not_fast"), match("not_fast"));
+        assertEquals(getErrorMatch("not_found"), match("not_found"));
+    }
+
+    @Test
+    public void testDiversityAttribute() {
+        assertEquals("", search(""));
+        assertEquals("", diversify("ok"));
+        assertEquals("", diversify("not_numeric"));
+        assertEquals(getErrorDiversity("not_single"), diversify("not_single"));
+        assertEquals("", diversify("not_fast"));
+        assertEquals(getErrorDiversity("not_found"), diversify("not_found"));
+    }
+
+    private String match(String m) {
+        return search("&ranking.matchPhase.attribute=" + m);
+    }
+
+    private String diversify(String m) {
+        return search("&ranking.matchPhase.attribute=ok&ranking.matchPhase.diversity.attribute=" + m);
+    }
+
+    private String search(String m) {
+        String q = "/?query=sddocname:test" + m;
+        Result r = doSearch(searcher, new Query(q), 0, 10);
+        if (r.hits().getError() != null) {
+            return r.hits().getError().toString();
+        }
+        return "";
+    }
+
+    private Result doSearch(Searcher searcher, Query query, int offset, int hits) {
+        query.setOffset(offset);
+        query.setHits(hits);
+        return createExecution(searcher).search(query);
+    }
+
+    private Execution createExecution(Searcher searcher) {
+        Execution.Context context = new Execution.Context(null, null, null, new RendererRegistry(), new SimpleLinguistics());
+        return new Execution(chainedAsSearchChain(searcher), context);
+    }
+
+    private Chain<Searcher> chainedAsSearchChain(Searcher topOfChain) {
+        List<Searcher> searchers = new ArrayList<>();
+        searchers.add(topOfChain);
+        return new Chain<>(searchers);
+    }
+
+}