| //===- unittest/Tooling/RecursiveASTVisitorTests/DeclRefExpr.cpp ----------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "TestVisitor.h" |
| |
| using namespace clang; |
| |
| namespace { |
| |
| class DeclRefExprVisitor : public ExpectedLocationVisitor<DeclRefExprVisitor> { |
| public: |
| DeclRefExprVisitor() : ShouldVisitImplicitCode(false) {} |
| |
| bool shouldVisitImplicitCode() const { return ShouldVisitImplicitCode; } |
| |
| void setShouldVisitImplicitCode(bool NewValue) { |
| ShouldVisitImplicitCode = NewValue; |
| } |
| |
| bool VisitDeclRefExpr(DeclRefExpr *Reference) { |
| Match(Reference->getNameInfo().getAsString(), Reference->getLocation()); |
| return true; |
| } |
| |
| private: |
| bool ShouldVisitImplicitCode; |
| }; |
| |
| TEST(RecursiveASTVisitor, VisitsBaseClassTemplateArguments) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("x", 2, 3); |
| EXPECT_TRUE(Visitor.runOver( |
| "void x(); template <void (*T)()> class X {};\nX<x> y;")); |
| } |
| |
| TEST(RecursiveASTVisitor, VisitsCXXForRangeStmtRange) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("x", 2, 25); |
| Visitor.ExpectMatch("x", 2, 30); |
| EXPECT_TRUE(Visitor.runOver( |
| "int x[5];\n" |
| "void f() { for (int i : x) { x[0] = 1; } }", |
| DeclRefExprVisitor::Lang_CXX11)); |
| } |
| |
| TEST(RecursiveASTVisitor, VisitsCallExpr) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("x", 1, 22); |
| EXPECT_TRUE(Visitor.runOver( |
| "void x(); void y() { x(); }")); |
| } |
| |
| TEST(RecursiveASTVisitor, VisitsExplicitLambdaCaptureInit) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("i", 1, 20); |
| EXPECT_TRUE(Visitor.runOver( |
| "void f() { int i; [i]{}; }", |
| DeclRefExprVisitor::Lang_CXX11)); |
| } |
| |
| TEST(RecursiveASTVisitor, VisitsUseOfImplicitLambdaCapture) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("i", 1, 24); |
| EXPECT_TRUE(Visitor.runOver( |
| "void f() { int i; [=]{ i; }; }", |
| DeclRefExprVisitor::Lang_CXX11)); |
| } |
| |
| TEST(RecursiveASTVisitor, VisitsImplicitLambdaCaptureInit) { |
| DeclRefExprVisitor Visitor; |
| Visitor.setShouldVisitImplicitCode(true); |
| // We're expecting the "i" in the lambda to be visited twice: |
| // - Once for the DeclRefExpr in the lambda capture initialization (whose |
| // source code location is set to the first use of the variable). |
| // - Once for the DeclRefExpr for the use of "i" inside the lambda. |
| Visitor.ExpectMatch("i", 1, 24, /*Times=*/2); |
| EXPECT_TRUE(Visitor.runOver( |
| "void f() { int i; [=]{ i; }; }", |
| DeclRefExprVisitor::Lang_CXX11)); |
| } |
| |
| TEST(RecursiveASTVisitor, VisitsLambdaInitCaptureInit) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("i", 1, 24); |
| EXPECT_TRUE(Visitor.runOver( |
| "void f() { int i; [a = i + 1]{}; }", |
| DeclRefExprVisitor::Lang_CXX14)); |
| } |
| |
| /* FIXME: According to Richard Smith this is a bug in the AST. |
| TEST(RecursiveASTVisitor, VisitsBaseClassTemplateArgumentsInInstantiation) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("x", 3, 43); |
| EXPECT_TRUE(Visitor.runOver( |
| "template <typename T> void x();\n" |
| "template <void (*T)()> class X {};\n" |
| "template <typename T> class Y : public X< x<T> > {};\n" |
| "Y<int> y;")); |
| } |
| */ |
| |
| TEST(RecursiveASTVisitor, VisitsExtension) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("s", 1, 24); |
| EXPECT_TRUE(Visitor.runOver( |
| "int s = __extension__ (s);\n")); |
| } |
| |
| TEST(RecursiveASTVisitor, VisitsCopyExprOfBlockDeclCapture) { |
| DeclRefExprVisitor Visitor; |
| Visitor.ExpectMatch("x", 3, 24); |
| EXPECT_TRUE(Visitor.runOver("void f(int(^)(int)); \n" |
| "void g() { \n" |
| " f([&](int x){ return x; }); \n" |
| "}", |
| DeclRefExprVisitor::Lang_OBJCXX11)); |
| } |
| |
| } // end anonymous namespace |