Fix Cursor handling of lists (#35)

cariad · Jul 21, 2023 · 64d357f · 64d357f
1 parent e3016ef
commit 64d357f
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 2 deletions.
diff --git a/docs/source/package/cursor.rst b/docs/source/package/cursor.rst
@@ -1,5 +1,5 @@
 Cursor
-=====
+======
 
 .. py:module:: rolumns
     :noindex:

diff --git a/rolumns/cursor.py b/rolumns/cursor.py
@@ -40,7 +40,7 @@ def __next__(self) -> Any:
         if isinstance(self._current, list):
             current_list = cast(Iterator[Any], self._current)
             self._resolved = iter(current_list)
-            self._current = next(resolved)
+            self._current = next(self._resolved)
 
         return self.current
 

diff --git a/tests/test_cursor.py b/tests/test_cursor.py
@@ -37,5 +37,25 @@ def test_group() -> None:
     assert child.cursor_group == ByPath("value")
 
 
+def test_list() -> None:
+    cursor = Cursor("value")
+
+    cursor.load(
+        {
+            "value": [
+                "one",
+                "two",
+                "three",
+            ],
+        },
+    )
+
+    assert list(cursor) == [
+        "one",
+        "two",
+        "three",
+    ]
+
+
 def test_str() -> None:
     assert str(Cursor("value")) == 'Cursor(ByPath("value"))'