unittests/Support/ErrorOrTest.cpp

   1 //===- unittests/ErrorOrTest.cpp - ErrorOr.h tests ------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "llvm/Support/ErrorOr.h"
  11
  12 #include "gtest/gtest.h"
  13
  14 #include <memory>
  15
  16 using namespace llvm;
  17
  18 namespace {
  19
  20 ErrorOr<int> t1() {return 1;}
  21 ErrorOr<int> t2() {return make_error_code(errc::invalid_argument);}
  22
  23 TEST(ErrorOr, SimpleValue) {
  24   ErrorOr<int> a = t1();
  25   EXPECT_TRUE(a);
  26   EXPECT_EQ(1, *a);
  27
  28   a = t2();
  29   EXPECT_FALSE(a);
  30   EXPECT_EQ(errc::invalid_argument, a);
  31   EXPECT_DEBUG_DEATH(*a, "Cannot get value when an error exists");
  32 }
  33
  34 #if LLVM_HAS_CXX11_STDLIB
  35 ErrorOr<std::unique_ptr<int> > t3() {
  36   return std::unique_ptr<int>(new int(3));
  37 }
  38 #endif
  39
  40 TEST(ErrorOr, Types) {
  41   int x;
  42   ErrorOr<int&> a(x);
  43   *a = 42;
  44   EXPECT_EQ(42, x);
  45
  46 #if LLVM_HAS_CXX11_STDLIB
  47   // Move only types.
  48   EXPECT_EQ(3, **t3());
  49 #endif
  50 }
  51 } // end anon namespace
  52
  53 struct InvalidArgError {
  54   InvalidArgError() {}
  55   InvalidArgError(std::string S) : ArgName(S) {}
  56   std::string ArgName;
  57 };
  58
  59 #if LLVM_HAS_CXX11_STDLIB
  60 namespace llvm {
  61 template<>
  62 struct ErrorOrUserDataTraits<InvalidArgError> : std::true_type {
  63   static error_code error() {
  64     return make_error_code(errc::invalid_argument);
  65   }
  66 };
  67 } // end namespace lld
  68
  69 ErrorOr<int> t4() {
  70   return InvalidArgError("adena");
  71 }
  72
  73 namespace {
  74 TEST(ErrorOr, UserErrorData) {
  75   ErrorOr<int> a = t4();
  76   EXPECT_EQ(errc::invalid_argument, a);
  77   EXPECT_EQ("adena", t4().getError<InvalidArgError>().ArgName);
  78 }
  79 } // end anon namespace
  80 #endif