Home | History | Annotate | Download | only in Support 
      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 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>(errc::broken_pipe));
     49   EXPECT_TRUE(ErrorOr<void>(errc::success));
     50 
     51 #if LLVM_HAS_CXX11_STDLIB
     52   // Move only types.
     53   EXPECT_EQ(3, **t3());
     54 #endif
     55 }
     56 
     57 struct B {};
     58 struct D : B {};
     59 
     60 TEST(ErrorOr, Covariant) {
     61   ErrorOr<B*> b(ErrorOr<D*>(0));
     62   b = ErrorOr<D*>(0);
     63 
     64 #if LLVM_HAS_CXX11_STDLIB
     65   ErrorOr<std::unique_ptr<B> > b1(ErrorOr<std::unique_ptr<D> >(0));
     66   b1 = ErrorOr<std::unique_ptr<D> >(0);
     67 #endif
     68 }
     69 } // end anon namespace
     70 
     71 struct InvalidArgError {
     72   InvalidArgError() {}
     73   InvalidArgError(std::string S) : ArgName(S) {}
     74   std::string ArgName;
     75 };
     76 
     77 namespace llvm {
     78 template<>
     79 struct ErrorOrUserDataTraits<InvalidArgError> : true_type {
     80   static error_code error() {
     81     return make_error_code(errc::invalid_argument);
     82   }
     83 };
     84 } // end namespace llvm
     85 
     86 ErrorOr<int> t4() {
     87   return InvalidArgError("adena");
     88 }
     89 
     90 ErrorOr<void> t5() {
     91   return InvalidArgError("pie");
     92 }
     93 
     94 namespace {
     95 TEST(ErrorOr, UserErrorData) {
     96   ErrorOr<int> a = t4();
     97   EXPECT_EQ(errc::invalid_argument, a);
     98   EXPECT_EQ("adena", t4().getError<InvalidArgError>().ArgName);
     99 
    100   ErrorOr<void> b = t5();
    101   EXPECT_EQ(errc::invalid_argument, b);
    102   EXPECT_EQ("pie", b.getError<InvalidArgError>().ArgName);
    103 }
    104 } // end anon namespace
    105