File tree Expand file tree Collapse file tree 1 file changed +23
-1
lines changed Expand file tree Collapse file tree 1 file changed +23
-1
lines changed Original file line number Diff line number Diff line change 1
1
import pytest
2
- from pynamodb .pagination import RateLimiter
2
+ from pynamodb .pagination import RateLimiter , PageIterator
3
3
4
4
5
5
class MockTime ():
@@ -77,3 +77,25 @@ def test_basic_rate_limiting_large_increment():
77
77
78
78
# The operation takes longer than the minimum wait, so rate limiting should have no effect
79
79
assert mock_time .time () == 1100.0
80
+
81
+
82
+ def test_page_iterator_with_rate_limit ():
83
+ def mock_operation ():
84
+ pass
85
+
86
+ args = None
87
+ kwargs = {'exclusive_start_key' : None }
88
+ rate_limit = 0.1
89
+ page_iter = PageIterator (mock_operation , args , kwargs , rate_limit )
90
+ assert page_iter ._rate_limiter .rate_limit == rate_limit
91
+
92
+
93
+ def test_page_iterator_page_size_getter ():
94
+ page_iter = PageIterator (None , None , {'limit' : 10 })
95
+ assert page_iter .page_size == 10
96
+
97
+
98
+ def test_page_iterator_page_size_setter ():
99
+ page_iter = PageIterator (None , None , {})
100
+ page_iter .page_size = 20
101
+ assert page_iter ._kwargs ['limit' ] == 20
You can’t perform that action at this time.
0 commit comments