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 #ifdef EXPECT_DEBUG_DEATH 32 EXPECT_DEBUG_DEATH(*a, "Cannot get value when an error exists"); 33 #endif 34 } 35 36 #if LLVM_HAS_CXX11_STDLIB 37 ErrorOr<std::unique_ptr<int> > t3() { 38 return std::unique_ptr<int>(new int(3)); 39 } 40 #endif 41 42 TEST(ErrorOr, Types) { 43 int x; 44 ErrorOr<int&> a(x); 45 *a = 42; 46 EXPECT_EQ(42, x); 47 48 EXPECT_FALSE(ErrorOr<void>(make_error_code(errc::broken_pipe))); 49 EXPECT_TRUE(ErrorOr<void>(make_error_code(errc::success))); 50 51 #if LLVM_HAS_CXX11_STDLIB 52 // Move only types. 53 EXPECT_EQ(3, **t3()); 54 #endif 55 } 56 } // end anon namespace 57 58 struct InvalidArgError { 59 InvalidArgError() {} 60 InvalidArgError(std::string S) : ArgName(S) {} 61 std::string ArgName; 62 }; 63 64 namespace llvm { 65 template<> 66 struct ErrorOrUserDataTraits<InvalidArgError> : true_type { 67 static error_code error() { 68 return make_error_code(errc::invalid_argument); 69 } 70 }; 71 } // end namespace llvm 72 73 ErrorOr<int> t4() { 74 return InvalidArgError("adena"); 75 } 76 77 ErrorOr<void> t5() { 78 return InvalidArgError("pie"); 79 } 80 81 namespace { 82 TEST(ErrorOr, UserErrorData) { 83 ErrorOr<int> a = t4(); 84 EXPECT_EQ(errc::invalid_argument, a); 85 EXPECT_EQ("adena", t4().getError<InvalidArgError>().ArgName); 86 87 ErrorOr<void> b = t5(); 88 EXPECT_EQ(errc::invalid_argument, b); 89 EXPECT_EQ("pie", b.getError<InvalidArgError>().ArgName); 90 } 91 } // end anon namespace 92