1 files changed, 65 insertions, 0 deletions
diff --git a/gnu/llvm/lldb/unittests/TestingSupport/TestUtilities.cpp b/gnu/llvm/lldb/unittests/TestingSupport/TestUtilities.cpp
new file mode 100644
index 00000000000..f4f6f79bac1
--- /dev/null
+++ b/gnu/llvm/lldb/unittests/TestingSupport/TestUtilities.cpp
@@ -0,0 +1,65 @@
+//===- TestUtilities.cpp ----------------------------------------*- 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 "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 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();
+}