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