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
|
//===--- TestWorkspace.cpp - Utility for writing multi-file tests -*- C++-*===//
//
// 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 "TestWorkspace.h"
#include "clang-include-cleaner/Record.h"
#include "index/FileIndex.h"
#include "gtest/gtest.h"
#include <memory>
#include <optional>
namespace clang {
namespace clangd {
std::unique_ptr<SymbolIndex> TestWorkspace::index() {
auto Index = std::make_unique<FileIndex>();
for (const auto &Input : Inputs) {
if (!Input.second.IsMainFile)
continue;
TU.Code = Input.second.Code;
TU.Filename = Input.first().str();
TU.preamble([&](CapturedASTCtx ASTCtx,
std::shared_ptr<const include_cleaner::PragmaIncludes> PI) {
auto &Ctx = ASTCtx.getASTContext();
auto &PP = ASTCtx.getPreprocessor();
Index->updatePreamble(testPath(Input.first()), "null", Ctx, PP, *PI);
});
ParsedAST MainAST = TU.build();
Index->updateMain(testPath(Input.first()), MainAST);
}
return Index;
}
std::optional<ParsedAST> TestWorkspace::openFile(llvm::StringRef Filename) {
auto It = Inputs.find(Filename);
if (It == Inputs.end()) {
ADD_FAILURE() << "Accessing non-existing file: " << Filename;
return std::nullopt;
}
TU.Code = It->second.Code;
TU.Filename = It->first().str();
return TU.build();
}
void TestWorkspace::addInput(llvm::StringRef Filename,
const SourceFile &Input) {
Inputs.insert(std::make_pair(Filename, Input));
TU.AdditionalFiles.insert(std::make_pair(Filename, Input.Code));
}
} // namespace clangd
} // namespace clang
|
