@@ -36,9 +36,10 @@ def test_constructor(self):
36
36
assert page .remaining == 3
37
37
assert page ._parent is parent
38
38
assert page ._item_to_value is item_to_value
39
+ assert page .raw_page is None
39
40
40
41
def test___iter__ (self ):
41
- page = page_iterator .Page (None , (), None )
42
+ page = page_iterator .Page (None , (), None , None )
42
43
assert iter (page ) is page
43
44
44
45
def test_iterator_calls_parent_item_to_value (self ):
@@ -69,6 +70,18 @@ def test_iterator_calls_parent_item_to_value(self):
69
70
item_to_value .assert_called_with (parent , 12 )
70
71
assert page .remaining == 97
71
72
73
+ def test_raw_page (self ):
74
+ parent = mock .sentinel .parent
75
+ item_to_value = mock .sentinel .item_to_value
76
+
77
+ raw_page = mock .sentinel .raw_page
78
+
79
+ page = page_iterator .Page (parent , (1 , 2 , 3 ), item_to_value , raw_page = raw_page )
80
+ assert page .raw_page is raw_page
81
+
82
+ with pytest .raises (AttributeError ):
83
+ page .raw_page = None
84
+
72
85
73
86
class PageIteratorImpl (page_iterator .Iterator ):
74
87
def _next_page (self ):
@@ -116,8 +129,7 @@ def test_pages_property_restart(self):
116
129
def test__page_iter_increment (self ):
117
130
iterator = PageIteratorImpl (None , None )
118
131
page = page_iterator .Page (
119
- iterator , ("item" ,), page_iterator ._item_to_value_identity
120
- )
132
+ iterator , ("item" ,), page_iterator ._item_to_value_identity )
121
133
iterator ._next_page = mock .Mock (side_effect = [page , None ])
122
134
123
135
assert iterator .num_results == 0
@@ -147,11 +159,9 @@ def test__items_iter(self):
147
159
# Make pages from mock responses
148
160
parent = mock .sentinel .parent
149
161
page1 = page_iterator .Page (
150
- parent , (item1 , item2 ), page_iterator ._item_to_value_identity
151
- )
162
+ parent , (item1 , item2 ), page_iterator ._item_to_value_identity )
152
163
page2 = page_iterator .Page (
153
- parent , (item3 ,), page_iterator ._item_to_value_identity
154
- )
164
+ parent , (item3 ,), page_iterator ._item_to_value_identity )
155
165
156
166
iterator = PageIteratorImpl (None , None )
157
167
iterator ._next_page = mock .Mock (side_effect = [page1 , page2 , None ])
0 commit comments