|
@@ -13,6 +13,7 @@
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "gtest/gtest.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
+#include "TestUtils.h"
|
|
#include <fstream>
|
|
#include <fstream>
|
|
#include <functional>
|
|
#include <functional>
|
|
#include <map>
|
|
#include <map>
|
|
@@ -352,77 +353,6 @@ TEST(libclang, ModuleMapDescriptor) {
|
|
clang_ModuleMapDescriptor_dispose(MMD);
|
|
clang_ModuleMapDescriptor_dispose(MMD);
|
|
}
|
|
}
|
|
|
|
|
|
-class LibclangParseTest : public ::testing::Test {
|
|
|
|
- std::set<std::string> Files;
|
|
|
|
- typedef std::unique_ptr<std::string> fixed_addr_string;
|
|
|
|
- std::map<fixed_addr_string, fixed_addr_string> UnsavedFileContents;
|
|
|
|
-public:
|
|
|
|
- std::string TestDir;
|
|
|
|
- CXIndex Index;
|
|
|
|
- CXTranslationUnit ClangTU;
|
|
|
|
- unsigned TUFlags;
|
|
|
|
- std::vector<CXUnsavedFile> UnsavedFiles;
|
|
|
|
-
|
|
|
|
- void SetUp() override {
|
|
|
|
- llvm::SmallString<256> Dir;
|
|
|
|
- ASSERT_FALSE(llvm::sys::fs::createUniqueDirectory("libclang-test", Dir));
|
|
|
|
- TestDir = Dir.str();
|
|
|
|
- TUFlags = CXTranslationUnit_DetailedPreprocessingRecord |
|
|
|
|
- clang_defaultEditingTranslationUnitOptions();
|
|
|
|
- Index = clang_createIndex(0, 0);
|
|
|
|
- ClangTU = nullptr;
|
|
|
|
- }
|
|
|
|
- void TearDown() override {
|
|
|
|
- clang_disposeTranslationUnit(ClangTU);
|
|
|
|
- clang_disposeIndex(Index);
|
|
|
|
- for (const std::string &Path : Files)
|
|
|
|
- llvm::sys::fs::remove(Path);
|
|
|
|
- llvm::sys::fs::remove(TestDir);
|
|
|
|
- }
|
|
|
|
- void WriteFile(std::string &Filename, const std::string &Contents) {
|
|
|
|
- if (!llvm::sys::path::is_absolute(Filename)) {
|
|
|
|
- llvm::SmallString<256> Path(TestDir);
|
|
|
|
- llvm::sys::path::append(Path, Filename);
|
|
|
|
- Filename = Path.str();
|
|
|
|
- Files.insert(Filename);
|
|
|
|
- }
|
|
|
|
- llvm::sys::fs::create_directories(llvm::sys::path::parent_path(Filename));
|
|
|
|
- std::ofstream OS(Filename);
|
|
|
|
- OS << Contents;
|
|
|
|
- assert(OS.good());
|
|
|
|
- }
|
|
|
|
- void MapUnsavedFile(std::string Filename, const std::string &Contents) {
|
|
|
|
- if (!llvm::sys::path::is_absolute(Filename)) {
|
|
|
|
- llvm::SmallString<256> Path(TestDir);
|
|
|
|
- llvm::sys::path::append(Path, Filename);
|
|
|
|
- Filename = Path.str();
|
|
|
|
- }
|
|
|
|
- auto it = UnsavedFileContents.insert(std::make_pair(
|
|
|
|
- fixed_addr_string(new std::string(Filename)),
|
|
|
|
- fixed_addr_string(new std::string(Contents))));
|
|
|
|
- UnsavedFiles.push_back({
|
|
|
|
- it.first->first->c_str(), // filename
|
|
|
|
- it.first->second->c_str(), // contents
|
|
|
|
- it.first->second->size() // length
|
|
|
|
- });
|
|
|
|
- }
|
|
|
|
- template<typename F>
|
|
|
|
- void Traverse(const F &TraversalFunctor) {
|
|
|
|
- CXCursor TuCursor = clang_getTranslationUnitCursor(ClangTU);
|
|
|
|
- std::reference_wrapper<const F> FunctorRef = std::cref(TraversalFunctor);
|
|
|
|
- clang_visitChildren(TuCursor,
|
|
|
|
- &TraverseStateless<std::reference_wrapper<const F>>,
|
|
|
|
- &FunctorRef);
|
|
|
|
- }
|
|
|
|
-private:
|
|
|
|
- template<typename TState>
|
|
|
|
- static CXChildVisitResult TraverseStateless(CXCursor cx, CXCursor parent,
|
|
|
|
- CXClientData data) {
|
|
|
|
- TState *State = static_cast<TState*>(data);
|
|
|
|
- return State->get()(cx, parent);
|
|
|
|
- }
|
|
|
|
-};
|
|
|
|
-
|
|
|
|
TEST_F(LibclangParseTest, AllSkippedRanges) {
|
|
TEST_F(LibclangParseTest, AllSkippedRanges) {
|
|
std::string Header = "header.h", Main = "main.cpp";
|
|
std::string Header = "header.h", Main = "main.cpp";
|
|
WriteFile(Header,
|
|
WriteFile(Header,
|