46a6f5ae14
This will allow moving the IncludeCleaner library essentials to Clang and decoupling them from the majority of clangd. The patch itself just moves the code, it doesn't change existing functionality. Reviewed By: sammccall Differential Revision: https://reviews.llvm.org/D119130
112 lines
3.3 KiB
C++
112 lines
3.3 KiB
C++
//===- unittest/Tooling/StandardLibrary.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/Tooling/Inclusions/StandardLibrary.h"
|
|
#include "clang/AST/Decl.h"
|
|
#include "clang/AST/DeclarationName.h"
|
|
#include "clang/Frontend/ASTUnit.h"
|
|
#include "clang/Tooling/Tooling.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Support/Casting.h"
|
|
#include "llvm/Support/ScopedPrinter.h"
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using ::testing::ElementsAre;
|
|
|
|
namespace clang {
|
|
namespace tooling {
|
|
namespace {
|
|
|
|
const NamedDecl &lookup(ASTUnit &AST, llvm::StringRef Name) {
|
|
auto &Ctx = AST.getASTContext();
|
|
TranslationUnitDecl *TU = Ctx.getTranslationUnitDecl();
|
|
auto Result = TU->lookup(DeclarationName(&Ctx.Idents.get(Name)));
|
|
assert(!Result.empty() && "Lookup failed");
|
|
assert(Result.isSingleResult() && "Lookup returned multiple results");
|
|
return *Result.front();
|
|
}
|
|
|
|
TEST(StdlibTest, All) {
|
|
auto VectorH = stdlib::Header::named("<vector>");
|
|
EXPECT_TRUE(VectorH);
|
|
EXPECT_EQ(llvm::to_string(*VectorH), "<vector>");
|
|
EXPECT_FALSE(stdlib::Header::named("HeadersTests.cpp"));
|
|
|
|
auto Vector = stdlib::Symbol::named("std::", "vector");
|
|
EXPECT_TRUE(Vector);
|
|
EXPECT_EQ(llvm::to_string(*Vector), "std::vector");
|
|
EXPECT_FALSE(stdlib::Symbol::named("std::", "dongle"));
|
|
EXPECT_FALSE(stdlib::Symbol::named("clang::", "ASTContext"));
|
|
|
|
EXPECT_EQ(Vector->header(), *VectorH);
|
|
EXPECT_THAT(Vector->headers(), ElementsAre(*VectorH));
|
|
}
|
|
|
|
TEST(StdlibTest, Recognizer) {
|
|
std::unique_ptr<ASTUnit> AST = buildASTFromCode(R"cpp(
|
|
namespace std {
|
|
inline namespace inl {
|
|
|
|
template <typename>
|
|
struct vector { class nested {}; };
|
|
|
|
class secret {};
|
|
|
|
} // inl
|
|
|
|
inline namespace __1 {
|
|
namespace chrono {
|
|
inline namespace chrono_inl {
|
|
class system_clock {};
|
|
} // chrono_inl
|
|
} // chrono
|
|
} // __1
|
|
|
|
} // std
|
|
|
|
// C Standard Library structure defined in <stdlib.h>
|
|
struct div_t {};
|
|
|
|
class vector {};
|
|
std::vector<int> vec;
|
|
std::vector<int>::nested nest;
|
|
std::secret sec;
|
|
std::chrono::system_clock clock;
|
|
|
|
div_t div;
|
|
)cpp");
|
|
|
|
auto &VectorNonstd = lookup(*AST, "vector");
|
|
auto *Vec =
|
|
cast<VarDecl>(lookup(*AST, "vec")).getType()->getAsCXXRecordDecl();
|
|
auto *Nest =
|
|
cast<VarDecl>(lookup(*AST, "nest")).getType()->getAsCXXRecordDecl();
|
|
auto *Clock =
|
|
cast<VarDecl>(lookup(*AST, "clock")).getType()->getAsCXXRecordDecl();
|
|
auto *Sec =
|
|
cast<VarDecl>(lookup(*AST, "sec")).getType()->getAsCXXRecordDecl();
|
|
auto *CDivT =
|
|
cast<VarDecl>(lookup(*AST, "div")).getType()->getAsCXXRecordDecl();
|
|
|
|
stdlib::Recognizer Recognizer;
|
|
|
|
EXPECT_EQ(Recognizer(&VectorNonstd), llvm::None);
|
|
EXPECT_EQ(Recognizer(Vec), stdlib::Symbol::named("std::", "vector"));
|
|
EXPECT_EQ(Recognizer(Nest), stdlib::Symbol::named("std::", "vector"));
|
|
EXPECT_EQ(Recognizer(Clock),
|
|
stdlib::Symbol::named("std::chrono::", "system_clock"));
|
|
EXPECT_EQ(Recognizer(CDivT), stdlib::Symbol::named("", "div_t"));
|
|
EXPECT_EQ(Recognizer(Sec), llvm::None);
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace tooling
|
|
} // namespace clang
|