Gitiles
Code Review Sign In
gerrit-public.fairphone.software / toolchain / llvm-project / 2146a0a90ea227530db7ec89d537f48a8967d038 / . / clang-tools-extra / unittests / change-namespace / ChangeNamespaceTests.cpp
blob: fb46bdbe7c41995b06069cdd01ef87254b4693cc [file] [log] [blame]
Eric Liu495b2112016-09-19 17:40:32 +0000[diff] [blame]1//===-- ChangeNamespaceTests.cpp - Change namespace unit tests ---*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include "ChangeNamespace.h"
11#include "unittests/Tooling/RewriterTestContext.h"
12#include "clang/ASTMatchers/ASTMatchFinder.h"
13#include "clang/Basic/FileManager.h"
14#include "clang/Basic/FileSystemOptions.h"
15#include "clang/Basic/VirtualFileSystem.h"
16#include "clang/Format/Format.h"
17#include "clang/Frontend/CompilerInstance.h"
18#include "clang/Frontend/PCHContainerOperations.h"
19#include "clang/Tooling/Refactoring.h"
20#include "clang/Tooling/Tooling.h"
21#include "llvm/ADT/IntrusiveRefCntPtr.h"
22#include "llvm/ADT/StringRef.h"
23#include "llvm/Support/MemoryBuffer.h"
24#include "gtest/gtest.h"
25#include <memory>
26#include <string>
27#include <vector>
28
29namespace clang {
30namespace change_namespace {
31namespace {
32
33class ChangeNamespaceTest : public ::testing::Test {
34public:
35 std::string runChangeNamespaceOnCode(llvm::StringRef Code) {
36 clang::RewriterTestContext Context;
37 clang::FileID ID = Context.createInMemoryFile(FileName, Code);
38
39 std::map<std::string, tooling::Replacements> FileToReplacements;
40 change_namespace::ChangeNamespaceTool NamespaceTool(
41 OldNamespace, NewNamespace, FilePattern, &FileToReplacements);
42 ast_matchers::MatchFinder Finder;
43 NamespaceTool.registerMatchers(&Finder);
44 std::unique_ptr<tooling::FrontendActionFactory> Factory =
45 tooling::newFrontendActionFactory(&Finder);
46 tooling::runToolOnCodeWithArgs(Factory->create(), Code, {"-std=c++11"},
47 FileName);
48 formatAndApplyAllReplacements(FileToReplacements, Context.Rewrite);
49 return format(Context.getRewrittenText(ID));
50 }
51
52 std::string format(llvm::StringRef Code) {
53 tooling::Replacements Replaces = format::reformat(
54 format::getLLVMStyle(), Code, {tooling::Range(0, Code.size())});
55 auto ChangedCode = tooling::applyAllReplacements(Code, Replaces);
56 EXPECT_TRUE(static_cast<bool>(ChangedCode));
57 if (!ChangedCode) {
58 llvm::errs() << llvm::toString(ChangedCode.takeError());
59 return "";
60 }
61 return *ChangedCode;
62 }
63
64protected:
65 std::string FileName = "input.cc";
66 std::string OldNamespace = "na::nb";
67 std::string NewNamespace = "x::y";
68 std::string FilePattern = "input.cc";
69};
70
71TEST_F(ChangeNamespaceTest, NoMatchingNamespace) {
72 std::string Code = "namespace na {\n"
73 "namespace nx {\n"
74 "class A {};\n"
75 "} // namespace nx\n"
76 "} // namespace na\n";
77 std::string Expected = "namespace na {\n"
78 "namespace nx {\n"
79 "class A {};\n"
80 "} // namespace nx\n"
81 "} // namespace na\n";
82 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
83}
84
85TEST_F(ChangeNamespaceTest, SimpleMoveWithoutTypeRefs) {
86 std::string Code = "namespace na {\n"
87 "namespace nb {\n"
88 "class A {};\n"
89 "} // namespace nb\n"
90 "} // namespace na\n";
91 std::string Expected = "\n\n"
92 "namespace x {\n"
93 "namespace y {\n"
94 "class A {};\n"
95 "} // namespace y\n"
96 "} // namespace x\n";
97 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
98}
99
100TEST_F(ChangeNamespaceTest, SimpleMoveIntoAnotherNestedNamespace) {
101 NewNamespace = "na::nc";
102 std::string Code = "namespace na {\n"
103 "namespace nb {\n"
104 "class A {};\n"
105 "} // namespace nb\n"
106 "} // namespace na\n";
107 std::string Expected = "namespace na {\n"
108 "\n"
109 "namespace nc {\n"
110 "class A {};\n"
111 "} // namespace nc\n"
112 "} // namespace na\n";
113 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
114}
115
116TEST_F(ChangeNamespaceTest, SimpleMoveNestedNamespace) {
117 NewNamespace = "na::x::y";
118 std::string Code = "namespace na {\n"
119 "class A {};\n"
120 "namespace nb {\n"
121 "class B {};\n"
122 "} // namespace nb\n"
123 "} // namespace na\n";
124 std::string Expected = "namespace na {\n"
125 "class A {};\n"
126 "\n"
127 "namespace x {\n"
128 "namespace y {\n"
129 "class B {};\n"
130 "} // namespace y\n"
131 "} // namespace x\n"
132 "} // namespace na\n";
133 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
134}
135
136TEST_F(ChangeNamespaceTest, SimpleMoveWithTypeRefs) {
137 std::string Code = "namespace na {\n"
138 "class C_A {};\n"
139 "namespace nc {\n"
140 "class C_C {};"
141 "} // namespace nc\n"
142 "namespace nb {\n"
143 "class C_X {\n"
144 "public:\n"
145 " C_A a;\n"
146 " nc::C_C c;\n"
147 "};\n"
148 "class C_Y {\n"
149 " C_X x;\n"
150 "};\n"
151 "} // namespace nb\n"
152 "} // namespace na\n";
153 std::string Expected = "namespace na {\n"
154 "class C_A {};\n"
155 "namespace nc {\n"
156 "class C_C {};"
157 "} // namespace nc\n"
158 "\n"
159 "} // namespace na\n"
160 "namespace x {\n"
161 "namespace y {\n"
162 "class C_X {\n"
163 "public:\n"
164 " na::C_A a;\n"
165 " na::nc::C_C c;\n"
166 "};\n"
167 "class C_Y {\n"
168 " C_X x;\n"
169 "};\n"
170 "} // namespace y\n"
171 "} // namespace x\n";
172 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
173}
174
175TEST_F(ChangeNamespaceTest, LeaveForwardDeclarationBehind) {
176 std::string Code = "namespace na {\n"
177 "namespace nb {\n"
178 "class FWD;\n"
179 "class A {\n"
180 " FWD *fwd;\n"
181 "};\n"
182 "} // namespace nb\n"
183 "} // namespace na\n";
184 std::string Expected = "namespace na {\n"
185 "namespace nb {\n"
186 "class FWD;\n"
187 "} // namespace nb\n"
188 "} // namespace na\n"
189 "namespace x {\n"
190 "namespace y {\n"
191 "\n"
192 "class A {\n"
193 " na::nb::FWD *fwd;\n"
194 "};\n"
195 "} // namespace y\n"
196 "} // namespace x\n";
197 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
198}
199
200TEST_F(ChangeNamespaceTest, MoveFunctions) {
201 std::string Code = "namespace na {\n"
202 "class C_A {};\n"
203 "namespace nc {\n"
204 "class C_C {};"
205 "} // namespace nc\n"
206 "namespace nb {\n"
207 "void fwd();\n"
208 "void f(C_A ca, nc::C_C cc) {\n"
209 " C_A ca_1 = ca;\n"
210 "}\n"
211 "} // namespace nb\n"
212 "} // namespace na\n";
213
214 std::string Expected = "namespace na {\n"
215 "class C_A {};\n"
216 "namespace nc {\n"
217 "class C_C {};"
218 "} // namespace nc\n"
219 "\n"
220 "} // namespace na\n"
221 "namespace x {\n"
222 "namespace y {\n"
223 "void fwd();\n"
224 "void f(na::C_A ca, na::nc::C_C cc) {\n"
225 " na::C_A ca_1 = ca;\n"
226 "}\n"
227 "} // namespace y\n"
228 "} // namespace x\n";
229 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
230}
231
Eric Liu68765a82016-09-21 15:06:12 +0000[diff] [blame]232TEST_F(ChangeNamespaceTest, FixUsingShadowDecl) {
233 std::string Code = "namespace na {\n"
234 "namespace nc {\n"
235 "class SAME {};\n"
236 "}\n"
237 "namespace nd {\n"
238 "class SAME {};\n"
239 "}\n"
240 "namespace nb {\n"
241 "using nc::SAME;\n"
242 "using YO = nc::SAME;\n"
243 "typedef nc::SAME IDENTICAL;\n"
244 "void f(nd::SAME Same) {}\n"
245 "} // namespace nb\n"
246 "} // namespace na\n";
247
248 std::string Expected = "namespace na {\n"
249 "namespace nc {\n"
250 "class SAME {};\n"
251 "}\n"
252 "namespace nd {\n"
253 "class SAME {};\n"
254 "}\n"
255 "\n"
256 "} // namespace na\n"
257 "namespace x {\n"
258 "namespace y {\n"
259 "using ::na::nc::SAME;\n"
260 "using YO = na::nc::SAME;\n"
261 "typedef na::nc::SAME IDENTICAL;\n"
262 "void f(na::nd::SAME Same) {}\n"
263 "} // namespace y\n"
264 "} // namespace x\n";
265 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
266}
267
268TEST_F(ChangeNamespaceTest, TypeInNestedNameSpecifier) {
269 std::string Code =
270 "namespace na {\n"
271 "class C_A {\n"
272 "public:\n"
273 " class Nested {\n"
274 " public:\n"
275 " static int NestedX;\n"
276 " static void nestedFunc() {}\n"
277 " };\n"
278 "};\n"
279 "namespace nb {\n"
280 "class C_X {\n"
281 " C_A na;\n"
282 " C_A::Nested nested;\n"
283 " void f() {\n"
284 " C_A::Nested::nestedFunc();\n"
285 " int X = C_A::Nested::NestedX;\n"
286 " }\n"
287 "};\n"
288 "} // namespace nb\n"
289 "} // namespace na\n";
290 std::string Expected =
291 "namespace na {\n"
292 "class C_A {\n"
293 "public:\n"
294 " class Nested {\n"
295 " public:\n"
296 " static int NestedX;\n"
297 " static void nestedFunc() {}\n"
298 " };\n"
299 "};\n"
300 "\n"
301 "} // namespace na\n"
302 "namespace x {\n"
303 "namespace y {\n"
304 "class C_X {\n"
305 " na::C_A na;\n"
306 " na::C_A::Nested nested;\n"
307 " void f() {\n"
308 " na::C_A::Nested::nestedFunc();\n"
309 " int X = na::C_A::Nested::NestedX;\n"
310 " }\n"
311 "};\n"
312 "} // namespace y\n"
313 "} // namespace x\n";
314 EXPECT_EQ(format(Expected), runChangeNamespaceOnCode(Code));
315}
316
Eric Liu495b2112016-09-19 17:40:32 +0000[diff] [blame]317} // anonymous namespace
318} // namespace change_namespace
319} // namespace clang