test/test_lua_classes.cpp

   1 #include <boost/tokenizer.hpp>
   2
   3 #include "test.h"
   4 #include <iostream>
   5
   6 namespace
   7 {
   8         LUABIND_ANONYMOUS_FIX int feedback = 0;
   9
  10         struct base
  11         {
  12                 virtual ~base()
  13                 {
  14 //                      std::cout << "~base\n";
  15                         feedback = -1;
  16                 }
  17
  18                 virtual const char* f()
  19                 {
  20                         return "base";
  21                 }
  22         };
  23
  24         struct baseWrap : base
  25         {
  26
  27                 luabind::object self_;
  28                 baseWrap(const luabind::object& self): self_(self) {}
  29
  30                 static const char* fS(base* obj)
  31                 {
  32                         return obj->base::f();
  33                 }
  34
  35                 virtual const char* f()
  36                 {
  37                         return luabind::call_member<const char*>(self_, "f");
  38                 }
  39         };
  40
  41         struct simple_class
  42         {
  43                 void f()
  44                 {
  45                         feedback = 3;
  46                 }
  47         };
  48
  49         struct no_copy
  50         {
  51                 no_copy() {}
  52                 virtual void f() = 0;
  53         private:
  54                 no_copy(const no_copy&) {}
  55         };
  56
  57         void set_feedback(int n)
  58         { feedback = n; }
  59
  60 } // anonymous namespace
  61
  62
  63 bool test_lua_classes()
  64 {
  65         using namespace luabind;
  66
  67         {
  68         lua_State* L = lua_open();
  69         lua_baselibopen(L);
  70         lua_closer c(L);
  71         int top = lua_gettop(L);
  72
  73         open(L);
  74
  75         module(L)
  76         [
  77                 class_<no_copy>("no copy"),
  78
  79                 class_<base, baseWrap>("base")
  80                         .def(constructor<>())
  81                         .def("f", &baseWrap::fS),
  82
  83                 class_<simple_class>("simple_class")
  84                         .def(constructor<>())
  85                         .def("f", &simple_class::f),
  86
  87                 def("set_feedback", &set_feedback)
  88         ];
  89
  90         dostring(L, "class 'derived' (base)\n"
  91                         "function derived:__init() super() end\n"
  92                         "function derived:f() return 'derived -> ' .. base.f(self) end\n");
  93
  94         dostring(L, "function create_derived() return derived() end");
  95
  96         base* ptr = call_function<base*>(L, "create_derived");
  97
  98         if (std::string("derived -> base") != ptr->f()) return false;
  99
 100         dostring(L, "class 'simple_derative' (simple_class)");
 101         dostring(L, "function simple_derative:__init() super() end\n");
 102         dostring(L, "a = simple_derative()");
 103         dostring(L, "a:f()");
 104         if (feedback != 3) return false;
 105
 106         dostring(L, "class 'simple_lua_class'\n");
 107         dostring(L, "function simple_lua_class:__init()\n"
 108                                                 "super()\n"
 109                                                 "self.test_var = 1\n"
 110                                                 "end\n"
 111                                                 "function simple_lua_class:f()\n"
 112                                                 "g = 5\n"
 113                                                 "end\n");
 114         dostring(L,     "a = simple_lua_class()");
 115         dostring(L,  "a:f()");
 116         if (feedback != 3) return false;
 117         object g = get_globals(L)["g"];
 118         if (object_cast<int>(g) != 5) return false;
 119
 120         if (top != lua_gettop(L)) return false;
 121
 122         dostring(L, "a = derived()");
 123
 124
 125         }
 126
 127         if (feedback != -1) return false;
 128
 129
 130         {
 131                 lua_State* L = lua_open();
 132                 lua_baselibopen(L);
 133                 lua_closer c(L);
 134
 135                 open(L);
 136
 137                 function(L, "set_feedback", &set_feedback);
 138
 139                 dostring(L, "class 'simple'");
 140                 dostring(L, "function simple:__init() set_feedback(321) end");
 141                 dostring(L, "function simple:__finalize() set_feedback(123) end");
 142
 143                 dostring(L, "a = simple()");
 144
 145                 if (feedback != 321) return false;
 146         }
 147
 148         if (feedback != 123) return false;
 149
 150         return true;
 151 }
 152