// Copyright (C) 2014 Andrzej Krzemienski.
//
// Use, modification, and distribution is subject to the Boost Software
// License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//
// See http://www.boost.org/lib/optional for documentation.
//
// You are welcome to contact the author at: akrzemi1@gmail.com


#include "boost/optional/optional.hpp"

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#include "boost/core/addressof.hpp"
#include "boost/core/enable_if.hpp"
#include "boost/core/lightweight_test.hpp"
#include "testable_classes.hpp"

using boost::optional;
using boost::none;

struct CountingClass
{
  static int count;
  static int assign_count;
  CountingClass() { ++count; }
  CountingClass(const CountingClass&) { ++count; }
  CountingClass& operator=(const CountingClass&) { ++assign_count; return *this; }
  ~CountingClass() { ++count; }
};

int CountingClass::count = 0;
int CountingClass::assign_count = 0;

void test_no_object_creation()
{
  BOOST_TEST_EQ(0, CountingClass::count);
  BOOST_TEST_EQ(0, CountingClass::assign_count);
  {
    CountingClass v1, v2;
    optional<CountingClass&> oA(v1);
    optional<CountingClass&> oB;
    optional<CountingClass&> oC = oA;
    oB = oA;
    *oB = v2;
    oC = none;
    oC = optional<CountingClass&>(v2);
    oB = none;
    oA = oB;
  }
  BOOST_TEST_EQ(4, CountingClass::count);
  BOOST_TEST_EQ(1, CountingClass::assign_count);
}

template <typename T>
typename boost::enable_if< has_arrow<T> >::type
test_arrow_const()
{
  const typename concrete_type_of<T>::type v(2);
  optional<const T&> o(v);
  BOOST_TEST(o);
  BOOST_TEST_EQ(o->val(), 2);
  BOOST_TEST(boost::addressof(o->val()) == boost::addressof(v.val()));
}

template <typename T>
typename boost::disable_if< has_arrow<T> >::type
test_arrow_const()
{
}

template <typename T>
typename boost::enable_if< has_arrow<T> >::type
test_arrow_noconst_const()
{
  typename concrete_type_of<T>::type v(2);
  optional<const T&> o(v);
  BOOST_TEST(o);
  BOOST_TEST_EQ(o->val(), 2);
  BOOST_TEST(boost::addressof(o->val()) == boost::addressof(v.val()));
  
  v.val() = 1;
  BOOST_TEST(o);
  BOOST_TEST_EQ(o->val(), 1);
  BOOST_TEST_EQ(v.val(), 1);
  BOOST_TEST(boost::addressof(o->val()) == boost::addressof(v.val()));
}

template <typename T>
typename boost::disable_if< has_arrow<T> >::type
test_arrow_noconst_const()
{
}

template <typename T>
typename boost::enable_if< has_arrow<T> >::type
test_arrow()
{
  typename concrete_type_of<T>::type v(2);
  optional<T&> o(v);
  BOOST_TEST(o);
  BOOST_TEST_EQ(o->val(), 2);
  BOOST_TEST(boost::addressof(o->val()) == boost::addressof(v.val()));
  
  v.val() = 1;
  BOOST_TEST(o);
  BOOST_TEST_EQ(o->val(), 1);
  BOOST_TEST_EQ(v.val(), 1);
  BOOST_TEST(boost::addressof(o->val()) == boost::addressof(v.val()));
  
  o->val() = 3;
  BOOST_TEST(o);
  BOOST_TEST_EQ(o->val(), 3);
  BOOST_TEST_EQ(v.val(), 3);
  BOOST_TEST(boost::addressof(o->val()) == boost::addressof(v.val()));
  
}

template <typename T>
typename boost::disable_if< has_arrow<T> >::type
test_arrow()
{
}

template <typename T>
void test_not_containing_value_for()
{
  optional<T&> o1;
  optional<T&> o2 = none;
  optional<T&> o3 = o1;
  
  BOOST_TEST(!o1);
  BOOST_TEST(!o2);
  BOOST_TEST(!o3);
  
  BOOST_TEST(o1 == none);
  BOOST_TEST(o2 == none);
  BOOST_TEST(o3 == none);
}

template <typename T>
void test_direct_init_for_const()
{
  const typename concrete_type_of<T>::type v(2);
  optional<const T&> o(v);
  
  BOOST_TEST(o);
  BOOST_TEST(o != none);
  BOOST_TEST(boost::addressof(*o) == boost::addressof(v));
  BOOST_TEST_EQ(val(*o), val(v));
  BOOST_TEST_EQ(val(*o), 2);
}

