test/test_operators.cpp

   1 #include "test.h"
   2 #include <iostream>
   3
   4 extern "C"
   5 {
   6         #include "lualib.h"
   7 }
   8
   9
  10 namespace
  11 {
  12         LUABIND_ANONYMOUS_FIX int feedback = 0;
  13
  14         struct operator_tester
  15         {
  16                 int operator+(int) const
  17                 {
  18                         feedback = 1;
  19                         return 4;
  20                 }
  21
  22                 float operator-() const
  23                 {
  24                         feedback = 2;
  25                         return 4.6f;
  26                 }
  27
  28                 float operator()() const
  29                 {
  30                         feedback = 3;
  31                         return 3.5f;
  32                 }
  33
  34                 float operator()(int a) const
  35                 {
  36                         feedback = 4;
  37                         return 3.5f + a;;
  38                 }
  39
  40                 float operator()(int a)
  41                 {
  42                         feedback = 7;
  43                         return 3.5f + a;;
  44                 }
  45
  46         };
  47
  48         int operator+(int, const operator_tester&)
  49         {
  50                 feedback = 5;
  51                 return 2;
  52         }
  53
  54         struct operator_tester2
  55         {
  56         };
  57
  58         float operator+(const operator_tester&, const operator_tester2&)
  59         {
  60                 feedback = 6;
  61                 return 6.3f;
  62         }
  63
  64         struct operator_tester3: public operator_tester {};
  65
  66         const operator_tester* make_const_test()
  67         {
  68                 static operator_tester o;
  69                 return &o;
  70         }
  71
  72         std::ostream& operator<<(std::ostream& os, const operator_tester&)
  73         {
  74                 os << "operator_tester"; feedback = 63; return os;
  75         }
  76
  77 } // anonymous namespace
  78
  79 bool test_operators()
  80 {
  81         using namespace luabind;
  82
  83         lua_State* L = lua_open();
  84         lua_closer c(L);
  85         lua_baselibopen(L);
  86
  87         int top = lua_gettop(L);
  88
  89         open(L);
  90
  91         if (feedback != 0) return false;
  92
  93         class_<operator_tester>(L, "operator_tester")
  94                 .def(constructor<>())
  95                 .def(tostring(const_self))
  96                 .def(self + int())
  97                 .def(other<int>() + self)
  98                 .def(-self)
  99                 .def(self())
 100                 .def(const_self(int()))
 101                 .def(self(int()))
 102                 ;
 103
 104         class_<operator_tester2>(L, "operator_tester2")
 105                 .def(constructor<>())
 106                 .def(other<const operator_tester&>() + self);
 107
 108         class_<operator_tester3, bases<operator_tester> >(L, "operator_tester3")
 109                 .def(constructor<>());
 110
 111
 112         function(L, "make_const_test", &make_const_test);
 113
 114         if (feedback != 0) return false;
 115
 116         if (dostring(L, "test = operator_tester()")) return false;
 117         if (dostring(L, "test2 = operator_tester2()")) return false;
 118         if (dostring(L, "test3 = operator_tester3()")) return false;
 119
 120         if (dostring(L, "test()")) return false;
 121         if (feedback != 3) return false;
 122         if (dostring(L, "test(5)")) return false;
 123         if (feedback != 7) return false;
 124
 125         dostring(L, "const_test = make_const_test()");
 126         if (dostring(L, "const_test(5)")) return false;
 127         if (feedback != 4) return false;
 128
 129         if (dostring(L, "a = -test")) return false;
 130         if (feedback != 2) return false;
 131
 132         if (dostring(L, "a = test + test2")) return false;
 133         if (feedback != 6) return false;
 134
 135         if (dostring(L, "a = 2 + test")) return false;
 136         if (feedback != 5) return false;
 137
 138         if (dostring(L, "a = test + 2")) return false;
 139         if (feedback != 1) return false;
 140
 141         if (dostring(L, "a = test3 + 6")) return false;
 142         if (feedback != 1) return false;
 143
 144         if (dostring(L, "a = test3 + test2")) return false;
 145         if (feedback != 6) return false;
 146
 147         if (dostring(L, "a = tostring(test)")) return false;
 148         if (feedback != 63) return false;
 149
 150         if (top != lua_gettop(L)) return false;
 151
 152         return true;
 153 }