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 namespace llvm { 60 template<> 61 struct ErrorOrUserDataTraits<InvalidArgError> : std::true_type { 62 static error_code error() { 63 return make_error_code(errc::invalid_argument); 64 } 65 }; 66 } // end namespace lld 67 68 ErrorOr<int> t4() { 69 return InvalidArgError("adena"); 70 } 71 72 namespace { 73 TEST(ErrorOr, UserErrorData) { 74 ErrorOr<int> a = t4(); 75 EXPECT_EQ(errc::invalid_argument, a); 76 EXPECT_EQ("adena", t4().getError<InvalidArgError>().ArgName); 77 } 78 } // end anon namespace 79