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 GetInputFilePath(const llvm::Twine & name)22std::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 fromYaml(llvm::StringRef Yaml)29llvm::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 if (!llvm::yaml::convertYAML(YIn, OS, [](const llvm::Twine &Msg) {})) 34 return llvm::createStringError(llvm::inconvertibleErrorCode(), 35 "convertYAML() failed"); 36 return TestFile(std::move(Buffer)); 37 } 38 fromYamlFile(const llvm::Twine & Name)39llvm::Expected<TestFile> TestFile::fromYamlFile(const llvm::Twine &Name) { 40 auto BufferOrError = 41 llvm::MemoryBuffer::getFile(GetInputFilePath(Name), /*FileSize*/ -1, 42 /*RequiresNullTerminator*/ false); 43 if (!BufferOrError) 44 return llvm::errorCodeToError(BufferOrError.getError()); 45 return fromYaml(BufferOrError.get()->getBuffer()); 46 } 47