1 files changed, 84 insertions, 0 deletions
diff --git a/container-search/src/test/java/com/yahoo/prelude/semantics/test/RuleBaseAbstractTestCase.java b/container-search/src/test/java/com/yahoo/prelude/semantics/test/RuleBaseAbstractTestCase.java
new file mode 100644
index 00000000000..102f4c95926
--- /dev/null
+++ b/container-search/src/test/java/com/yahoo/prelude/semantics/test/RuleBaseAbstractTestCase.java
@@ -0,0 +1,84 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.prelude.semantics.test;
+
+import com.yahoo.component.chain.Chain;
+import com.yahoo.language.Linguistics;
+import com.yahoo.language.simple.SimpleLinguistics;
+import com.yahoo.search.Query;
+import com.yahoo.prelude.semantics.RuleBase;
+import com.yahoo.prelude.semantics.RuleBaseException;
+import com.yahoo.prelude.semantics.SemanticSearcher;
+import com.yahoo.search.Searcher;
+import com.yahoo.search.rendering.RendererRegistry;
+import com.yahoo.search.searchchain.Execution;
+import com.yahoo.search.test.QueryTestCase;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Tests semantic searching
+ *
+ * @author bratseth
+ */
+@SuppressWarnings("deprecation")
+public abstract class RuleBaseAbstractTestCase extends junit.framework.TestCase {
+
+    protected final String root="src/test/java/com/yahoo/prelude/semantics/test/rulebases/";
+    protected final SemanticSearcher searcher;
+
+    protected RuleBaseAbstractTestCase(String name,String ruleBaseName) {
+        this(name,ruleBaseName,null);
+    }
+
+    protected RuleBaseAbstractTestCase(String name,String ruleBaseName,String automataFileName) {
+        super(name);
+        searcher = createSearcher(ruleBaseName,automataFileName);
+    }
+
+    public void setUp() {
+    }
+
+    protected SemanticSearcher createSearcher(String ruleBaseName,String automataFileName) {
+        try {
+            if (automataFileName!=null)
+                automataFileName=root + automataFileName;
+            RuleBase ruleBase = RuleBase.createFromFile(root + ruleBaseName,automataFileName);
+            return new SemanticSearcher(ruleBase);
+        } catch (Exception e) {
+            throw new RuleBaseException("Initialization of rule base '" + ruleBaseName + "' failed",e);
+        }
+    }
+
+    protected Query assertSemantics(String result, String input) {
+        return assertSemantics(result, input, 0);
+    }
+
+    protected Query assertSemantics(String result, String input, int tracelevel) {
+        return assertSemantics(result, input, tracelevel, Query.Type.ALL);
+    }
+
+    protected Query assertSemantics(String result, String input, int tracelevel, Query.Type queryType) {
+        Query query=new Query("?query=" + QueryTestCase.httpEncode(input) + "&tracelevel=0&tracelevel.rules=" + tracelevel +
+                             "&language=und&type=" + queryType.toString());
+        return assertSemantics(result, query);
+    }
+
+    protected Query assertSemantics(String result, Query query) {
+        createExecution(searcher).search(query);
+        assertEquals(result, query.getModel().getQueryTree().getRoot().toString());
+        return 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);
+    }
+
+}