test/test_lua_classes.cpp

   1 // Copyright (c) 2005 Daniel Wallin, Arvid Norberg
   2
   3 // Permission is hereby granted, free of charge, to any person obtaining a
   4 // copy of this software and associated documentation files (the "Software"),
   5 // to deal in the Software without restriction, including without limitation
   6 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
   7 // and/or sell copies of the Software, and to permit persons to whom the
   8 // Software is furnished to do so, subject to the following conditions:
   9
  10 // The above copyright notice and this permission notice shall be included
  11 // in all copies or substantial portions of the Software.
  12
  13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
  14 // ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
  15 // TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
  16 // PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
  17 // SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR
  18 // ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  19 // ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE
  21 // OR OTHER DEALINGS IN THE SOFTWARE.
  22
  23 #include "test.hpp"
  24
  25 #include <luabind/luabind.hpp>
  26 #include <luabind/wrapper_base.hpp>
  27 #include <luabind/adopt_policy.hpp>
  28 #include <boost/shared_ptr.hpp>
  29
  30 namespace luabind {
  31
  32 #ifdef BOOST_NO_ARGUMENT_DEPENDENT_LOOKUP
  33     template<class T>
  34     T* get_pointer(boost::shared_ptr<T> const& p) { return p.get(); }
  35 #endif
  36
  37     template<class A>
  38     boost::shared_ptr<const A>* get_const_holder(boost::shared_ptr<A>*)
  39     {
  40         return 0;
  41     }
  42 }
  43
  44 using namespace luabind;
  45
  46 struct A : counted_type<A>
  47 {
  48     virtual ~A() {}
  49
  50     virtual std::string f()
  51     { return "A:f()"; }
  52
  53     virtual std::string g() const
  54     { return "A:g()"; }
  55 };
  56
  57 struct A_wrap : A, wrap_base
  58 {
  59     std::string f()
  60     {
  61         return call_member<std::string>(this, "f");
  62     }
  63
  64     static std::string default_f(A* p)
  65     { return p->A::f(); }
  66
  67     std::string g() const
  68     {
  69         return call_member<std::string>(this, "g");
  70     }
  71
  72     static std::string default_g(A const* p)
  73     { return p->A::g(); }
  74 };
  75
  76 struct B : A
  77 {
  78     virtual std::string f()
  79     { return "B:f()"; }
  80 };
  81
  82 struct B_wrap : B, wrap_base
  83 {
  84     virtual std::string f()
  85     { return call_member<std::string>(this, "f"); }
  86
  87     static std::string default_f(B* p)
  88     { return p->B::f(); }
  89
  90     virtual std::string g() const
  91     { return call_member<std::string>(this, "g"); }
  92
  93     static std::string default_g(B const* p)
  94     { return p->B::g(); }
  95 };
  96
  97
  98 struct base : counted_type<base>
  99 {
 100     virtual ~base() {}
 101
 102     virtual std::string f()
 103     {
 104         return "base:f()";
 105     }
 106
 107     virtual std::string g() const { return ""; }
 108 };
 109
 110 base* filter(base* p) { return p; }
 111
 112 struct base_wrap : base, wrap_base
 113 {
 114     virtual std::string f()
 115     {
 116                 return call_member<std::string>(this, "f");
 117     }
 118
 119     static std::string default_f(base* p)
 120     {
 121         return p->base::f();
 122     }
 123
 124         virtual std::string g() const
 125         {
 126                 return call_member<std::string>(this, "g");
 127         }
 128 };
 129
 130 struct T_ // vc6.5, don't name your types T!
 131 {
 132         int f(int) { return 1; }
 133 };
 134
 135 struct U : T_
 136 {
 137         int g() { return 3; }
 138         int f(int, int) { return 2; }
 139 };
 140
 141 void test_main(lua_State* L)
 142 {
 143         module(L)
 144         [
 145                 class_<A, A_wrap, boost::shared_ptr<A> >("A")
 146                         .def(constructor<>())
 147                         .def("f", &A::f, &A_wrap::default_f)
 148                         .def("g", &A::g, &A_wrap::default_g),
 149
 150                 class_<B, A, B_wrap, boost::shared_ptr<A> >("B")
 151                         .def(constructor<>())
 152                         .def("f", &B::f, &B_wrap::default_f)
 153                         .def("g", &B::g, &B_wrap::default_g),
 154
 155         def("filter", &filter),
 156
 157         class_<base, base_wrap>("base")
 158                         .def(constructor<>())
 159                         .def("f", &base::f, &base_wrap::default_f)
 160                         .def("g", &base::g),
 161
 162                 class_<T_>("T")
 163                         .def("f", &T_::f),
 164
 165                 class_<U, T_>("U")
 166                         .def(constructor<>())
 167                         .def("f", &U::f)
 168                         .def("g", &U::g)
 169         ];
 170
 171         DOSTRING(L,
 172                 "u = U()\n"
 173                 "assert(u:f(0) == 1)\n"
 174                 "assert(u:f(0,0) == 2)\n"
 175                 "assert(u:g() == 3)\n");
 176
 177         DOSTRING(L,
 178                 "u = U()\n"
 179                 "assert(u:f(0) == 1)\n"
 180                 "assert(u:f(0,0) == 2)\n"
 181                 "assert(u:g() == 3)\n");
 182
 183         DOSTRING(L,
 184                 "function base:fun()\n"
 185                 "  return 4\n"
 186                 "end\n"
 187                 "ba = base()\n"
 188                 "assert(ba:fun() == 4)");
 189
 190     DOSTRING(L,
 191         "class 'derived' (base)\n"
 192         "  function derived:__init() super() end\n"
 193         "  function derived:f()\n"
 194         "    return 'derived:f() : ' .. base.f(self)\n"
 195         "  end\n"
 196
 197                 "class 'empty_derived' (base)\n"
 198                 "  function empty_derived:__init() super() end\n"
 199
 200                 "class 'C' (B)\n"
 201         "  function C:__init() super() end\n"
 202                 "  function C:f() return 'C:f()' end\n"
 203
 204         "function make_derived()\n"
 205         "  return derived()\n"
 206         "end\n"
 207
 208         "function make_empty_derived()\n"
 209         "  return empty_derived()\n"
 210         "end\n"
 211
 212                 "function adopt_ptr(x)\n"
 213                 "  a = x\n"
 214                 "end\n");
 215
 216         DOSTRING(L,
 217                 "function gen_error()\n"
 218                 "  assert(0 == 1)\n"
 219                 "end\n");
 220
 221         DOSTRING(L,
 222                 "a = A()\n"
 223                 "b = B()\n"
 224                 "c = C()\n"
 225
 226                 "assert(c:f() == 'C:f()')\n"
 227                 "assert(b:f() == 'B:f()')\n"
 228                 "assert(a:f() == 'A:f()')\n"
 229                 "assert(b:g() == 'A:g()')\n"
 230                 "assert(c:g() == 'A:g()')\n"
 231
 232                 "assert(C.f(c) == 'C:f()')\n"
 233                 "assert(B.f(c) == 'B:f()')\n"
 234                 "assert(A.f(c) == 'A:f()')\n"
 235                 "assert(A.g(c) == 'A:g()')\n");
 236
 237 #ifndef LUABIND_NO_EXCEPTONS
 238         {
 239                 LUABIND_CHECK_STACK(L);
 240
 241                 try { call_function<int>(L, "gen_error"); }
 242                 catch (luabind::error&)
 243                 {
 244             bool result(
 245                 lua_tostring(L, -1) == std::string("[string \"function "
 246                     "gen_error()...\"]:2: assertion failed!"));
 247                         TEST_CHECK(result);
 248                         lua_pop(L, 1);
 249                 }
 250         }
 251
 252         {
 253                 A a;
 254
 255                 DOSTRING(L, "function test_ref(x) end");
 256                 call_function<void>(L, "test_ref", boost::ref(a));
 257         }
 258
 259         {
 260                 LUABIND_CHECK_STACK(L);
 261
 262                 try { call_function<void>(L, "gen_error"); }
 263                 catch (luabind::error&)
 264                 {
 265             bool result(
 266                 lua_tostring(L, -1) == std::string("[string \"function "
 267                     "gen_error()...\"]:2: assertion failed!"));
 268                         TEST_CHECK(result);
 269                         lua_pop(L, 1);
 270                 }
 271         }
 272
 273         {
 274                 LUABIND_CHECK_STACK(L);
 275
 276                 try { call_function<void*>(L, "gen_error") [ adopt(result) ]; }
 277                 catch (luabind::error&)
 278                 {
 279             bool result(
 280                 lua_tostring(L, -1) == std::string("[string \"function "
 281                     "gen_error()...\"]:2: assertion failed!"));
 282                         TEST_CHECK(result);
 283                         lua_pop(L, 1);
 284                 }
 285         }
 286
 287 #endif
 288
 289         base* ptr;
 290         {
 291                 LUABIND_CHECK_STACK(L);
 292
 293                 TEST_NOTHROW(
 294                         object a = globals(L)["ba"];
 295                         TEST_CHECK(call_member<int>(a, "fun") == 4);
 296                 );
 297         }
 298
 299         {
 300                 LUABIND_CHECK_STACK(L);
 301
 302                 object make_derived = globals(L)["make_derived"];
 303                 TEST_NOTHROW(
 304                         call_function<void>(make_derived)
 305                         );
 306         }
 307
 308         std::auto_ptr<base> own_ptr;
 309         {
 310                 LUABIND_CHECK_STACK(L);
 311
 312                 TEST_NOTHROW(
 313                     own_ptr = std::auto_ptr<base>(
 314                 call_function<base*>(L, "make_derived") [ adopt(result) ])
 315                         );
 316         }
 317
 318         // make sure the derived lua-part is still referenced by
 319         // the adopted c++ pointer
 320         DOSTRING(L,
 321                 "collectgarbage()\n"
 322                 "collectgarbage()\n"
 323                 "collectgarbage()\n"
 324                 "collectgarbage()\n"
 325                 "collectgarbage()\n");
 326
 327     TEST_NOTHROW(
 328         TEST_CHECK(own_ptr->f() == "derived:f() : base:f()")
 329     );
 330         own_ptr = std::auto_ptr<base>();
 331
 332         // test virtual functions that are not overridden by lua
 333     TEST_NOTHROW(
 334         own_ptr = std::auto_ptr<base>(
 335             call_function<base*>(L, "make_empty_derived") [ adopt(result) ])
 336         );
 337     TEST_NOTHROW(
 338         TEST_CHECK(own_ptr->f() == "base:f()")
 339         );
 340     TEST_NOTHROW(
 341         call_function<void>(L, "adopt_ptr", own_ptr.get()) [ adopt(_1) ]
 342     );
 343         own_ptr.release();
 344
 345         // test virtual functions that are overridden by lua
 346     TEST_NOTHROW(
 347         ptr = call_function<base*>(L, "derived")
 348     );
 349
 350     TEST_NOTHROW(
 351         TEST_CHECK(ptr->f() == "derived:f() : base:f()")
 352     );
 353
 354         // test virtual function dispatch from within lua
 355         DOSTRING(L,
 356                 "a = derived()\n"
 357         "b = filter(a)\n"
 358         "assert(b:f() == 'derived:f() : base:f()')\n");
 359
 360         // test back references
 361         DOSTRING(L,
 362                 "a = derived()\n"
 363                 "assert(a == filter(a))\n");
 364 }
 365