1 files changed, 111 insertions, 0 deletions
diff --git a/config-model/src/test/java/com/yahoo/searchdefinition/DiversityTestCase.java b/config-model/src/test/java/com/yahoo/searchdefinition/DiversityTestCase.java
new file mode 100644
index 00000000000..6aebe765bc4
--- /dev/null
+++ b/config-model/src/test/java/com/yahoo/searchdefinition/DiversityTestCase.java
@@ -0,0 +1,111 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.searchdefinition;
+
+import com.yahoo.config.model.application.provider.BaseDeployLogger;
+import com.yahoo.search.query.ranking.Diversity;
+import com.yahoo.searchdefinition.parser.ParseException;
+import org.junit.Test;
+import static org.junit.Assert.fail;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * Created by balder on 3/10/15.
+ */
+public class DiversityTestCase {
+    @Test
+    public void testDiversity() throws ParseException {
+        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
+        SearchBuilder builder = new SearchBuilder(rankProfileRegistry);
+        builder.importString(
+                "search test {\n" +
+                        "    document test { \n" +
+                        "        field a type int { \n" +
+                        "            indexing: attribute \n" +
+                        "            attribute: fast-search\n" +
+                        "        }\n" +
+                        "        field b type int {\n" +
+                        "            indexing: attribute \n" +
+                        "        }\n" +
+                        "    }\n" +
+                        "    \n" +
+                        "    rank-profile parent {\n" +
+                        "        match-phase {\n" +
+                        "            diversity {\n" +
+                        "                attribute: b\n" +
+                        "                min-groups: 74\n" +
+                        "                cutoff-factor: 17.3\n" +
+                        "                cutoff-strategy: strict" +
+                        "            }\n" +
+                        "            attribute: a\n" +
+                        "            max-hits: 120\n" +
+                        "            max-filter-coverage: 0.065" +
+                        "        }\n" +
+                        "    }\n" +
+                        "}\n");
+        builder.build();
+        Search s = builder.getSearch();
+        RankProfile.MatchPhaseSettings matchPhase = rankProfileRegistry.getRankProfile(s, "parent").getMatchPhaseSettings();
+        RankProfile.DiversitySettings diversity = matchPhase.getDiversity();
+        assertEquals("b", diversity.getAttribute());
+        assertEquals(74, diversity.getMinGroups());
+        assertEquals(17.3, diversity.getCutoffFactor(), 1e-16);
+        assertEquals(Diversity.CutoffStrategy.strict, diversity.getCutoffStrategy());
+        assertEquals(120, matchPhase.getMaxHits());
+        assertEquals("a", matchPhase.getAttribute());
+        assertEquals(0.065, matchPhase.getMaxFilterCoverage(), 1e-16);
+    }
+
+    private static String getMessagePrefix() {
+        return "In search definition 'test', rank-profile 'parent': diversity attribute 'b' ";
+    }
+    @Test
+    public void requireSingleNumericOrString() throws ParseException {
+        SearchBuilder builder = getSearchBuilder("field b type predicate { indexing: attribute }");
+
+        try {
+            builder.build();
+            fail("Should throw.");
+        } catch (IllegalArgumentException e) {
+            assertEquals(getMessagePrefix() + "must be single value numeric, or enumerated attribute, but it is 'predicate'", e.getMessage());
+        }
+    }
+
+    @Test
+    public void requireSingle() throws ParseException {
+        SearchBuilder builder = getSearchBuilder("field b type array<int> { indexing: attribute }");
+
+        try {
+            builder.build();
+            fail("Should throw.");
+        } catch (IllegalArgumentException e) {
+            assertEquals(getMessagePrefix() + "must be single value numeric, or enumerated attribute, but it is 'Array<int>'", e.getMessage());
+        }
+    }
+    private SearchBuilder getSearchBuilder(String diversity) throws ParseException {
+        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
+        SearchBuilder builder = new SearchBuilder(rankProfileRegistry);
+        builder.importString(
+                "search test {\n" +
+                        "    document test { \n" +
+                        "        field a type int { \n" +
+                        "            indexing: attribute \n" +
+                        "            attribute: fast-search\n" +
+                        "        }\n" +
+                        diversity +
+                        "    }\n" +
+                        "    \n" +
+                        "    rank-profile parent {\n" +
+                        "        match-phase {\n" +
+                        "            diversity {\n" +
+                        "                attribute: b\n" +
+                        "                min-groups: 74\n" +
+                        "            }\n" +
+                        "            attribute: a\n" +
+                        "            max-hits: 120\n" +
+                        "        }\n" +
+                        "    }\n" +
+                        "}\n");
+        return builder;
+    }
+}