238 lines
8.5 KiB
C++
238 lines
8.5 KiB
C++
//===- unittest/ASTMatchers/Dynamic/ParserTest.cpp - Parser unit tests -===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===-------------------------------------------------------------------===//
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "../ASTMatchersTest.h"
|
|
#include "clang/ASTMatchers/Dynamic/Parser.h"
|
|
#include "clang/ASTMatchers/Dynamic/Registry.h"
|
|
#include "gtest/gtest.h"
|
|
#include "llvm/ADT/StringMap.h"
|
|
|
|
namespace clang {
|
|
namespace ast_matchers {
|
|
namespace dynamic {
|
|
namespace {
|
|
|
|
class MockSema : public Parser::Sema {
|
|
public:
|
|
virtual ~MockSema() {}
|
|
|
|
uint64_t expectMatcher(StringRef MatcherName) {
|
|
ast_matchers::internal::Matcher<Stmt> M = stmt();
|
|
ExpectedMatchers.insert(std::make_pair(MatcherName, M));
|
|
return M.getID();
|
|
}
|
|
|
|
void parse(StringRef Code) {
|
|
Diagnostics Error;
|
|
VariantValue Value;
|
|
Parser::parseExpression(Code, this, &Value, &Error);
|
|
Values.push_back(Value);
|
|
Errors.push_back(Error.toStringFull());
|
|
}
|
|
|
|
VariantMatcher actOnMatcherExpression(StringRef MatcherName,
|
|
const SourceRange &NameRange,
|
|
StringRef BindID,
|
|
ArrayRef<ParserValue> Args,
|
|
Diagnostics *Error) {
|
|
MatcherInfo ToStore = { MatcherName, NameRange, Args, BindID };
|
|
Matchers.push_back(ToStore);
|
|
return VariantMatcher::SingleMatcher(
|
|
ExpectedMatchers.find(MatcherName)->second);
|
|
}
|
|
|
|
struct MatcherInfo {
|
|
StringRef MatcherName;
|
|
SourceRange NameRange;
|
|
std::vector<ParserValue> Args;
|
|
std::string BoundID;
|
|
};
|
|
|
|
std::vector<std::string> Errors;
|
|
std::vector<VariantValue> Values;
|
|
std::vector<MatcherInfo> Matchers;
|
|
std::map<std::string, ast_matchers::internal::Matcher<Stmt> >
|
|
ExpectedMatchers;
|
|
};
|
|
|
|
TEST(ParserTest, ParseUnsigned) {
|
|
MockSema Sema;
|
|
Sema.parse("0");
|
|
Sema.parse("123");
|
|
Sema.parse("0x1f");
|
|
Sema.parse("12345678901");
|
|
Sema.parse("1a1");
|
|
EXPECT_EQ(5U, Sema.Values.size());
|
|
EXPECT_EQ(0U, Sema.Values[0].getUnsigned());
|
|
EXPECT_EQ(123U, Sema.Values[1].getUnsigned());
|
|
EXPECT_EQ(31U, Sema.Values[2].getUnsigned());
|
|
EXPECT_EQ("1:1: Error parsing unsigned token: <12345678901>", Sema.Errors[3]);
|
|
EXPECT_EQ("1:1: Error parsing unsigned token: <1a1>", Sema.Errors[4]);
|
|
}
|
|
|
|
TEST(ParserTest, ParseString) {
|
|
MockSema Sema;
|
|
Sema.parse("\"Foo\"");
|
|
Sema.parse("\"\"");
|
|
Sema.parse("\"Baz");
|
|
EXPECT_EQ(3ULL, Sema.Values.size());
|
|
EXPECT_EQ("Foo", Sema.Values[0].getString());
|
|
EXPECT_EQ("", Sema.Values[1].getString());
|
|
EXPECT_EQ("1:1: Error parsing string token: <\"Baz>", Sema.Errors[2]);
|
|
}
|
|
|
|
bool matchesRange(const SourceRange &Range, unsigned StartLine,
|
|
unsigned EndLine, unsigned StartColumn, unsigned EndColumn) {
|
|
EXPECT_EQ(StartLine, Range.Start.Line);
|
|
EXPECT_EQ(EndLine, Range.End.Line);
|
|
EXPECT_EQ(StartColumn, Range.Start.Column);
|
|
EXPECT_EQ(EndColumn, Range.End.Column);
|
|
return Range.Start.Line == StartLine && Range.End.Line == EndLine &&
|
|
Range.Start.Column == StartColumn && Range.End.Column == EndColumn;
|
|
}
|
|
|
|
llvm::Optional<DynTypedMatcher> getSingleMatcher(const VariantValue &Value) {
|
|
llvm::Optional<DynTypedMatcher> Result =
|
|
Value.getMatcher().getSingleMatcher();
|
|
EXPECT_TRUE(Result.hasValue());
|
|
return Result;
|
|
}
|
|
|
|
TEST(ParserTest, ParseMatcher) {
|
|
MockSema Sema;
|
|
const uint64_t ExpectedFoo = Sema.expectMatcher("Foo");
|
|
const uint64_t ExpectedBar = Sema.expectMatcher("Bar");
|
|
const uint64_t ExpectedBaz = Sema.expectMatcher("Baz");
|
|
Sema.parse(" Foo ( Bar ( 17), Baz( \n \"B A,Z\") ) .bind( \"Yo!\") ");
|
|
for (size_t i = 0, e = Sema.Errors.size(); i != e; ++i) {
|
|
EXPECT_EQ("", Sema.Errors[i]);
|
|
}
|
|
|
|
EXPECT_EQ(1ULL, Sema.Values.size());
|
|
EXPECT_EQ(ExpectedFoo, getSingleMatcher(Sema.Values[0])->getID());
|
|
|
|
EXPECT_EQ(3ULL, Sema.Matchers.size());
|
|
const MockSema::MatcherInfo Bar = Sema.Matchers[0];
|
|
EXPECT_EQ("Bar", Bar.MatcherName);
|
|
EXPECT_TRUE(matchesRange(Bar.NameRange, 1, 1, 8, 17));
|
|
EXPECT_EQ(1ULL, Bar.Args.size());
|
|
EXPECT_EQ(17U, Bar.Args[0].Value.getUnsigned());
|
|
|
|
const MockSema::MatcherInfo Baz = Sema.Matchers[1];
|
|
EXPECT_EQ("Baz", Baz.MatcherName);
|
|
EXPECT_TRUE(matchesRange(Baz.NameRange, 1, 2, 19, 10));
|
|
EXPECT_EQ(1ULL, Baz.Args.size());
|
|
EXPECT_EQ("B A,Z", Baz.Args[0].Value.getString());
|
|
|
|
const MockSema::MatcherInfo Foo = Sema.Matchers[2];
|
|
EXPECT_EQ("Foo", Foo.MatcherName);
|
|
EXPECT_TRUE(matchesRange(Foo.NameRange, 1, 2, 2, 12));
|
|
EXPECT_EQ(2ULL, Foo.Args.size());
|
|
EXPECT_EQ(ExpectedBar, getSingleMatcher(Foo.Args[0].Value)->getID());
|
|
EXPECT_EQ(ExpectedBaz, getSingleMatcher(Foo.Args[1].Value)->getID());
|
|
EXPECT_EQ("Yo!", Foo.BoundID);
|
|
}
|
|
|
|
using ast_matchers::internal::Matcher;
|
|
|
|
TEST(ParserTest, FullParserTest) {
|
|
Diagnostics Error;
|
|
llvm::Optional<DynTypedMatcher> VarDecl(Parser::parseMatcherExpression(
|
|
"varDecl(hasInitializer(binaryOperator(hasLHS(integerLiteral()),"
|
|
" hasOperatorName(\"+\"))))",
|
|
&Error));
|
|
EXPECT_EQ("", Error.toStringFull());
|
|
Matcher<Decl> M = VarDecl->unconditionalConvertTo<Decl>();
|
|
EXPECT_TRUE(matches("int x = 1 + false;", M));
|
|
EXPECT_FALSE(matches("int x = true + 1;", M));
|
|
EXPECT_FALSE(matches("int x = 1 - false;", M));
|
|
EXPECT_FALSE(matches("int x = true - 1;", M));
|
|
|
|
llvm::Optional<DynTypedMatcher> HasParameter(Parser::parseMatcherExpression(
|
|
"functionDecl(hasParameter(1, hasName(\"x\")))", &Error));
|
|
EXPECT_EQ("", Error.toStringFull());
|
|
M = HasParameter->unconditionalConvertTo<Decl>();
|
|
|
|
EXPECT_TRUE(matches("void f(int a, int x);", M));
|
|
EXPECT_FALSE(matches("void f(int x, int a);", M));
|
|
|
|
EXPECT_TRUE(!Parser::parseMatcherExpression(
|
|
"hasInitializer(\n binaryOperator(hasLHS(\"A\")))",
|
|
&Error).hasValue());
|
|
EXPECT_EQ("1:1: Error parsing argument 1 for matcher hasInitializer.\n"
|
|
"2:5: Error parsing argument 1 for matcher binaryOperator.\n"
|
|
"2:20: Error building matcher hasLHS.\n"
|
|
"2:27: Incorrect type for arg 1. "
|
|
"(Expected = Matcher<Expr>) != (Actual = String)",
|
|
Error.toStringFull());
|
|
}
|
|
|
|
std::string ParseWithError(StringRef Code) {
|
|
Diagnostics Error;
|
|
VariantValue Value;
|
|
Parser::parseExpression(Code, &Value, &Error);
|
|
return Error.toStringFull();
|
|
}
|
|
|
|
std::string ParseMatcherWithError(StringRef Code) {
|
|
Diagnostics Error;
|
|
Parser::parseMatcherExpression(Code, &Error);
|
|
return Error.toStringFull();
|
|
}
|
|
|
|
TEST(ParserTest, Errors) {
|
|
EXPECT_EQ(
|
|
"1:5: Error parsing matcher. Found token <123> while looking for '('.",
|
|
ParseWithError("Foo 123"));
|
|
EXPECT_EQ(
|
|
"1:9: Error parsing matcher. Found token <123> while looking for ','.",
|
|
ParseWithError("Foo(\"A\" 123)"));
|
|
EXPECT_EQ(
|
|
"1:4: Error parsing matcher. Found end-of-code while looking for ')'.",
|
|
ParseWithError("Foo("));
|
|
EXPECT_EQ("1:1: End of code found while looking for token.",
|
|
ParseWithError(""));
|
|
EXPECT_EQ("Input value is not a matcher expression.",
|
|
ParseMatcherWithError("\"A\""));
|
|
EXPECT_EQ("1:1: Error parsing argument 1 for matcher Foo.\n"
|
|
"1:5: Invalid token <(> found when looking for a value.",
|
|
ParseWithError("Foo(("));
|
|
EXPECT_EQ("1:7: Expected end of code.", ParseWithError("expr()a"));
|
|
EXPECT_EQ("1:11: Malformed bind() expression.",
|
|
ParseWithError("isArrow().biind"));
|
|
EXPECT_EQ("1:15: Malformed bind() expression.",
|
|
ParseWithError("isArrow().bind"));
|
|
EXPECT_EQ("1:16: Malformed bind() expression.",
|
|
ParseWithError("isArrow().bind(foo"));
|
|
EXPECT_EQ("1:21: Malformed bind() expression.",
|
|
ParseWithError("isArrow().bind(\"foo\""));
|
|
EXPECT_EQ("1:1: Error building matcher isArrow.\n"
|
|
"1:1: Matcher does not support binding.",
|
|
ParseWithError("isArrow().bind(\"foo\")"));
|
|
EXPECT_EQ("Input value has unresolved overloaded type: "
|
|
"Matcher<DoStmt|ForStmt|WhileStmt>",
|
|
ParseMatcherWithError("hasBody(stmt())"));
|
|
}
|
|
|
|
TEST(ParserTest, OverloadErrors) {
|
|
EXPECT_EQ("1:1: Error building matcher callee.\n"
|
|
"1:8: Candidate 1: Incorrect type for arg 1. "
|
|
"(Expected = Matcher<Stmt>) != (Actual = String)\n"
|
|
"1:8: Candidate 2: Incorrect type for arg 1. "
|
|
"(Expected = Matcher<Decl>) != (Actual = String)",
|
|
ParseWithError("callee(\"A\")"));
|
|
}
|
|
|
|
} // end anonymous namespace
|
|
} // end namespace dynamic
|
|
} // end namespace ast_matchers
|
|
} // end namespace clang
|