1 files changed, 114 insertions, 0 deletions
diff --git a/searchlib/src/tests/query/query_visitor_test.cpp b/searchlib/src/tests/query/query_visitor_test.cpp
new file mode 100644
index 00000000000..b98e14604e3
--- /dev/null
+++ b/searchlib/src/tests/query/query_visitor_test.cpp
@@ -0,0 +1,114 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// Unit tests for query_visitor.
+
+#include <vespa/fastos/fastos.h>
+#include <vespa/log/log.h>
+LOG_SETUP("query_visitor_test");
+
+#include <vespa/searchlib/query/tree/intermediatenodes.h>
+#include <vespa/searchlib/query/tree/point.h>
+#include <vespa/searchlib/query/tree/queryvisitor.h>
+#include <vespa/searchlib/query/tree/simplequery.h>
+#include <vespa/searchlib/query/tree/termnodes.h>
+#include <vespa/vespalib/testkit/testapp.h>
+
+using namespace search::query;
+
+namespace {
+
+class Test : public vespalib::TestApp {
+    void requireThatAllNodesCanBeVisited();
+
+    template <class T> void checkVisit(T *node);
+
+public:
+    int Main();
+};
+
+int
+Test::Main()
+{
+    TEST_INIT("query_visitor_test");
+
+    TEST_DO(requireThatAllNodesCanBeVisited());
+
+    TEST_DONE();
+}
+
+class MyVisitor : public QueryVisitor
+{
+public:
+    template <typename T>
+    bool &isVisited() {
+        static bool b;
+        return b;
+    }
+
+    virtual void visit(And &) { isVisited<And>() = true; }
+    virtual void visit(AndNot &) { isVisited<AndNot>() = true; }
+    virtual void visit(Equiv &) { isVisited<Equiv>() = true; }
+    virtual void visit(NumberTerm &) { isVisited<NumberTerm>() = true; }
+    virtual void visit(LocationTerm &) { isVisited<LocationTerm>() = true; }
+    virtual void visit(Near &) { isVisited<Near>() = true; }
+    virtual void visit(ONear &) { isVisited<ONear>() = true; }
+    virtual void visit(Or &) { isVisited<Or>() = true; }
+    virtual void visit(Phrase &) { isVisited<Phrase>() = true; }
+    virtual void visit(PrefixTerm &) { isVisited<PrefixTerm>() = true; }
+    virtual void visit(RangeTerm &) { isVisited<RangeTerm>() = true; }
+    virtual void visit(Rank &) { isVisited<Rank>() = true; }
+    virtual void visit(StringTerm &) { isVisited<StringTerm>() = true; }
+    virtual void visit(SubstringTerm &) { isVisited<SubstringTerm>() = true; }
+    virtual void visit(SuffixTerm &) { isVisited<SuffixTerm>() = true; }
+    virtual void visit(WeakAnd &) { isVisited<WeakAnd>() = true; }
+    virtual void visit(WeightedSetTerm &)
+    { isVisited<WeightedSetTerm>() = true; }
+    virtual void visit(DotProduct &) { isVisited<DotProduct>() = true; }
+    virtual void visit(WandTerm &) { isVisited<WandTerm>() = true; }
+    virtual void visit(PredicateQuery &)
+    { isVisited<PredicateQuery>() = true; }
+    virtual void visit(RegExpTerm &) { isVisited<RegExpTerm>() = true; }
+};
+
+template <class T>
+void Test::checkVisit(T *node) {
+    Node::UP query(node);
+    MyVisitor visitor;
+    visitor.isVisited<T>() = false;
+    query->accept(visitor);
+    ASSERT_TRUE(visitor.isVisited<T>());
+}
+
+void Test::requireThatAllNodesCanBeVisited() {
+    checkVisit<And>(new SimpleAnd);
+    checkVisit<AndNot>(new SimpleAndNot);
+    checkVisit<Near>(new SimpleNear(0));
+    checkVisit<ONear>(new SimpleONear(0));
+    checkVisit<Or>(new SimpleOr);
+    checkVisit<Phrase>(new SimplePhrase("field", 0, Weight(42)));
+    checkVisit<WeightedSetTerm>(
+            new SimpleWeightedSetTerm("field", 0, Weight(42)));
+    checkVisit<DotProduct>(new SimpleDotProduct("field", 0, Weight(42)));
+    checkVisit<WandTerm>(
+            new SimpleWandTerm("field", 0, Weight(42), 57, 67, 77.7));
+    checkVisit<Rank>(new SimpleRank);
+    checkVisit<NumberTerm>(
+            new SimpleNumberTerm("0.42", "field", 0, Weight(0)));
+    const Location location(Point(10, 10), 20, 0);
+    checkVisit<LocationTerm>(
+            new SimpleLocationTerm(location, "field", 0, Weight(0)));
+    checkVisit<PrefixTerm>(new SimplePrefixTerm("t", "field", 0, Weight(0)));
+    checkVisit<RangeTerm>(
+            new SimpleRangeTerm(Range(0, 1), "field", 0, Weight(0)));
+    checkVisit<StringTerm>(new SimpleStringTerm("t", "field", 0, Weight(0)));
+    checkVisit<SubstringTerm>(
+            new SimpleSubstringTerm("t", "field", 0, Weight(0)));
+    checkVisit<SuffixTerm>(new SimpleSuffixTerm("t", "field", 0, Weight(0)));
+    checkVisit<PredicateQuery>(
+            new SimplePredicateQuery(PredicateQueryTerm::UP(),
+                                     "field", 0, Weight(0)));
+    checkVisit<RegExpTerm>(new SimpleRegExpTerm("t", "field", 0, Weight(0)));
+}
+
+}  // namespace
+
+TEST_APPHOOK(Test);