test/test_iterator.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
  25 #include <luabind/luabind.hpp>
  26
  27 #include <vector>
  28 #include <luabind/iterator_policy.hpp>
  29
  30 namespace
  31 {
  32         struct IteratorTest: counted_type<IteratorTest>
  33         {
  34                 IteratorTest()
  35                 {
  36                         names.push_back("first one");
  37                         names.push_back("foobar");
  38                         names.push_back("last one");
  39                 }
  40
  41                 std::vector<std::string> names;
  42         };
  43
  44 } // anonymous namespace
  45
  46 void test_iterator()
  47 {
  48     COUNTER_GUARD(IteratorTest);
  49
  50         lua_state L;
  51
  52         using namespace luabind;
  53
  54         module(L)
  55         [
  56                 class_<IteratorTest>("A")
  57                         .def(constructor<>())
  58                         .def_readonly("names", &IteratorTest::names, return_stl_iterator)
  59         ];
  60
  61         DOSTRING(L,
  62                 "a = A()\n"
  63                 "b = ''\n"
  64                 "for name in a.names do\n"
  65                 "       b = b .. ' ' .. name\n"
  66                 "end\n"
  67                 "assert(b == ' first one foobar last one')");
  68 }