@@ -36,9 +36,10 @@ def test_constructor(self):
3636 assert page .remaining == 3
3737 assert page ._parent is parent
3838 assert page ._item_to_value is item_to_value
39+ assert page .raw_page is None
3940
4041 def test___iter__ (self ):
41- page = page_iterator .Page (None , (), None )
42+ page = page_iterator .Page (None , (), None , None )
4243 assert iter (page ) is page
4344
4445 def test_iterator_calls_parent_item_to_value (self ):
@@ -69,6 +70,18 @@ def test_iterator_calls_parent_item_to_value(self):
6970 item_to_value .assert_called_with (parent , 12 )
7071 assert page .remaining == 97
7172
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+
7285
7386class PageIteratorImpl (page_iterator .Iterator ):
7487 def _next_page (self ):
@@ -116,8 +129,7 @@ def test_pages_property_restart(self):
116129 def test__page_iter_increment (self ):
117130 iterator = PageIteratorImpl (None , None )
118131 page = page_iterator .Page (
119- iterator , ("item" ,), page_iterator ._item_to_value_identity
120- )
132+ iterator , ("item" ,), page_iterator ._item_to_value_identity )
121133 iterator ._next_page = mock .Mock (side_effect = [page , None ])
122134
123135 assert iterator .num_results == 0
@@ -147,11 +159,9 @@ def test__items_iter(self):
147159 # Make pages from mock responses
148160 parent = mock .sentinel .parent
149161 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 )
152163 page2 = page_iterator .Page (
153- parent , (item3 ,), page_iterator ._item_to_value_identity
154- )
164+ parent , (item3 ,), page_iterator ._item_to_value_identity )
155165
156166 iterator = PageIteratorImpl (None , None )
157167 iterator ._next_page = mock .Mock (side_effect = [page1 , page2 , None ])
0 commit comments