1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
//===- unittest/Tooling/LookupTest.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"
#include "clang/Tooling/Core/Lookup.h"
using namespace clang;
namespace {
struct GetDeclsVisitor : TestVisitor<GetDeclsVisitor> {
std::function<void(CallExpr *)> OnCall;
SmallVector<Decl *, 4> DeclStack;
bool VisitCallExpr(CallExpr *Expr) {
OnCall(Expr);
return true;
}
bool TraverseDecl(Decl *D) {
DeclStack.push_back(D);
bool Ret = TestVisitor::TraverseDecl(D);
DeclStack.pop_back();
return Ret;
}
};
TEST(LookupTest, replaceNestedName) {
GetDeclsVisitor Visitor;
auto replaceCallExpr = [&](const CallExpr *Expr,
StringRef ReplacementString) {
const auto *Callee = cast<DeclRefExpr>(Expr->getCallee()->IgnoreImplicit());
const ValueDecl *FD = Callee->getDecl();
return tooling::replaceNestedName(
Callee->getQualifier(), Visitor.DeclStack.back()->getDeclContext(), FD,
ReplacementString);
};
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("bar", replaceCallExpr(Expr, "::bar"));
};
Visitor.runOver("namespace a { void foo(); }\n"
"namespace a { void f() { foo(); } }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("bar", replaceCallExpr(Expr, "::a::bar"));
};
Visitor.runOver("namespace a { void foo(); }\n"
"namespace a { void f() { foo(); } }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("a::bar", replaceCallExpr(Expr, "::a::bar"));
};
Visitor.runOver("namespace a { void foo(); }\n"
"namespace b { void f() { a::foo(); } }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("a::bar", replaceCallExpr(Expr, "::a::bar"));
};
Visitor.runOver("namespace a { void foo(); }\n"
"namespace b { namespace a { void foo(); }\n"
"void f() { a::foo(); } }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("c::bar", replaceCallExpr(Expr, "::a::c::bar"));
};
Visitor.runOver("namespace a { namespace b { void foo(); }\n"
"void f() { b::foo(); } }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("bar", replaceCallExpr(Expr, "::a::bar"));
};
Visitor.runOver("namespace a { namespace b { void foo(); }\n"
"void f() { b::foo(); } }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("bar", replaceCallExpr(Expr, "::bar"));
};
Visitor.runOver("void foo(); void f() { foo(); }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("::bar", replaceCallExpr(Expr, "::bar"));
};
Visitor.runOver("void foo(); void f() { ::foo(); }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("a::bar", replaceCallExpr(Expr, "::a::bar"));
};
Visitor.runOver("namespace a { void foo(); }\nvoid f() { a::foo(); }\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("a::bar", replaceCallExpr(Expr, "::a::bar"));
};
Visitor.runOver("namespace a { int foo(); }\nauto f = a::foo();\n");
Visitor.OnCall = [&](CallExpr *Expr) {
EXPECT_EQ("bar", replaceCallExpr(Expr, "::a::bar"));
};
Visitor.runOver(
"namespace a { int foo(); }\nusing a::foo;\nauto f = foo();\n");
}
} // end anonymous namespace
|
