third_party/boost/smart_ptr/test/shared_from_test.cpp

// shared_from_test.cpp
//
// Copyright 2019 Peter Dimov
//
// Distributed under the Boost Software License, Version 1.0.
// http://www.boost.org/LICENSE_1_0.txt)

#include <boost/smart_ptr/enable_shared_from.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/core/lightweight_test.hpp>

//

class X
{
private:

    int m_;

public:

    X(): m_() {}
};

class Y: public boost::enable_shared_from
{
};

class Z: public X, public Y
{
public:

    boost::shared_ptr<Z> shared_from_this()
    {
        return boost::shared_from( this );
    }
};

void null_deleter( void const* )
{
}

int main()
{
    boost::shared_ptr<Z> p( new Z );

    try
    {
        boost::shared_ptr<Z> q = p->shared_from_this();

        BOOST_TEST_EQ( p, q );
        BOOST_TEST( !( p < q ) && !( q < p ) );
    }
    catch( boost::bad_weak_ptr const & )
    {
        BOOST_ERROR( "p->shared_from_this() failed" );
    }

    Z v2( *p );

    try
    {
        boost::shared_ptr<Z> q = v2.shared_from_this();
        BOOST_ERROR( "v2.shared_from_this() failed to throw" );
    }
    catch( boost::bad_weak_ptr const & )
    {
    }

    *p = Z();

    try
    {
        boost::shared_ptr<Z> q = p->shared_from_this();

        BOOST_TEST_EQ( p, q );
        BOOST_TEST( !( p < q ) && !( q < p ) );
    }
    catch( boost::bad_weak_ptr const & )
    {
        BOOST_ERROR( "p->shared_from_this() threw bad_weak_ptr after *p = Z()" );
    }

    {
        boost::shared_ptr<Z> p2( p.get(), null_deleter );
    }

    try
    {
        boost::shared_ptr<Z> q = p->shared_from_this();

        BOOST_TEST_EQ( p, q );
        BOOST_TEST( !( p < q ) && !( q < p ) );
    }
    catch( boost::bad_weak_ptr const& )
    {
        BOOST_ERROR( "p->shared_from_this() failed" );
    }

    return boost::report_errors();
}