325 changes: 325 additions & 0 deletions test/atomic_sp_test.cpp
@@ -0,0 +1,325 @@
#include <boost/config.hpp>

// atomic_sp_test.cpp
//
// Copyright 2008, 2017 Peter Dimov
//
// Distributed under 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


#include <boost/smart_ptr/atomic_shared_ptr.hpp>
#include <boost/core/lightweight_test.hpp>
#include <boost/memory_order.hpp>

//

struct X
{
};

#define BOOST_TEST_SP_EQ( p, q ) BOOST_TEST( p == q && !( p < q ) && !( q < p ) )

int main()
{
// default constructor

{
boost::atomic_shared_ptr<X> apx;

boost::shared_ptr<X> p2 = apx.load();

BOOST_TEST_EQ( p2.get(), (X*)0 );
BOOST_TEST_EQ( p2.use_count(), 0 );
}

// shared_ptr constructor

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> p2 = apx.load();

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// shared_ptr assignment

{
boost::shared_ptr<X> px0( new X );
boost::atomic_shared_ptr<X> apx( px0 );

boost::shared_ptr<X> px( new X );
apx = px;

boost::shared_ptr<X> p2 = apx.load();

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// load, w/ mo

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> p2 = apx.load( boost::memory_order_acquire );

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// operator shared_ptr

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> p2 = apx;

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// store

{
boost::shared_ptr<X> px0( new X );
boost::atomic_shared_ptr<X> apx( px0 );

boost::shared_ptr<X> px( new X );
apx.store( px );

boost::shared_ptr<X> p2 = apx.load();

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// store, w/ mo

{
boost::shared_ptr<X> px0( new X );
boost::atomic_shared_ptr<X> apx( px0 );

boost::shared_ptr<X> px( new X );
apx.store( px, boost::memory_order_release );

boost::shared_ptr<X> p2 = apx.load();

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// exchange

{
boost::shared_ptr<X> px0( new X );
boost::atomic_shared_ptr<X> apx( px0 );

boost::shared_ptr<X> px( new X );
boost::shared_ptr<X> p1 = apx.exchange( px );

BOOST_TEST_EQ( px0, p1 );
BOOST_TEST_EQ( px0.use_count(), 2 );
BOOST_TEST_EQ( p1.use_count(), 2 );

boost::shared_ptr<X> p2 = apx.load();

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// exchange, w/ mo

{
boost::shared_ptr<X> px0( new X );
boost::atomic_shared_ptr<X> apx( px0 );

boost::shared_ptr<X> px( new X );
boost::shared_ptr<X> p1 = apx.exchange( px, boost::memory_order_acq_rel );

BOOST_TEST_EQ( px0, p1 );
BOOST_TEST_EQ( px0.use_count(), 2 );
BOOST_TEST_EQ( p1.use_count(), 2 );

boost::shared_ptr<X> p2 = apx.load();

BOOST_TEST_EQ( px, p2 );
BOOST_TEST_EQ( px.use_count(), 3 );
BOOST_TEST_EQ( p2.use_count(), 3 );
}

// compare_exchange_weak

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> px2( new X );
boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_weak( cmp, px2 );
BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_weak( cmp, px2 );
BOOST_TEST( r );
BOOST_TEST_SP_EQ( apx.load(), px2 );
}

// compare_exchange_weak, w/ mo

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> px2( new X );
boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_weak( cmp, px2, boost::memory_order_seq_cst, boost::memory_order_seq_cst );
BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_weak( cmp, px2, boost::memory_order_seq_cst, boost::memory_order_seq_cst );
BOOST_TEST( r );
BOOST_TEST_SP_EQ( apx.load(), px2 );
}

// compare_exchange_weak, rv

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_weak( cmp, boost::shared_ptr<X>() );

BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_weak( cmp, boost::shared_ptr<X>() );

BOOST_TEST( r );
BOOST_TEST( apx.load().get() == 0 );
BOOST_TEST( apx.load().use_count() == 0 );
}

// compare_exchange_weak, rv, w/ mo

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_weak( cmp, boost::shared_ptr<X>(), boost::memory_order_seq_cst, boost::memory_order_seq_cst );

BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_weak( cmp, boost::shared_ptr<X>(), boost::memory_order_seq_cst, boost::memory_order_seq_cst );

BOOST_TEST( r );
BOOST_TEST( apx.load().get() == 0 );
BOOST_TEST( apx.load().use_count() == 0 );
}

// compare_exchange_strong

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> px2( new X );
boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_strong( cmp, px2 );
BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_strong( cmp, px2 );
BOOST_TEST( r );
BOOST_TEST_SP_EQ( apx.load(), px2 );
}

// compare_exchange_strong, w/ mo

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> px2( new X );
boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_strong( cmp, px2, boost::memory_order_seq_cst, boost::memory_order_seq_cst );
BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_strong( cmp, px2, boost::memory_order_seq_cst, boost::memory_order_seq_cst );
BOOST_TEST( r );
BOOST_TEST_SP_EQ( apx.load(), px2 );
}

// compare_exchange_strong, rv

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_strong( cmp, boost::shared_ptr<X>() );

BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_strong( cmp, boost::shared_ptr<X>() );

BOOST_TEST( r );
BOOST_TEST( apx.load().get() == 0 );
BOOST_TEST( apx.load().use_count() == 0 );
}

// compare_exchange_strong, rv, w/ mo

{
boost::shared_ptr<X> px( new X );
boost::atomic_shared_ptr<X> apx( px );

boost::shared_ptr<X> cmp;

bool r = apx.compare_exchange_strong( cmp, boost::shared_ptr<X>(), boost::memory_order_seq_cst, boost::memory_order_seq_cst );

BOOST_TEST( !r );
BOOST_TEST_SP_EQ( apx.load(), px );
BOOST_TEST_SP_EQ( cmp, px );

r = apx.compare_exchange_strong( cmp, boost::shared_ptr<X>(), boost::memory_order_seq_cst, boost::memory_order_seq_cst );

BOOST_TEST( r );
BOOST_TEST( apx.load().get() == 0 );
BOOST_TEST( apx.load().use_count() == 0 );
}

return boost::report_errors();
}