// 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);