lldb/unittests/TestingSupport/TestUtilities.cpp

   1 //===-- TestUtilities.cpp -------------------------------------------------===//
   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 "TestUtilities.h"
  10 #include "llvm/ADT/SmallString.h"
  11 #include "llvm/ObjectYAML/yaml2obj.h"
  12 #include "llvm/Support/FileSystem.h"
  13 #include "llvm/Support/Path.h"
  14 #include "llvm/Support/Program.h"
  15 #include "llvm/Support/YAMLTraits.h"
  16 #include "gtest/gtest.h"
  17
  18 using namespace lldb_private;
  19
  20 extern const char *TestMainArgv0;
  21
  22 std::string lldb_private::GetInputFilePath(const llvm::Twine &name) {
  23   llvm::SmallString<128> result = llvm::sys::path::parent_path(TestMainArgv0);
  24   llvm::sys::fs::make_absolute(result);
  25   llvm::sys::path::append(result, "Inputs", name);
  26   return std::string(result.str());
  27 }
  28
  29 llvm::Expected<TestFile> TestFile::fromYaml(llvm::StringRef Yaml) {
  30   std::string Buffer;
  31   llvm::raw_string_ostream OS(Buffer);
  32   llvm::yaml::Input YIn(Yaml);
  33   std::string ErrorMsg("convertYAML() failed: ");
  34   if (!llvm::yaml::convertYAML(YIn, OS, [&ErrorMsg](const llvm::Twine &Msg) {
  35         ErrorMsg += Msg.str();
  36       }))
  37     return llvm::createStringError(llvm::inconvertibleErrorCode(), ErrorMsg);
  38   return TestFile(std::move(Buffer));
  39 }
  40
  41 llvm::Expected<TestFile> TestFile::fromYamlFile(const llvm::Twine &Name) {
  42   auto BufferOrError =
  43       llvm::MemoryBuffer::getFile(GetInputFilePath(Name), /*IsText=*/false,
  44                                   /*RequiresNullTerminator=*/false);
  45   if (!BufferOrError)
  46     return llvm::errorCodeToError(BufferOrError.getError());
  47   return fromYaml(BufferOrError.get()->getBuffer());
  48 }