smart_ptr/test/weak_from_this_test2.cpp

//
//  weak_from_this_test2.cpp
//
//  Tests weak_from_this in a destructor
//
//  Copyright (c) 2014, 2015 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/enable_shared_from_this.hpp>
#include <boost/weak_ptr.hpp>
#include <boost/core/lightweight_test.hpp>

class X: public boost::enable_shared_from_this< X >
{
private:

    boost::weak_ptr<X> px_;

public:

    X()
    {
        boost::weak_ptr<X> p1 = weak_from_this();
        BOOST_TEST( p1._empty() );
        BOOST_TEST( p1.expired() );
    }

    void check()
    {
        boost::weak_ptr<X> p2 = weak_from_this();
        BOOST_TEST( !p2.expired() );

        BOOST_TEST( p2.lock().get() == this );

        px_ = p2;
    }

    ~X()
    {
        boost::weak_ptr<X> p3 = weak_from_this();
        BOOST_TEST( p3.expired() );

        BOOST_TEST( !(px_ < p3) && !(p3 < px_) );
    }
};

int main()
{
    {
        boost::shared_ptr< X > px( new X );
        px->check();
    }

    return boost::report_errors();
}