summaryrefslogtreecommitdiff
path: root/include/llvm/Testing/Support/SupportHelpers.h
blob: c4dd414b80dbc28a1e1ace054e1f9c6cf6a29469 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
//===- Testing/Support/SupportHelpers.h -----------------------------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_TESTING_SUPPORT_SUPPORTHELPERS_H
#define LLVM_TESTING_SUPPORT_SUPPORTHELPERS_H

#include "llvm/ADT/StringRef.h"
#include "llvm/Support/Error.h"
#include "gtest/gtest-printers.h"

namespace llvm {
namespace detail {
struct ErrorHolder {
  bool Success;
  std::string Message;
};

template <typename T> struct ExpectedHolder : public ErrorHolder {
  Optional<T *> Value;
};

inline void PrintTo(const ErrorHolder &Err, std::ostream *Out) {
  *Out << (Err.Success ? "succeeded" : "failed");
  if (!Err.Success) {
    *Out << "  (" << StringRef(Err.Message).trim().str() << ")";
  }
}

template <typename T>
void PrintTo(const ExpectedHolder<T> &Item, std::ostream *Out) {
  if (Item.Success) {
    *Out << "succeeded with value \"" << ::testing::PrintToString(**Item.Value)
         << "\"";
  } else {
    PrintTo(static_cast<const ErrorHolder &>(Item), Out);
  }
}
} // namespace detail
} // namespace llvm

#endif