V-Gears/lib/luabind/test/test_automatic_smart_ptr.cpp

// Copyright Daniel Wallin 2009. 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)

#include "test.hpp"
#include <luabind/luabind.hpp>
#include <boost/shared_ptr.hpp>

struct X
{
    X(int value)
      : value(value)
    {
        ++alive;
    }

    ~X()
    {
        --alive;
    }

    int value;

    static int alive;
};

int X::alive = 0;

struct ptr
{
    ptr(X* p)
      : p(p)
    {}

    ptr(ptr const& other)
      : p(other.p)
    {
        const_cast<ptr&>(other).p = 0;
    }

    ~ptr()
    {
        delete p;
    }

    X* p;
};

X* get_pointer(ptr const& p)
{
    return p.p;
}

std::auto_ptr<X> make1()
{
    return std::auto_ptr<X>(new X(1));
}

boost::shared_ptr<X> make2()
{
    return boost::shared_ptr<X>(new X(2));
}

ptr make3()
{
    return ptr(new X(3));
}

void test_main(lua_State* L)
{
    using namespace luabind;

    module(L) [
        class_<X>("X")
          .def_readonly("value", &X::value),

        def("make1", make1),
        def("make2", make2),
        def("make3", make3)
    ];

    DOSTRING(L,
        "x1 = make1()\n"
        "x2 = make2()\n"
        "x3 = make3()\n"
    );

    TEST_CHECK(X::alive == 3);

    DOSTRING(L,
        "assert(x1.value == 1)\n"
        "assert(x2.value == 2)\n"
        "assert(x3.value == 3)\n"
    );

    DOSTRING(L,
        "x1 = nil\n"
        "x2 = nil\n"
        "x3 = nil\n"
        "collectgarbage()\n"
    );

    assert(X::alive == 0);
}