template <typename T>
void test_direct_init_for_noconst_const()
{
  typename concrete_type_of<T>::type v(2);
  optional<const T&> o(v);
  
  BOOST_TEST(o);
  BOOST_TEST(o != none);
  BOOST_TEST(boost::addressof(*o) == boost::addressof(v));
  BOOST_TEST_EQ(val(*o), val(v));
  BOOST_TEST_EQ(val(*o), 2);
   
  val(v) = 9;
  BOOST_TEST(boost::addressof(*o) == boost::addressof(v));
  BOOST_TEST_EQ(val(*o), val(v));
  BOOST_TEST_EQ(val(*o), 9);
  BOOST_TEST_EQ(val(v), 9);
}

template <typename T>
void test_direct_init_for()
{
  typename concrete_type_of<T>::type v(2);
  optional<T&> o(v);
  
  BOOST_TEST(o);
  BOOST_TEST(o != none);
  BOOST_TEST(boost::addressof(*o) == boost::addressof(v));
  BOOST_TEST_EQ(val(*o), val(v));
  BOOST_TEST_EQ(val(*o), 2);
   
  val(v) = 9;
  BOOST_TEST(boost::addressof(*o) == boost::addressof(v));
  BOOST_TEST_EQ(val(*o), val(v));
  BOOST_TEST_EQ(val(*o), 9);
  BOOST_TEST_EQ(val(v), 9);
  
  val(*o) = 7;
  BOOST_TEST(boost::addressof(*o) == boost::addressof(v));
  BOOST_TEST_EQ(val(*o), val(v));
  BOOST_TEST_EQ(val(*o), 7);
  BOOST_TEST_EQ(val(v), 7);
}

template <typename T, typename U>
void test_clearing_the_value()
{
  typename concrete_type_of<T>::type v(2);
  optional<U&> o1(v), o2(v);
  
  BOOST_TEST(o1);
  BOOST_TEST(o1 != none);
  BOOST_TEST(o2);
  BOOST_TEST(o2 != none);
  
  o1 = none;
  BOOST_TEST(!o1);
  BOOST_TEST(o1 == none);
  BOOST_TEST(o2);
  BOOST_TEST(o2 != none);
  BOOST_TEST_EQ(val(*o2), 2);
  BOOST_TEST(boost::addressof(*o2) == boost::addressof(v));
  BOOST_TEST_EQ(val(v), 2);
}

template <typename T, typename U>
void test_equality()
{
  typename concrete_type_of<T>::type v1(1), v2(2), v2_(2), v3(3);
  optional<U&> o1(v1), o2(v2), o2_(v2_), o3(v3), o3_(v3), oN, oN_;
  // o2 and o2_ point to different objects; o3 and o3_ point to the same object
  
  BOOST_TEST(oN  == oN);
  BOOST_TEST(oN  == oN_);
  BOOST_TEST(oN_ == oN);
  BOOST_TEST(o1  == o1);
  BOOST_TEST(o2  == o2);
  BOOST_TEST(o2  == o2_);
  BOOST_TEST(o2_ == o2);
  BOOST_TEST(o3  == o3);
  BOOST_TEST(o3  == o3_);
  BOOST_TEST(!(oN == o1));
  BOOST_TEST(!(o1 == oN));
  BOOST_TEST(!(o2 == o1));
  BOOST_TEST(!(o2 == oN));
  
  BOOST_TEST(!(oN  != oN));
  BOOST_TEST(!(oN  != oN_));
  BOOST_TEST(!(oN_ != oN));
  BOOST_TEST(!(o1  != o1));
  BOOST_TEST(!(o2  != o2));
  BOOST_TEST(!(o2  != o2_));
  BOOST_TEST(!(o2_ != o2));
  BOOST_TEST(!(o3  != o3));
  BOOST_TEST(!(o3  != o3_));
  BOOST_TEST( (oN  != o1));
  BOOST_TEST( (o1  != oN));
  BOOST_TEST( (o2  != o1));
  BOOST_TEST( (o2  != oN));
}

