@@ -270,68 +270,72 @@ def test_mask_call_invalid_mask(self, make_api_call):
270270 else :
271271 self .fail ('No exception raised' )
272272
273+ @patch ('SoftLayer.API.Client.iter_call' )
274+ def test_iterate (self , _iter_call ):
275+ self .client ['SERVICE' ].METHOD (iter = True )
276+ _iter_call .assert_called_with ('SoftLayer_SERVICE' , 'METHOD' , iter = True )
277+
278+ @patch ('SoftLayer.API.Client.iter_call' )
279+ def test_service_iter_call (self , _iter_call ):
280+ self .client ['SERVICE' ].iter_call ('METHOD' )
281+ _iter_call .assert_called_with ('SoftLayer_SERVICE' , 'METHOD' )
282+
273283 @patch ('SoftLayer.API.Client.call' )
274- def test_iterate (self , _call ):
284+ def test_iter_call (self , _call ):
275285 # chunk=100, no limit
276286 _call .side_effect = [range (100 ), range (100 , 125 )]
277- result = list (self .client [ 'SERVICE' ]. METHOD ( iter = True ))
287+ result = list (self .client . iter_call ( 'SERVICE' , ' METHOD' , iter = True ))
278288
279289 self .assertEquals (range (125 ), result )
280290 _call .assert_has_calls ([
281- call ('SoftLayer_SERVICE' , 'METHOD' ,
282- limit = 100 , iter = True , offset = 0 ),
283- call ('SoftLayer_SERVICE' , 'METHOD' ,
284- limit = 100 , iter = True , offset = 100 ),
291+ call ('SERVICE' , 'METHOD' , limit = 100 , iter = False , offset = 0 ),
292+ call ('SERVICE' , 'METHOD' , limit = 100 , iter = False , offset = 100 ),
285293 ])
286294 _call .reset_mock ()
287295
288296 # chunk=100, no limit. Requires one extra request.
289297 _call .side_effect = [range (100 ), range (100 , 200 ), []]
290- result = list (self .client [ 'SERVICE' ]. METHOD ( iter = True ))
298+ result = list (self .client . iter_call ( 'SERVICE' , ' METHOD' , iter = True ))
291299 self .assertEquals (range (200 ), result )
292300 _call .assert_has_calls ([
293- call ('SoftLayer_SERVICE' , 'METHOD' ,
294- limit = 100 , iter = True , offset = 0 ),
295- call ('SoftLayer_SERVICE' , 'METHOD' ,
296- limit = 100 , iter = True , offset = 100 ),
297- call ('SoftLayer_SERVICE' , 'METHOD' ,
298- limit = 100 , iter = True , offset = 200 ),
301+ call ('SERVICE' , 'METHOD' , limit = 100 , iter = False , offset = 0 ),
302+ call ('SERVICE' , 'METHOD' , limit = 100 , iter = False , offset = 100 ),
303+ call ('SERVICE' , 'METHOD' , limit = 100 , iter = False , offset = 200 ),
299304 ])
300305 _call .reset_mock ()
301306
302307 # chunk=25, limit=30
303308 _call .side_effect = [range (0 , 25 ), range (25 , 30 )]
304- result = list (
305- self . client [ 'SERVICE' ]. METHOD ( iter = True , limit = 30 , chunk = 25 ))
309+ result = list (self . client . iter_call (
310+ 'SERVICE' , ' METHOD' , iter = True , limit = 30 , chunk = 25 ))
306311 self .assertEquals (range (30 ), result )
307312 _call .assert_has_calls ([
308- call ('SoftLayer_SERVICE ' , 'METHOD' , iter = True , limit = 25 , offset = 0 ),
309- call ('SoftLayer_SERVICE ' , 'METHOD' , iter = True , limit = 5 , offset = 25 ),
313+ call ('SERVICE ' , 'METHOD' , iter = False , limit = 25 , offset = 0 ),
314+ call ('SERVICE ' , 'METHOD' , iter = False , limit = 5 , offset = 25 ),
310315 ])
311316 _call .reset_mock ()
312317
313318 # A non-list was returned
314319 _call .side_effect = ["test" ]
315- result = list (self .client [ 'SERVICE' ]. METHOD ( iter = True ))
320+ result = list (self .client . iter_call ( 'SERVICE' , ' METHOD' , iter = True ))
316321 self .assertEquals (["test" ], result )
317322 _call .assert_has_calls ([
318- call ('SoftLayer_SERVICE' , 'METHOD' ,
319- iter = True , limit = 100 , offset = 0 ),
323+ call ('SERVICE' , 'METHOD' , iter = False , limit = 100 , offset = 0 ),
320324 ])
321325 _call .reset_mock ()
322326
323327 # chunk=25, limit=30, offset=12
324328 _call .side_effect = [range (0 , 25 ), range (25 , 30 )]
325- result = list (self .client [ 'SERVICE' ]. METHOD (
326- iter = True , limit = 30 , chunk = 25 , offset = 12 ))
329+ result = list (self .client . iter_call (
330+ 'SERVICE' , 'METHOD' , iter = True , limit = 30 , chunk = 25 , offset = 12 ))
327331 self .assertEquals (range (30 ), result )
328332 _call .assert_has_calls ([
329- call ('SoftLayer_SERVICE' , 'METHOD' ,
330- iter = True , limit = 25 , offset = 12 ),
331- call ('SoftLayer_SERVICE' , 'METHOD' , iter = True , limit = 5 , offset = 37 ),
333+ call ('SERVICE' , 'METHOD' , iter = False , limit = 25 , offset = 12 ),
334+ call ('SERVICE' , 'METHOD' , iter = False , limit = 5 , offset = 37 ),
332335 ])
333336
334337 # Chunk size of 0 is invalid
335338 self .assertRaises (
336339 AttributeError ,
337- lambda : list (self .client ['SERVICE' ].METHOD (iter = True , chunk = 0 )))
340+ lambda : list (self .client .iter_call (
341+ 'SERVICE' , 'METHOD' , iter = True , chunk = 0 )))
0 commit comments