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