template <typename T, typename U>
void test_order()
{
  typename concrete_type_of<T>::type v1(1), v2(2), v2_(2), v3(3);
  optional<U&> o1(v1), o2(v2), o2_(v2_), o3(v3), o3_(v3), oN, oN_;
  // o2 and o2_ point to different objects; o3 and o3_ point to the same object
  
  BOOST_TEST(!(oN  < oN));
  BOOST_TEST(!(oN  < oN_));
  BOOST_TEST(!(oN_ < oN));
  BOOST_TEST(!(o1  < o1));
  BOOST_TEST(!(o2  < o2));
  BOOST_TEST(!(o2  < o2_));
  BOOST_TEST(!(o2_ < o2));
  BOOST_TEST(!(o3  < o3));
  BOOST_TEST(!(o3  < o3_));
  
  BOOST_TEST( (oN  <= oN));
  BOOST_TEST( (oN  <= oN_));
  BOOST_TEST( (oN_ <= oN));
  BOOST_TEST( (o1  <= o1));
  BOOST_TEST( (o2  <= o2));
  BOOST_TEST( (o2  <= o2_));
  BOOST_TEST( (o2_ <= o2));
  BOOST_TEST( (o3  <= o3));
  BOOST_TEST( (o3  <= o3_));
  
  BOOST_TEST(!(oN  > oN));
  BOOST_TEST(!(oN  > oN_));
  BOOST_TEST(!(oN_ > oN));
  BOOST_TEST(!(o1  > o1));
  BOOST_TEST(!(o2  > o2));
  BOOST_TEST(!(o2  > o2_));
  BOOST_TEST(!(o2_ > o2));
  BOOST_TEST(!(o3  > o3));
  BOOST_TEST(!(o3  > o3_));
  
  BOOST_TEST( (oN  >= oN));
  BOOST_TEST( (oN  >= oN_));
  BOOST_TEST( (oN_ >= oN));
  BOOST_TEST( (o1  >= o1));
  BOOST_TEST( (o2  >= o2));
  BOOST_TEST( (o2  >= o2_));
  BOOST_TEST( (o2_ >= o2));
  BOOST_TEST( (o3  >= o3));
  BOOST_TEST( (o3  >= o3_));
  
  BOOST_TEST( (oN  < o1));
  BOOST_TEST( (oN_ < o1));
  BOOST_TEST( (oN  < o2));
  BOOST_TEST( (oN_ < o2));
  BOOST_TEST( (oN  < o2_));
  BOOST_TEST( (oN_ < o2_));
  BOOST_TEST( (oN  < o3));
  BOOST_TEST( (oN_ < o3));
  BOOST_TEST( (oN  < o3_));
  BOOST_TEST( (oN_ < o3_));
  BOOST_TEST( (o1  < o2));
  BOOST_TEST( (o1  < o2_));
  BOOST_TEST( (o1  < o3));
  BOOST_TEST( (o1  < o3_));
  BOOST_TEST( (o2  < o3));
  BOOST_TEST( (o2_ < o3));
  BOOST_TEST( (o2  < o3_));
  BOOST_TEST( (o2_ < o3_));
  
  BOOST_TEST( (oN  <= o1));
  BOOST_TEST( (oN_ <= o1));
  BOOST_TEST( (oN  <= o2));
  BOOST_TEST( (oN_ <= o2));
  BOOST_TEST( (oN  <= o2_));
  BOOST_TEST( (oN_ <= o2_));
  BOOST_TEST( (oN  <= o3));
  BOOST_TEST( (oN_ <= o3));
  BOOST_TEST( (oN  <= o3_));
  BOOST_TEST( (oN_ <= o3_));
  BOOST_TEST( (o1  <= o2));
  BOOST_TEST( (o1  <= o2_));
  BOOST_TEST( (o1  <= o3));
  BOOST_TEST( (o1  <= o3_));
  BOOST_TEST( (o2  <= o3));
  BOOST_TEST( (o2_ <= o3));
  BOOST_TEST( (o2  <= o3_));
  BOOST_TEST( (o2_ <= o3_));
  
  BOOST_TEST(!(oN  > o1));
  BOOST_TEST(!(oN_ > o1));
  BOOST_TEST(!(oN  > o2));
  BOOST_TEST(!(oN_ > o2));
  BOOST_TEST(!(oN  > o2_));
  BOOST_TEST(!(oN_ > o2_));
  BOOST_TEST(!(oN  > o3));
  BOOST_TEST(!(oN_ > o3));
  BOOST_TEST(!(oN  > o3_));
  BOOST_TEST(!(oN_ > o3_));
  BOOST_TEST(!(o1  > o2));
  BOOST_TEST(!(o1  > o2_));
  BOOST_TEST(!(o1  > o3));
  BOOST_TEST(!(o1  > o3_));
  BOOST_TEST(!(o2  > o3));
  BOOST_TEST(!(o2_ > o3));
  BOOST_TEST(!(o2  > o3_));
  BOOST_TEST(!(o2_ > o3_));
  
  BOOST_TEST(!(oN  >= o1));
  BOOST_TEST(!(oN_ >= o1));
  BOOST_TEST(!(oN  >= o2));
  BOOST_TEST(!(oN_ >= o2));
  BOOST_TEST(!(oN  >= o2_));
  BOOST_TEST(!(oN_ >= o2_));
  BOOST_TEST(!(oN  >= o3));
  BOOST_TEST(!(oN_ >= o3));
  BOOST_TEST(!(oN  >= o3_));
  BOOST_TEST(!(oN_ >= o3_));
  BOOST_TEST(!(o1  >= o2));
  BOOST_TEST(!(o1  >= o2_));
  BOOST_TEST(!(o1  >= o3));
  BOOST_TEST(!(o1  >= o3_));
  BOOST_TEST(!(o2  >= o3));
  BOOST_TEST(!(o2_ >= o3));
  BOOST_TEST(!(o2  >= o3_));
  BOOST_TEST(!(o2_ >= o3_));
  
  BOOST_TEST(!(o1  < oN));
  BOOST_TEST(!(o1  < oN_));
  BOOST_TEST(!(o2  < oN));
  BOOST_TEST(!(o2  < oN_));
  BOOST_TEST(!(o2_ < oN));
  BOOST_TEST(!(o2_ < oN_));
  BOOST_TEST(!(o3  < oN));
  BOOST_TEST(!(o3  < oN_));
  BOOST_TEST(!(o3_ < oN));
  BOOST_TEST(!(o3_ < oN_));
  BOOST_TEST(!(o2  < oN));
  BOOST_TEST(!(o2_ < oN_));
  BOOST_TEST(!(o3  < oN));
  BOOST_TEST(!(o3_ < oN_));
  BOOST_TEST(!(o3  < oN));
  BOOST_TEST(!(o3  < oN_));
  BOOST_TEST(!(o3_ < oN));
  BOOST_TEST(!(o3_ < oN_));
}

