ClangRenameTest.h
4.34 KB
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
109
110
111
112
113
114
115
116
//===-- ClangRenameTests.cpp - clang-rename unit tests --------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_CLANG_UNITTESTS_RENAME_CLANGRENAMETEST_H
#define LLVM_CLANG_UNITTESTS_RENAME_CLANGRENAMETEST_H
#include "unittests/Tooling/RewriterTestContext.h"
#include "clang/ASTMatchers/ASTMatchFinder.h"
#include "clang/Basic/FileManager.h"
#include "clang/Basic/FileSystemOptions.h"
#include "clang/Format/Format.h"
#include "clang/Frontend/CompilerInstance.h"
#include "clang/Frontend/PCHContainerOperations.h"
#include "clang/Tooling/Refactoring.h"
#include "clang/Tooling/Refactoring/Rename/RenamingAction.h"
#include "clang/Tooling/Refactoring/Rename/USRFindingAction.h"
#include "clang/Tooling/Tooling.h"
#include "llvm/ADT/IntrusiveRefCntPtr.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/Format.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Support/VirtualFileSystem.h"
#include "gtest/gtest.h"
#include <memory>
#include <string>
#include <vector>
namespace clang {
namespace clang_rename {
namespace test {
struct Case {
std::string Before;
std::string After;
std::string OldName;
std::string NewName;
};
class ClangRenameTest : public testing::Test,
public testing::WithParamInterface<Case> {
protected:
void AppendToHeader(StringRef Code) { HeaderContent += Code.str(); }
std::string runClangRenameOnCode(llvm::StringRef Code,
llvm::StringRef OldName,
llvm::StringRef NewName) {
std::string NewCode;
llvm::raw_string_ostream(NewCode) << llvm::format(
"#include \"%s\"\n%s", HeaderName.c_str(), Code.str().c_str());
tooling::FileContentMappings FileContents = {{HeaderName, HeaderContent},
{CCName, NewCode}};
clang::RewriterTestContext Context;
Context.createInMemoryFile(HeaderName, HeaderContent);
clang::FileID InputFileID = Context.createInMemoryFile(CCName, NewCode);
tooling::USRFindingAction FindingAction({}, {std::string(OldName)}, false);
std::unique_ptr<tooling::FrontendActionFactory> USRFindingActionFactory =
tooling::newFrontendActionFactory(&FindingAction);
if (!tooling::runToolOnCodeWithArgs(
USRFindingActionFactory->create(), NewCode, {"-std=c++11"}, CCName,
"clang-rename", std::make_shared<PCHContainerOperations>(),
FileContents))
return "";
const std::vector<std::vector<std::string>> &USRList =
FindingAction.getUSRList();
std::vector<std::string> NewNames = {std::string(NewName)};
std::map<std::string, tooling::Replacements> FileToReplacements;
tooling::QualifiedRenamingAction RenameAction(NewNames, USRList,
FileToReplacements);
auto RenameActionFactory = tooling::newFrontendActionFactory(&RenameAction);
if (!tooling::runToolOnCodeWithArgs(
RenameActionFactory->create(), NewCode, {"-std=c++11"}, CCName,
"clang-rename", std::make_shared<PCHContainerOperations>(),
FileContents))
return "";
formatAndApplyAllReplacements(FileToReplacements, Context.Rewrite, "llvm");
return Context.getRewrittenText(InputFileID);
}
void CompareSnippets(StringRef Expected, StringRef Actual) {
std::string ExpectedCode;
llvm::raw_string_ostream(ExpectedCode) << llvm::format(
"#include \"%s\"\n%s", HeaderName.c_str(), Expected.str().c_str());
EXPECT_EQ(format(ExpectedCode), format(Actual));
}
std::string format(llvm::StringRef Code) {
tooling::Replacements Replaces = format::reformat(
format::getLLVMStyle(), Code, {tooling::Range(0, Code.size())});
auto ChangedCode = tooling::applyAllReplacements(Code, Replaces);
EXPECT_TRUE(static_cast<bool>(ChangedCode));
if (!ChangedCode) {
llvm::errs() << llvm::toString(ChangedCode.takeError());
return "";
}
return *ChangedCode;
}
std::string HeaderContent;
std::string HeaderName = "header.h";
std::string CCName = "input.cc";
};
} // namespace test
} // namespace clang_rename
} // namesdpace clang
#endif