125 lines
3.8 KiB
C++
125 lines
3.8 KiB
C++
//===- unittest/Tooling/RecursiveASTVisitorTests/DeclRefExpr.cpp ----------===//
|
|
//
|
|
// 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 "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 "i" to be visited twice: once for the initialization expr
|
|
// for the captured variable "i" outside of the lambda body, and again for
|
|
// the use of "i" inside the lambda.
|
|
Visitor.ExpectMatch("i", 1, 20, /*Times=*/1);
|
|
Visitor.ExpectMatch("i", 1, 24, /*Times=*/1);
|
|
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
|