template <typename T, typename U>
void test_swap()
{
  typename concrete_type_of<T>::type v1(1), v2(2);
  optional<U&> o1(v1), o1_(v1), o2(v2), o2_(v2), oN, oN_;
  
  swap(o1, o1);
  BOOST_TEST(o1);
  BOOST_TEST(boost::addressof(*o1) == boost::addressof(v1));
  
  swap(oN, oN_);
  BOOST_TEST(!oN);
  BOOST_TEST(!oN_);
  
  swap(o1, oN);
  BOOST_TEST(!o1);
  BOOST_TEST(oN);
  BOOST_TEST(boost::addressof(*oN) == boost::addressof(v1));
  
  swap(oN, o1);
  BOOST_TEST(!oN);
  BOOST_TEST(o1);
  BOOST_TEST(boost::addressof(*o1) == boost::addressof(v1));
  
  swap(o1_, o2_);
  BOOST_TEST(o1_);
  BOOST_TEST(o2_);
  BOOST_TEST(boost::addressof(*o1_) == boost::addressof(v2));
  BOOST_TEST(boost::addressof(*o2_) == boost::addressof(v1));
}

template <typename T, typename U>
void test_convertability_of_compatible_reference_types()
{
  typename concrete_type_of<T>::type v1(1);
  optional<T&> oN, o1(v1);
  optional<U&> uN(oN), u1(o1);
  BOOST_TEST(!uN);
  BOOST_TEST(u1);
  BOOST_TEST(boost::addressof(*u1) == boost::addressof(*o1));
  
  uN = o1;
  u1 = oN;
  BOOST_TEST(!u1);
  BOOST_TEST(uN);
  BOOST_TEST(boost::addressof(*uN) == boost::addressof(*o1));
}

template <typename T>
void test_optional_ref()
{
  test_not_containing_value_for<T>();
  test_direct_init_for<T>();
  test_clearing_the_value<T, T>();
  test_arrow<T>();
  test_equality<T, T>();
  test_order<T, T>();
  test_swap<T, T>();
}

template <typename T>
void test_optional_const_ref()
{
  test_not_containing_value_for<const T>();
  test_direct_init_for_const<T>();
  test_direct_init_for_noconst_const<T>();
  test_clearing_the_value<const T, const T>();
  test_clearing_the_value<T, const T>();
  test_arrow_const<T>();
  test_arrow_noconst_const<T>();
  test_equality<const T, const T>();
  test_equality<T, const T>();
  test_order<const T, const T>();
  test_order<T, const T>();
  test_swap<const T, const T>();
  test_swap<T, const T>();
}

int main()
{
  test_optional_ref<int>();
  test_optional_ref<ScopeGuard>();
  test_optional_ref<Abstract>();
  test_optional_ref< optional<int> >();
  
  test_optional_const_ref<int>();
  test_optional_const_ref<ScopeGuard>();
  test_optional_const_ref<Abstract>();
  test_optional_const_ref< optional<int> >();
  
  test_convertability_of_compatible_reference_types<int, const int>();
  test_convertability_of_compatible_reference_types<Impl, Abstract>();
  test_convertability_of_compatible_reference_types<Impl, const Abstract>();
  test_convertability_of_compatible_reference_types<const Impl, const Abstract>();
  test_convertability_of_compatible_reference_types<optional<int>, const optional<int> >();
  
  return boost::report_errors();
}
