forked from OSchip/llvm-project
66 lines
2.3 KiB
C++
66 lines
2.3 KiB
C++
//===-- TestUtilities.cpp -------------------------------------------------===//
|
|
//
|
|
// 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 "TestUtilities.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/ObjectYAML/yaml2obj.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/Program.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace lldb_private;
|
|
|
|
extern const char *TestMainArgv0;
|
|
|
|
std::string lldb_private::GetInputFilePath(const llvm::Twine &name) {
|
|
llvm::SmallString<128> result = llvm::sys::path::parent_path(TestMainArgv0);
|
|
llvm::sys::fs::make_absolute(result);
|
|
llvm::sys::path::append(result, "Inputs", name);
|
|
return std::string(result.str());
|
|
}
|
|
|
|
llvm::Expected<TestFile> TestFile::fromYaml(llvm::StringRef Yaml) {
|
|
const auto *Info = testing::UnitTest::GetInstance()->current_test_info();
|
|
assert(Info);
|
|
llvm::SmallString<128> Name;
|
|
int FD;
|
|
if (std::error_code EC = llvm::sys::fs::createTemporaryFile(
|
|
llvm::Twine(Info->test_case_name()) + "-" + Info->name(), "test", FD,
|
|
Name))
|
|
return llvm::errorCodeToError(EC);
|
|
llvm::FileRemover Remover(Name);
|
|
{
|
|
llvm::raw_fd_ostream OS(FD, /*shouldClose*/ true);
|
|
llvm::yaml::Input YIn(Yaml);
|
|
if (!llvm::yaml::convertYAML(YIn, OS, [](const llvm::Twine &Msg) {}))
|
|
return llvm::createStringError(llvm::inconvertibleErrorCode(),
|
|
"convertYAML() failed");
|
|
}
|
|
return TestFile(Name, std::move(Remover));
|
|
}
|
|
|
|
llvm::Expected<TestFile> TestFile::fromYamlFile(const llvm::Twine &Name) {
|
|
auto BufferOrError =
|
|
llvm::MemoryBuffer::getFile(GetInputFilePath(Name), /*FileSize*/ -1,
|
|
/*RequiresNullTerminator*/ false);
|
|
if (!BufferOrError)
|
|
return llvm::errorCodeToError(BufferOrError.getError());
|
|
return fromYaml(BufferOrError.get()->getBuffer());
|
|
}
|
|
|
|
TestFile::~TestFile() {
|
|
if (!Name)
|
|
return;
|
|
if (std::error_code EC =
|
|
llvm::sys::fs::remove(*Name, /*IgnoreNonExisting*/ false))
|
|
GTEST_LOG_(WARNING) << "Failed to delete `" << Name->c_str()
|
|
<< "`: " << EC.message();
|
|
}
|