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 test_main(lua_State* L)
  43 {
  44     using namespace luabind;
  45
  46     disable_super_deprecation();
  47
  48     module(L) [
  49         class_<Base, Base_wrap>("Base")
  50             .def(constructor<>()),
  51
  52         def("take_ownership", &take_ownership, adopt(_1))
  53     ];
  54
  55     DOSTRING(L,
  56         "x = Base()\n"
  57     );
  58
  59     TEST_CHECK(Base::count == 1);
  60
  61     DOSTRING(L,
  62         "x = nil\n"
  63         "collectgarbage('collect')\n"
  64     );
  65
  66     TEST_CHECK(Base::count == 0);
  67
  68     DOSTRING(L,
  69         "class 'Derived' (Base)\n"
  70         "  function Derived:__init()\n"
  71         "      super()\n"
  72         "      self.x = Base()\n"
  73         "  end\n"
  74     );
  75
  76     DOSTRING(L,
  77         "x = Derived()\n"
  78     );
  79
  80     TEST_CHECK(Base::count == 2);
  81
  82     DOSTRING(L,
  83         "x = nil\n"
  84         "collectgarbage('collect')\n"
  85         "collectgarbage('collect')\n"
  86     );
  87
  88     TEST_CHECK(Base::count == 0);
  89
  90     DOSTRING(L,
  91         "x = nil\n"
  92         "collectgarbage('collect')\n"
  93     );
  94
  95     TEST_CHECK(Base::count == 0);
  96
  97      DOSTRING(L,
  98         "class 'Derived2' (Derived)\n"
  99         "  function Derived2:__init()\n"
 100         "      super()\n"
 101         "  end\n"
 102     );
 103
 104     DOSTRING(L,
 105         "x = Derived2()\n"
 106     );
 107
 108     TEST_CHECK(Base::count == 2);
 109
 110     DOSTRING(L,
 111         "x = nil\n"
 112         "collectgarbage('collect')\n"
 113         "collectgarbage('collect')\n"
 114     );
 115
 116     TEST_CHECK(Base::count == 0);
 117
 118     DOSTRING(L,
 119         "x = Derived()\n"
 120     );
 121
 122     TEST_CHECK(Base::count == 2);
 123
 124     DOSTRING(L,
 125         "take_ownership(x)\n"
 126         "x = nil\n"
 127         "collectgarbage('collect')\n"
 128         "collectgarbage('collect')\n"
 129     );
 130
 131     TEST_CHECK(Base::count == 2);
 132
 133     delete adopted;
 134
 135     DOSTRING(L,
 136         "collectgarbage('collect')\n"
 137         "collectgarbage('collect')\n"
 138     );
 139
 140     TEST_CHECK(Base::count == 0);
 141
 142     DOSTRING(L,
 143         "x = Derived2()\n"
 144     );
 145
 146     TEST_CHECK(Base::count == 2);
 147
 148     DOSTRING(L,
 149         "take_ownership(x)\n"
 150         "x = nil\n"
 151         "collectgarbage('collect')\n"
 152         "collectgarbage('collect')\n"
 153     );
 154
 155     TEST_CHECK(Base::count == 2);
 156
 157     delete adopted;
 158
 159     DOSTRING(L,
 160         "collectgarbage('collect')\n"
 161         "collectgarbage('collect')\n"
 162     );
 163
 164     TEST_CHECK(Base::count == 0);
 165 }