test/test_iterator.cpp

   1 #include "test.h"
   2
   3 extern "C"
   4 {
   5         #include "lauxlib.h"
   6         #include "lualib.h"
   7 }
   8
   9 #include <vector>
  10 #include <luabind/iterator_policy.hpp>
  11
  12 namespace
  13 {
  14         LUABIND_ANONYMOUS_FIX int feedback = 0;
  15
  16         struct IteratorTest
  17         {
  18                 IteratorTest()
  19                 {
  20                         names.push_back("first one");
  21                         names.push_back("foobar");
  22                         names.push_back("last one");
  23                 }
  24
  25                 std::vector<std::string> names;
  26         };
  27
  28 } // anonymous namespace
  29
  30 bool test_iterator()
  31 {
  32         using namespace luabind;
  33
  34         lua_State* L = lua_open();
  35         lua_baselibopen(L);
  36         lua_closer c(L);
  37
  38         open(L);
  39
  40         module(L)
  41         [
  42                 class_<IteratorTest>("A")
  43                         .def(constructor<>())
  44                         .def_readonly("names", &IteratorTest::names, return_stl_iterator)
  45         ];
  46
  47         dostring(L, "a = A()");
  48         dostring(L, "b = ''");
  49         dostring(L, "for name in a.names do b = b .. name end");
  50
  51         if (object_cast<std::string>(get_globals(L)["b"]) != "first onefoobarlast one") return false;
  52
  53         return true;
  54 }
  55