clang-tools-extra/clangd/unittests/TestWorkspace.cpp

   1 //===--- TestWorkspace.cpp - Utility for writing multi-file tests -*- C++-*===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "TestWorkspace.h"
  10 #include "index/FileIndex.h"
  11 #include "gtest/gtest.h"
  12 #include <optional>
  13
  14 namespace clang {
  15 namespace clangd {
  16
  17 std::unique_ptr<SymbolIndex> TestWorkspace::index() {
  18   auto Index = std::make_unique<FileIndex>();
  19   for (const auto &Input : Inputs) {
  20     if (!Input.second.IsMainFile)
  21       continue;
  22     TU.Code = Input.second.Code;
  23     TU.Filename = Input.first().str();
  24     TU.preamble(
  25         [&](CapturedASTCtx ASTCtx,
  26             const std::shared_ptr<const CanonicalIncludes> CanonIncludes) {
  27           auto &Ctx = ASTCtx.getASTContext();
  28           auto &PP = ASTCtx.getPreprocessor();
  29           Index->updatePreamble(testPath(Input.first()), "null", Ctx, PP,
  30                                 *CanonIncludes);
  31         });
  32     ParsedAST MainAST = TU.build();
  33     Index->updateMain(testPath(Input.first()), MainAST);
  34   }
  35   return Index;
  36 }
  37
  38 std::optional<ParsedAST> TestWorkspace::openFile(llvm::StringRef Filename) {
  39   auto It = Inputs.find(Filename);
  40   if (It == Inputs.end()) {
  41     ADD_FAILURE() << "Accessing non-existing file: " << Filename;
  42     return std::nullopt;
  43   }
  44   TU.Code = It->second.Code;
  45   TU.Filename = It->first().str();
  46   return TU.build();
  47 }
  48
  49 void TestWorkspace::addInput(llvm::StringRef Filename,
  50                              const SourceFile &Input) {
  51   Inputs.insert(std::make_pair(Filename, Input));
  52   TU.AdditionalFiles.insert(std::make_pair(Filename, Input.Code));
  53 }
  54 } // namespace clangd
  55 } // namespace clang