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
|
//===--- IncludeSpellerTest.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 "clang-include-cleaner/IncludeSpeller.h"
#include "clang-include-cleaner/Analysis.h"
#include "clang-include-cleaner/Types.h"
#include "clang/Lex/Preprocessor.h"
#include "clang/Testing/TestAST.h"
#include "clang/Tooling/Inclusions/StandardLibrary.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/Path.h"
#include "gtest/gtest.h"
#include <assert.h>
#include <string>
namespace clang::include_cleaner {
namespace {
const char *testRoot() {
#ifdef _WIN32
return "C:\\include-cleaner-test";
#else
return "/include-cleaner-test";
#endif
}
std::string testPath(llvm::StringRef File) {
assert(llvm::sys::path::is_relative(File) && "FileName should be relative");
llvm::SmallString<32> NativeFile = File;
llvm::sys::path::native(NativeFile, llvm::sys::path::Style::native);
llvm::SmallString<32> Path;
llvm::sys::path::append(Path, llvm::sys::path::Style::native, testRoot(),
NativeFile);
return std::string(Path.str());
}
class DummyIncludeSpeller : public IncludeSpeller {
public:
std::string operator()(const IncludeSpeller::Input &Input) const override {
if (Input.H.kind() == Header::Standard)
return "<bits/stdc++.h>";
if (Input.H.kind() != Header::Physical)
return "";
llvm::StringRef AbsolutePath = Input.H.physical()->tryGetRealPathName();
std::string RootWithSeparator{testRoot()};
RootWithSeparator += llvm::sys::path::get_separator();
if (!AbsolutePath.consume_front(llvm::StringRef{RootWithSeparator}))
return "";
return "\"" + AbsolutePath.str() + "\"";
}
};
TEST(IncludeSpeller, IsRelativeToTestRoot) {
TestInputs Inputs;
Inputs.ExtraArgs.push_back("-isystemdir");
Inputs.ExtraFiles[testPath("foo.h")] = "";
Inputs.ExtraFiles["dir/header.h"] = "";
TestAST AST{Inputs};
auto &FM = AST.fileManager();
auto &HS = AST.preprocessor().getHeaderSearchInfo();
const auto *MainFile = AST.sourceManager().getFileEntryForID(
AST.sourceManager().getMainFileID());
EXPECT_EQ("\"foo.h\"", spellHeader({Header{*FM.getFile(testPath("foo.h"))},
HS, MainFile}));
EXPECT_EQ("<header.h>",
spellHeader({Header{*FM.getFile("dir/header.h")}, HS, MainFile}));
}
TEST(IncludeSpeller, CanOverrideSystemHeaders) {
TestAST AST("");
auto &HS = AST.preprocessor().getHeaderSearchInfo();
const auto *MainFile = AST.sourceManager().getFileEntryForID(
AST.sourceManager().getMainFileID());
EXPECT_EQ("<bits/stdc++.h>",
spellHeader({Header{*tooling::stdlib::Header::named("<vector>")},
HS, MainFile}));
}
IncludeSpellingStrategy::Add<DummyIncludeSpeller>
Speller("dummy", "Dummy Include Speller");
} // namespace
} // namespace clang::include_cleaner
|
