//===-- QueryEngineTest.cpp - clang-query test ----------------------------===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// #include "Query.h" #include "QueryParser.h" #include "QuerySession.h" #include "clang/ASTMatchers/ASTMatchers.h" #include "clang/ASTMatchers/Dynamic/VariantValue.h" #include "clang/Frontend/ASTUnit.h" #include "clang/Tooling/Tooling.h" #include "llvm/ADT/STLExtras.h" #include "llvm/Support/raw_ostream.h" #include "gtest/gtest.h" #include <string> using namespace clang; using namespace clang::ast_matchers; using namespace clang::ast_matchers::dynamic; using namespace clang::query; using namespace clang::tooling; class QueryEngineTest : public ::testing::Test { ArrayRef<std::unique_ptr<ASTUnit>> mkASTUnit2(std::unique_ptr<ASTUnit> a, std::unique_ptr<ASTUnit> b) { ASTs[0] = std::move(a); ASTs[1] = std::move(b); return ArrayRef<std::unique_ptr<ASTUnit>>(ASTs); } protected: QueryEngineTest() : S(mkASTUnit2(buildASTFromCode("void foo1(void) {}\nvoid foo2(void) {}", "foo.cc"), buildASTFromCode("void bar1(void) {}\nvoid bar2(void) {}", "bar.cc"))), OS(Str) {} std::unique_ptr<ASTUnit> ASTs[2]; QuerySession S; std::string Str; llvm::raw_string_ostream OS; }; TEST_F(QueryEngineTest, Basic) { DynTypedMatcher FnMatcher = functionDecl(); DynTypedMatcher FooMatcher = functionDecl(hasName("foo1")); std::string FooMatcherString = "functionDecl(hasName(\"foo1\"))"; EXPECT_TRUE(NoOpQuery().run(OS, S)); EXPECT_EQ("", OS.str()); Str.clear(); EXPECT_FALSE(InvalidQuery("Parse error").run(OS, S)); EXPECT_EQ("Parse error\n", OS.str()); Str.clear(); EXPECT_TRUE(HelpQuery().run(OS, S)); EXPECT_TRUE(OS.str().find("Available commands:") != std::string::npos); Str.clear(); EXPECT_TRUE(MatchQuery("functionDecl()", FnMatcher).run(OS, S)); EXPECT_TRUE(OS.str().find("foo.cc:1:1: note: \"root\" binds here") != std::string::npos); EXPECT_TRUE(OS.str().find("foo.cc:2:1: note: \"root\" binds here") != std::string::npos); EXPECT_TRUE(OS.str().find("bar.cc:1:1: note: \"root\" binds here") != std::string::npos); EXPECT_TRUE(OS.str().find("bar.cc:2:1: note: \"root\" binds here") != std::string::npos); EXPECT_TRUE(OS.str().find("4 matches.") != std::string::npos); Str.clear(); EXPECT_TRUE(MatchQuery(FooMatcherString, FooMatcher).run(OS, S)); EXPECT_TRUE(OS.str().find("foo.cc:1:1: note: \"root\" binds here") != std::string::npos); EXPECT_TRUE(OS.str().find("1 match.") != std::string::npos); Str.clear(); EXPECT_TRUE( SetExclusiveOutputQuery(&QuerySession::PrintOutput).run(OS, S)); EXPECT_TRUE(MatchQuery(FooMatcherString, FooMatcher).run(OS, S)); EXPECT_TRUE(OS.str().find("Binding for \"root\":\nvoid foo1()") != std::string::npos); Str.clear(); EXPECT_TRUE( SetExclusiveOutputQuery(&QuerySession::DetailedASTOutput).run(OS, S)); EXPECT_TRUE(MatchQuery(FooMatcherString, FooMatcher).run(OS, S)); EXPECT_TRUE(OS.str().find("FunctionDecl") != std::string::npos); Str.clear(); EXPECT_TRUE(EnableOutputQuery(&QuerySession::DiagOutput).run(OS, S)); EXPECT_TRUE(EnableOutputQuery(&QuerySession::DetailedASTOutput).run(OS, S)); EXPECT_TRUE(MatchQuery(FooMatcherString, FooMatcher).run(OS, S)); { auto Output = OS.str(); EXPECT_TRUE(Output.find("FunctionDecl") != std::string::npos); EXPECT_TRUE(Output.find("foo.cc:1:1: note: \"root\" binds here") != std::string::npos); } Str.clear(); EXPECT_TRUE(SetQuery<bool>(&QuerySession::BindRoot, false).run(OS, S)); EXPECT_TRUE(MatchQuery(FooMatcherString, FooMatcher).run(OS, S)); EXPECT_TRUE(OS.str().find("No bindings.") != std::string::npos); Str.clear(); EXPECT_FALSE(MatchQuery("isMain()", isMain()).run(OS, S)); EXPECT_EQ("Not a valid top-level matcher.\n", OS.str()); } TEST_F(QueryEngineTest, LetAndMatch) { EXPECT_TRUE(QueryParser::parse("let x \"foo1\"", S)->run(OS, S)); EXPECT_EQ("", OS.str()); Str.clear(); EXPECT_TRUE(QueryParser::parse("let y hasName(x)", S)->run(OS, S)); EXPECT_EQ("", OS.str()); Str.clear(); EXPECT_TRUE(QueryParser::parse("match functionDecl(y)", S)->run(OS, S)); EXPECT_TRUE(OS.str().find("foo.cc:1:1: note: \"root\" binds here") != std::string::npos); EXPECT_TRUE(OS.str().find("1 match.") != std::string::npos); Str.clear(); EXPECT_TRUE(QueryParser::parse("unlet x", S)->run(OS, S)); EXPECT_EQ("", OS.str()); Str.clear(); EXPECT_FALSE(QueryParser::parse("let y hasName(x)", S)->run(OS, S)); EXPECT_EQ("1:2: Error parsing argument 1 for matcher hasName.\n" "1:10: Value not found: x\n", OS.str()); Str.clear(); }