test/test_yield.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 #include "test.hpp"
  24 #include <luabind/luabind.hpp>
  25
  26 struct test_class : counted_type<test_class>
  27 {
  28         test_class() {}
  29         int f() const
  30         {
  31                 return 5;
  32         }
  33 };
  34
  35 int f(int a)
  36 {
  37         return 9;
  38 }
  39
  40 int j(lua_State* L)
  41 {
  42         lua_pushnumber(L, 9);
  43         return lua_yield(L, 1);
  44 }
  45
  46 void f() {}
  47
  48 COUNTER_GUARD(test_class);
  49
  50 void test_main(lua_State* L)
  51 {
  52         using namespace luabind;
  53
  54         module(L)
  55         [
  56                 class_<test_class>("test")
  57                         .def(constructor<>())
  58                         .def("f", &test_class::f, luabind::yield)
  59         ];
  60
  61         DOSTRING(L,
  62                 "function h()\n"
  63                 "       coroutine.yield(4)"
  64                 "end");
  65
  66         {
  67                 lua_State* thread = lua_newthread(L);
  68         TEST_CHECK(resume_function<int>(thread, "h") == 4);
  69         lua_pop(L, 1); // pop thread
  70     }
  71
  72         DOSTRING(L,
  73                 "function g(str)\n"
  74                 "       assert(str == 'foobar')\n"
  75                 "       a = test()"
  76                 "       for i = 1, 10 do\n"
  77                 "               assert(a:f() == i + 4)\n"
  78                 "       end\n"
  79                 "end");
  80
  81         {
  82         lua_State* thread = lua_newthread(L);
  83
  84         TEST_CHECK(resume_function<int>(thread, "g", "foobar") == 5);
  85                 for (int i = 1; i < 10; ++i)
  86                 {
  87                         TEST_CHECK(resume<int>(thread, i + 4) == 5);
  88                 }
  89                 lua_pop(L, 1); // pop thread
  90         }
  91
  92
  93         {
  94         lua_State* thread = lua_newthread(L);
  95                 object g = globals(thread)["g"];
  96
  97         TEST_CHECK(resume_function<int>(g, "foobar") == 5);
  98                 for (int i = 1; i < 10; ++i)
  99                 {
 100                         TEST_CHECK(resume<int>(thread, i + 4) == 5);
 101                 }
 102                 lua_pop(L, 1); // pop thread
 103         }
 104
 105 }
 106