test/test_object.cpp

   1 // Copyright (c) 2003 Daniel Wallin and 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
  24 #include "test.hpp"
  25 #include <luabind/luabind.hpp>
  26 #include <luabind/adopt_policy.hpp>
  27 #include <luabind/detail/debug.hpp>
  28 #include <luabind/error.hpp>
  29 #include <luabind/operator.hpp>
  30
  31 #include <boost/lexical_cast.hpp>
  32
  33 #include <utility>
  34
  35 using namespace luabind;
  36
  37 int test_object_param(const object& table)
  38 {
  39         LUABIND_CHECK_STACK(table.interpreter());
  40
  41         object current_object;
  42         current_object = table;
  43         lua_State* L = table.interpreter();
  44
  45         if (type(table) == LUA_TTABLE)
  46         {
  47                 int sum1 = 0;
  48                 for (iterator i(table), e; i != e; ++i)
  49                 {
  50                         assert(type(*i) == LUA_TNUMBER);
  51                         sum1 += object_cast<int>(*i);
  52                 }
  53
  54                 int sum2 = 0;
  55                 for (raw_iterator i(table), e; i != e; ++i)
  56                 {
  57                         assert(type(*i) == LUA_TNUMBER);
  58                         sum2 += object_cast<int>(*i);
  59                 }
  60
  61                 // test iteration of empty table
  62                 object empty_table = newtable(L);
  63                 for (iterator i(empty_table), e; i != e; ++i)
  64                 {}
  65
  66                 table["sum1"] = sum1;
  67                 table["sum2"] = sum2;
  68                 table["blurp"] = 5;
  69                 table["sum3"] = table["sum1"];
  70                 return 0;
  71         }
  72         else
  73         {
  74                 if (type(table) != LUA_TNIL)
  75                 {
  76                         return 1;
  77                 }
  78                 else
  79                 {
  80                         return 2;
  81                 }
  82         }
  83 }
  84
  85 int test_fun()
  86 {
  87         return 42;
  88 }
  89
  90 struct test_param : counted_type<test_param>
  91 {
  92         luabind::object obj;
  93         luabind::object obj2;
  94
  95         bool operator==(test_param const& rhs) const
  96         { return obj == rhs.obj && obj2 == rhs.obj2; }
  97 };
  98
  99 COUNTER_GUARD(test_param);
 100
 101 int test_match(const luabind::object& o)
 102 {
 103         return 0;
 104 }
 105
 106 int test_match(int i)
 107 {
 108         return 1;
 109 }
 110
 111 void test_match_object(
 112         luabind::object p1
 113   , luabind::object p2
 114   , luabind::object p3)
 115 {
 116         p1["ret"] = 1;
 117         p2["ret"] = 2;
 118         p3["ret"] = 3;
 119 }
 120
 121 void test_call(lua_State* L)
 122 {
 123     DOSTRING(L,
 124         "function sum(...)\n"
 125         "    local result = 0\n"
 126         "    for _,x in pairs({...}) do\n"
 127         "        result = result + x\n"
 128         "    end\n"
 129         "    return result\n"
 130         "end\n"
 131     );
 132
 133     object sum = globals(L)["sum"];
 134
 135     TEST_CHECK(object_cast<int>(sum()) == 0);
 136     TEST_CHECK(object_cast<int>(sum(1)) == 1);
 137     TEST_CHECK(object_cast<int>(sum(1,2)) == 3);
 138     TEST_CHECK(object_cast<int>(sum(1,2,3)) == 6);
 139     TEST_CHECK(object_cast<int>(sum(1,2,3,4)) == 10);
 140     TEST_CHECK(object_cast<int>(sum(1,2,3,4,5)) == 15);
 141 }
 142
 143 void test_metatable(lua_State* L)
 144 {
 145         object metatable = newtable(L);
 146         object G = globals(L);
 147
 148         DOSTRING(L,
 149                 "function index_function(table, key)\n"
 150                 "  return key*2\n"
 151                 "end\n"
 152
 153                 "index_table = {}"
 154         );
 155
 156         metatable["__index"] = G["index_function"];
 157         setmetatable(G["index_table"], metatable);
 158
 159         DOSTRING(L,
 160                 "assert(index_table[0] == 0)\n"
 161                 "assert(index_table[1] == 2)\n"
 162                 "assert(index_table[2] == 4)\n"
 163         );
 164
 165         DOSTRING(L,
 166                 "setmetatable(index_table, { ['luabind.metatable'] = 1 })"
 167         );
 168
 169         assert(getmetatable(G["index_table"])["luabind.metatable"] == 1);
 170         assert(type(getmetatable(G["index_table"])["nonexistent"]) == LUA_TNIL);
 171 }
 172
 173 int with_upvalues(lua_State *)
 174 {
 175         return 0;
 176 }
 177
 178 void test_upvalues(lua_State* L)
 179 {
 180         lua_pushnumber(L, 3);
 181         lua_pushnumber(L, 4);
 182         lua_pushcclosure(L, &with_upvalues, 2);
 183
 184         object f(from_stack(L, -1));
 185         lua_pop(L, 1);
 186
 187         assert(getupvalue(f, 1) == 3);
 188         assert(getupvalue(f, 2) == 4);
 189
 190         setupvalue(f, 1, object(L, 4));
 191         assert(getupvalue(f, 1) == 4);
 192         assert(getupvalue(f, 2) == 4);
 193
 194         setupvalue(f, 2, object(L, 5));
 195         assert(getupvalue(f, 1) == 4);
 196         assert(getupvalue(f, 2) == 5);
 197 }
 198
 199 void test_explicit_conversions(lua_State* L)
 200 {
 201         lua_pushcclosure(L, &with_upvalues, 0);
 202         object f(from_stack(L, -1));
 203         lua_pop(L, 1);
 204         assert(tocfunction(f) == &with_upvalues);
 205
 206         int* p = static_cast<int*>(lua_newuserdata(L, sizeof(int)));
 207         *p = 1234;
 208         object x(from_stack(L, -1));
 209         lua_pop(L, 1);
 210         assert(*touserdata<int>(x) == 1234);
 211 }
 212
 213 int with_argument(argument const& arg)
 214 {
 215         return object_cast<int>(arg) * 2;
 216 }
 217
 218 int with_table_argument(argument const& arg, argument const& key)
 219 {
 220         return object_cast<int>(arg[key]);
 221 }
 222
 223 void test_argument(lua_State* L)
 224 {
 225     module(L) [
 226         def("with_argument", &with_argument),
 227         def("with_table_argument", &with_table_argument)
 228     ];
 229
 230     DOSTRING(L,
 231         "assert(with_argument(1) == 2)\n"
 232         "assert(with_argument(2) == 4)\n"
 233     );
 234
 235     DOSTRING(L,
 236         "x = { foo = 1, bar = 2 }\n"
 237
 238         "assert(with_table_argument(x, 'foo') == 1)\n"
 239         "assert(with_table_argument(x, 'bar') == 2)\n"
 240     );
 241 }
 242
 243 void test_bool_convertible(lua_State* L)
 244 {
 245     DOSTRING(L,
 246         "x1 = nil\n"
 247         "x2 = false\n"
 248         "x3 = 0\n"
 249         "x4 = 1\n"
 250     );
 251
 252     DOSTRING(L,
 253         "assert(not x1)\n"
 254         "assert(not x2)\n"
 255         "assert(x3)\n"
 256         "assert(x4)\n"
 257     );
 258
 259     object G = globals(L);
 260
 261     object x1 = G["x1"];
 262     object x2 = G["x2"];
 263     object x3 = G["x3"];
 264     object x4 = G["x4"];
 265
 266     assert(!x1);
 267     assert(!x2);
 268     assert(x3);
 269     assert(x4);
 270
 271     assert(!G["x1"]);
 272     assert(!G["x2"]);
 273     assert(G["x3"]);
 274     assert(G["x4"]);
 275 }
 276
 277 void test_main(lua_State* L)
 278 {
 279         using namespace luabind;
 280
 281         module(L)
 282         [
 283                 def("test_object_param", &test_object_param),
 284                 def("test_fun", &test_fun),
 285                 def("test_match", (int(*)(const luabind::object&))&test_match),
 286                 def("test_match", (int(*)(int))&test_match),
 287                 def("test_match_object", &test_match_object),
 288
 289                 class_<test_param>("test_param")
 290                         .def(constructor<>())
 291                         .def_readwrite("obj", &test_param::obj)
 292                         .def_readonly("obj2", &test_param::obj2)
 293                         .def(const_self == const_self)
 294         ];
 295
 296         object uninitialized;
 297         TEST_CHECK(!uninitialized);
 298         TEST_CHECK(!uninitialized.is_valid());
 299
 300         test_param temp_object;
 301         globals(L)["temp"] = temp_object;
 302         TEST_CHECK(object_cast<test_param>(globals(L)["temp"]) == temp_object);
 303         globals(L)["temp"] = &temp_object;
 304         TEST_CHECK(object_cast<test_param const*>(globals(L)["temp"]) == &temp_object);
 305         TEST_CHECK(globals(L)["temp"] == temp_object);
 306
 307         // test the registry
 308         object reg = registry(L);
 309         reg["__a"] = "foobar";
 310         TEST_CHECK(object_cast<std::string>(registry(L)["__a"]) == "foobar");
 311
 312         DOSTRING(L,
 313                 "t = 2\n"
 314                 "assert(test_object_param(t) == 1)");
 315
 316         DOSTRING(L, "assert(test_object_param(nil) == 2)");
 317         DOSTRING(L, "t = { ['oh'] = 4, 3, 5, 7, 13 }");
 318         DOSTRING(L, "assert(test_object_param(t) == 0)");
 319         DOSTRING(L, "assert(t.sum1 == 4 + 3 + 5 + 7 + 13)");
 320         DOSTRING(L, "assert(t.sum2 == 4 + 3 + 5 + 7 + 13)");
 321         DOSTRING(L, "assert(t.blurp == 5)");
 322
 323         object g = globals(L);
 324         object ret = g["test_fun"]();
 325         TEST_CHECK(object_cast<int>(ret) == 42);
 326
 327         DOSTRING(L, "function test_param_policies(x, y) end");
 328         object test_param_policies = g["test_param_policies"];
 329         int a = type(test_param_policies);
 330         TEST_CHECK(a == LUA_TFUNCTION);
 331
 332         luabind::object obj;
 333         obj = luabind::object();
 334
 335         // call the function and tell lua to adopt the pointer passed as first argument
 336         test_param_policies(5, new test_param())[adopt(_2)];
 337
 338         DOSTRING(L, "assert(test_match(7) == 1)");
 339         DOSTRING(L, "assert(test_match('oo') == 0)");
 340
 341         DOSTRING(L,
 342                 "t = test_param()\n"
 343                 "t.obj = 'foo'\n"
 344                 "assert(t.obj == 'foo')\n"
 345                 "assert(t.obj2 == nil)");
 346
 347         DOSTRING(L,
 348                 "function test_object_policies(a) glob = a\n"
 349                 "return 6\n"
 350                 "end");
 351         object test_object_policies = g["test_object_policies"];
 352         object ret_val = test_object_policies("teststring")[detail::null_type()];
 353         TEST_CHECK(object_cast<int>(ret_val) == 6);
 354         TEST_CHECK(ret_val == 6);
 355         TEST_CHECK(6 == ret_val);
 356         g["temp_val"] = 6;
 357         TEST_CHECK(ret_val == g["temp_val"]);
 358         object temp_val = g["temp_val"];
 359         TEST_CHECK(ret_val == temp_val);
 360
 361         g["temp"] = "test string";
 362         TEST_CHECK(boost::lexical_cast<std::string>(g["temp"]) == "test string");
 363         g["temp"] = 6;
 364         TEST_CHECK(boost::lexical_cast<std::string>(g["temp"]) == "6");
 365
 366         TEST_CHECK(object_cast<std::string>(g["glob"]) == "teststring");
 367         TEST_CHECK(object_cast<std::string>(gettable(g, "glob")) == "teststring");
 368         TEST_CHECK(object_cast<std::string>(rawget(g, "glob")) == "teststring");
 369
 370         object t = newtable(L);
 371         TEST_CHECK(iterator(t) == iterator());
 372         TEST_CHECK(raw_iterator(t) == raw_iterator());
 373
 374         t["foo"] = "bar";
 375
 376         TEST_CHECK(object_cast<std::string>(t["foo"]) == "bar");
 377         TEST_CHECK(object_cast<std::string>(*iterator(t)) == "bar");
 378         TEST_CHECK(object_cast<std::string>(*raw_iterator(t)) == "bar");
 379
 380         t["foo"] = nil; // luabind::nil_type
 381
 382         TEST_CHECK(iterator(t) == iterator());
 383         TEST_CHECK(raw_iterator(t) == raw_iterator());
 384
 385         t["foo"] = "bar";
 386         iterator it1(t);
 387         *it1 = nil;
 388
 389         TEST_CHECK(iterator(t) == iterator());
 390         TEST_CHECK(raw_iterator(t) == raw_iterator());
 391
 392         t["foo"] = "bar";
 393         raw_iterator it2(t);
 394         *it2 = nil;
 395
 396         TEST_CHECK(iterator(t) == iterator());
 397         TEST_CHECK(raw_iterator(t) == raw_iterator());
 398
 399         DOSTRING(L,
 400                 "p1 = {}\n"
 401                 "p2 = {}\n"
 402                 "p3 = {}\n"
 403                 "test_match_object(p1, p2, p3)\n"
 404                 "assert(p1.ret == 1)\n"
 405                 "assert(p2.ret == 2)\n"
 406                 "assert(p3.ret == 3)\n");
 407
 408
 409 #ifndef LUABIND_NO_EXCEPTIONS
 410
 411         try
 412         {
 413                 object not_initialized;
 414                 int i = object_cast<int>(not_initialized);
 415                 (void)i;
 416                 TEST_ERROR("invalid cast succeeded");
 417         }
 418         catch(luabind::cast_failed&) {}
 419
 420 #endif
 421
 422     object not_initialized;
 423     TEST_CHECK(!object_cast_nothrow<int>(not_initialized));
 424          TEST_CHECK(!not_initialized.is_valid());
 425          TEST_CHECK(!not_initialized);
 426
 427     DOSTRING(L, "t = { {1}, {2}, {3}, {4} }");
 428
 429     int inner_sum = 0;
 430
 431     for (iterator i(globals(L)["t"]), e; i != e; ++i)
 432     {
 433         inner_sum += object_cast<int>((*i)[1]);
 434     }
 435
 436     TEST_CHECK(inner_sum == 1 + 2 + 3 + 4);
 437
 438     DOSTRING(L, "t = { {1, 2}, {3, 4}, {5, 6}, {7, 8} }");
 439
 440     inner_sum = 0;
 441
 442     for (iterator i(globals(L)["t"]), e; i != e; ++i)
 443     {
 444         for (iterator j(*i), e; j != e; ++j)
 445         {
 446             inner_sum += object_cast<int>(*j);
 447         }
 448     }
 449
 450     TEST_CHECK(inner_sum == 1 + 2 + 3 + 4 + 5 + 6 + 7 + 8);
 451     TEST_CHECK(object_cast<int>(globals(L)["t"][2][2]) == 4);
 452
 453     test_call(L);
 454     test_metatable(L);
 455     test_upvalues(L);
 456     test_explicit_conversions(L);
 457     test_argument(L);
 458     test_bool_convertible(L);
 459 }
 460