1 files changed, 133 insertions, 0 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/config/SchemaInfoTester.java b/container-search/src/test/java/com/yahoo/search/config/SchemaInfoTester.java
new file mode 100644
index 00000000000..d5b4522f3aa
--- /dev/null
+++ b/container-search/src/test/java/com/yahoo/search/config/SchemaInfoTester.java
@@ -0,0 +1,133 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.search.config;
+
+import com.yahoo.container.QrSearchersConfig;
+import com.yahoo.prelude.fastsearch.DocumentdbInfoConfig;
+import com.yahoo.search.Query;
+import com.yahoo.tensor.TensorType;
+import com.yahoo.yolean.Exceptions;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * @author bratseth
+ */
+public class SchemaInfoTester {
+
+    private final SchemaInfo schemaInfo;
+
+    SchemaInfoTester() {
+        this.schemaInfo = createSchemaInfo();
+    }
+
+    SchemaInfo schemaInfo() { return schemaInfo; }
+
+    Query query(String sources, String restrict) {
+        Map<String, String> params = new HashMap<>();
+        if ( ! sources.isEmpty())
+            params.put("sources", sources);
+        if ( ! restrict.isEmpty())
+            params.put("restrict", restrict);
+        return new Query.Builder().setSchemaInfo(schemaInfo)
+                                  .setRequestMap(params)
+                                  .build();
+    }
+
+    void assertInput(TensorType expectedType, String sources, String restrict, String rankProfile, String feature) {
+        assertEquals(expectedType,
+                     schemaInfo.newSession(query(sources, restrict)).rankProfileInput(feature, rankProfile));
+    }
+
+    void assertInputConflict(TensorType expectedType, String sources, String restrict, String rankProfile, String feature) {
+        try {
+            assertInput(expectedType, sources, restrict, rankProfile, feature);
+        }
+        catch (IllegalArgumentException e) {
+            assertEquals("Conflicting input type declarations for '" + feature + "'",
+                         e.getMessage().split(":")[0]);
+        }
+    }
+
+    static SchemaInfo createSchemaInfo() {
+        List<Schema> schemas = new ArrayList<>();
+        RankProfile common = new RankProfile.Builder("commonProfile")
+                .addInput("query(myTensor1)", TensorType.fromSpec("tensor(a{},b{})"))
+                .addInput("query(myTensor2)", TensorType.fromSpec("tensor(x[2],y[2])"))
+                .addInput("query(myTensor3)", TensorType.fromSpec("tensor(x[2],y[2])"))
+                .addInput("query(myTensor4)", TensorType.fromSpec("tensor<float>(x[5])"))
+                .build();
+        schemas.add(new Schema.Builder("a")
+                            .add(common)
+                            .add(new RankProfile.Builder("inconsistent")
+                                         .addInput("query(myTensor1)", TensorType.fromSpec("tensor(a{},b{})"))
+                                         .build())
+                            .build());
+        schemas.add(new Schema.Builder("b")
+                            .add(common)
+                            .add(new RankProfile.Builder("inconsistent")
+                                         .addInput("query(myTensor1)", TensorType.fromSpec("tensor(x[10])"))
+                                         .build())
+                            .add(new RankProfile.Builder("bOnly")
+                                         .addInput("query(myTensor1)", TensorType.fromSpec("tensor(a{},b{})"))
+                                         .build())
+                            .build());
+        Map<String, List<String>> clusters = new HashMap<>();
+        clusters.put("ab", List.of("a", "b"));
+        clusters.put("a", List.of("a"));
+        return new SchemaInfo(schemas, clusters);
+    }
+
+    /** Creates the same schema info as createSchemaInfo from config objects. */
+    static SchemaInfo createSchemaInfoFromConfig() {
+        var indexInfoConfig = new IndexInfoConfig.Builder();
+
+        var rankProfileCommon = new DocumentdbInfoConfig.Documentdb.Rankprofile.Builder();
+        rankProfileCommon.name("commonProfile");
+        rankProfileCommon.input(new DocumentdbInfoConfig.Documentdb.Rankprofile.Input.Builder().name("query(myTensor1)").type("tensor(a{},b{})"));
+        rankProfileCommon.input(new DocumentdbInfoConfig.Documentdb.Rankprofile.Input.Builder().name("query(myTensor2)").type("tensor(x[2],y[2])"));
+        rankProfileCommon.input(new DocumentdbInfoConfig.Documentdb.Rankprofile.Input.Builder().name("query(myTensor3)").type("tensor(x[2],y[2])"));
+        rankProfileCommon.input(new DocumentdbInfoConfig.Documentdb.Rankprofile.Input.Builder().name("query(myTensor4)").type("tensor<float>(x[5])"));
+
+        var documentDbInfoInfoConfig = new DocumentdbInfoConfig.Builder();
+
+        var documentDbA = new DocumentdbInfoConfig.Documentdb.Builder();
+        documentDbA.name("a");
+        documentDbA.rankprofile(rankProfileCommon);
+        var rankProfileInconsistentA = new DocumentdbInfoConfig.Documentdb.Rankprofile.Builder();
+        rankProfileInconsistentA.name("inconsistent");
+        rankProfileInconsistentA.input(new DocumentdbInfoConfig.Documentdb.Rankprofile.Input.Builder().name("query(myTensor1)").type("tensor(a{},b{})"));
+        documentDbA.rankprofile(rankProfileInconsistentA);
+        documentDbInfoInfoConfig.documentdb(documentDbA);
+
+        var documentDbB = new DocumentdbInfoConfig.Documentdb.Builder();
+        documentDbB.name("b");
+        documentDbB.rankprofile(rankProfileCommon);
+        var rankProfileInconsistentB = new DocumentdbInfoConfig.Documentdb.Rankprofile.Builder();
+        rankProfileInconsistentB.name("inconsistent");
+        rankProfileInconsistentB.input(new DocumentdbInfoConfig.Documentdb.Rankprofile.Input.Builder().name("query(myTensor1)").type("tensor(x[10])"));
+        documentDbB.rankprofile(rankProfileInconsistentB);
+        var rankProfileBOnly = new DocumentdbInfoConfig.Documentdb.Rankprofile.Builder();
+        rankProfileBOnly.name("bOnly");
+        rankProfileBOnly.input(new DocumentdbInfoConfig.Documentdb.Rankprofile.Input.Builder().name("query(myTensor1)").type("tensor(a{},b{})"));
+        documentDbB.rankprofile(rankProfileBOnly);
+        documentDbInfoInfoConfig.documentdb(documentDbB);
+
+        var qrSearchersConfig = new QrSearchersConfig.Builder();
+        var clusterAB = new QrSearchersConfig.Searchcluster.Builder();
+        clusterAB.name("ab");
+        clusterAB.searchdef("a").searchdef("b");
+        qrSearchersConfig.searchcluster(clusterAB);
+        var clusterA = new QrSearchersConfig.Searchcluster.Builder();
+        clusterA.name("a");
+        clusterA.searchdef("a");
+        qrSearchersConfig.searchcluster(clusterA);
+
+        return new SchemaInfo(indexInfoConfig.build(), documentDbInfoInfoConfig.build(), qrSearchersConfig.build());
+    }
+
+}