test/test_adopt.cpp

   1 // Copyright Daniel Wallin 2008. Use, modification and distribution is
   2 // subject to the Boost Software License, Version 1.0. (See accompanying
   3 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   4
   5 #include "test.hpp"
   6 #include <luabind/luabind.hpp>
   7 #include <luabind/adopt_policy.hpp>
   8 #include <iostream>
   9
  10 struct Base
  11 {
  12     Base()
  13     {
  14         count++;
  15     }
  16
  17     virtual ~Base()
  18     {
  19         count--;
  20     }
  21
  22     static int count;
  23 };
  24
  25 int Base::count = 0;
  26
  27 struct Base_wrap : Base, luabind::wrap_base
  28 {};
  29
  30 void destroy(Base* p)
  31 {
  32     delete p;
  33 }
  34
  35 Base* adopted = 0;
  36
  37 void take_ownership(Base* p)
  38 {
  39     adopted = p;
  40 }
  41
  42 void not_null(Base* p)
  43 {
  44     TEST_CHECK(p);
  45 }
  46
  47 void test_main(lua_State* L)
  48 {
  49     using namespace luabind;
  50
  51     disable_super_deprecation();
  52
  53     module(L) [
  54         class_<Base, Base_wrap>("Base")
  55             .def(constructor<>()),
  56
  57         def("take_ownership", &take_ownership, adopt(_1)),
  58         def("not_null", &not_null)
  59     ];
  60
  61     DOSTRING(L,
  62         "x = Base()\n"
  63     );
  64
  65     TEST_CHECK(Base::count == 1);
  66
  67     DOSTRING(L,
  68         "x = nil\n"
  69         "collectgarbage('collect')\n"
  70     );
  71
  72     TEST_CHECK(Base::count == 0);
  73
  74     DOSTRING(L,
  75         "class 'Derived' (Base)\n"
  76         "  function Derived:__init()\n"
  77         "      super()\n"
  78         "      self.x = Base()\n"
  79         "  end\n"
  80     );
  81
  82     DOSTRING(L,
  83         "x = Derived()\n"
  84     );
  85
  86     TEST_CHECK(Base::count == 2);
  87
  88     DOSTRING(L,
  89         "x = nil\n"
  90         "collectgarbage('collect')\n"
  91         "collectgarbage('collect')\n"
  92     );
  93
  94     TEST_CHECK(Base::count == 0);
  95
  96     DOSTRING(L,
  97         "x = nil\n"
  98         "collectgarbage('collect')\n"
  99     );
 100
 101     TEST_CHECK(Base::count == 0);
 102
 103      DOSTRING(L,
 104         "class 'Derived2' (Derived)\n"
 105         "  function Derived2:__init()\n"
 106         "      super()\n"
 107         "  end\n"
 108     );
 109
 110     DOSTRING(L,
 111         "x = Derived2()\n"
 112     );
 113
 114     TEST_CHECK(Base::count == 2);
 115
 116     DOSTRING(L,
 117         "x = nil\n"
 118         "collectgarbage('collect')\n"
 119         "collectgarbage('collect')\n"
 120     );
 121
 122     TEST_CHECK(Base::count == 0);
 123
 124     DOSTRING(L,
 125         "x = Derived()\n"
 126     );
 127
 128     TEST_CHECK(Base::count == 2);
 129
 130     DOSTRING(L,
 131         "take_ownership(x)\n"
 132         "x = nil\n"
 133         "collectgarbage('collect')\n"
 134         "collectgarbage('collect')\n"
 135     );
 136
 137     TEST_CHECK(Base::count == 2);
 138
 139     delete adopted;
 140
 141     DOSTRING(L,
 142         "collectgarbage('collect')\n"
 143         "collectgarbage('collect')\n"
 144     );
 145
 146     TEST_CHECK(Base::count == 0);
 147
 148     DOSTRING(L,
 149         "x = Derived2()\n"
 150     );
 151
 152     TEST_CHECK(Base::count == 2);
 153
 154     DOSTRING(L,
 155         "take_ownership(x)\n"
 156         "x = nil\n"
 157         "collectgarbage('collect')\n"
 158         "collectgarbage('collect')\n"
 159     );
 160
 161     TEST_CHECK(Base::count == 2);
 162
 163     delete adopted;
 164
 165     DOSTRING(L,
 166         "collectgarbage('collect')\n"
 167         "collectgarbage('collect')\n"
 168     );
 169
 170     TEST_CHECK(Base::count == 0);
 171
 172     DOSTRING(L,
 173         "x = Derived()\n"
 174         "take_ownership(x)\n"
 175         "not_null(x)\n"
 176     );
 177
 178     delete adopted;
 179
 180     DOSTRING(L,
 181         "x = nil\n"
 182         "collectgarbage('collect')\n"
 183     );
 184
 185     TEST_CHECK(Base::count == 0);
 186 }