aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/test_page_iterator.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/test_page_iterator.py')
-rw-r--r--tests/unit/test_page_iterator.py20
1 files changed, 20 insertions, 0 deletions
diff --git a/tests/unit/test_page_iterator.py b/tests/unit/test_page_iterator.py
index 8359537..97b0657 100644
--- a/tests/unit/test_page_iterator.py
+++ b/tests/unit/test_page_iterator.py
@@ -109,6 +109,26 @@ class TestIterator(object):
assert iterator.next_page_token == token
assert iterator.num_results == 0
+ def test_next(self):
+ iterator = PageIteratorImpl(None, None)
+ page_1 = page_iterator.Page(
+ iterator, ("item 1.1", "item 1.2"), page_iterator._item_to_value_identity
+ )
+ page_2 = page_iterator.Page(
+ iterator, ("item 2.1",), page_iterator._item_to_value_identity
+ )
+ iterator._next_page = mock.Mock(side_effect=[page_1, page_2, None])
+
+ result = next(iterator)
+ assert result == "item 1.1"
+ result = next(iterator)
+ assert result == "item 1.2"
+ result = next(iterator)
+ assert result == "item 2.1"
+
+ with pytest.raises(StopIteration):
+ next(iterator)
+
def test_pages_property_starts(self):
iterator = PageIteratorImpl(None, None)