updated tests to cope with additional paramater (step_count)

test/test_ims.py

@@ -59,18 +59,16 @@ def test_ims_class_filtered_entities(mocker):
     ds = inventory_provider.db.ims.IMS(
         'dummy_base', 'dummy_username', 'dummy_password', 'dummy_bt')
 
-    list(ds.get_filtered_entities('Node', 'dummy_param=dummy value'))
+    list(ds.get_filtered_entities(
+        'Node', 'dummy_param=dummy value', step_count=50))
     mock_get.assert_called_once_with(
         'dummy_base/ims/Node/filtered/dummy_param=dummy value',
         headers={'Authorization': 'Bearer dummy_bt'},
         params={
             'paginatorStartElement': 0,
-            'paginatorNumberOfElements':
-                inventory_provider.db.ims.IMS.NO_OF_ELEMENTS_PER_ITERATION
+            'paginatorNumberOfElements': 50
         })
 
-    inventory_provider.db.ims.IMS.NO_OF_ELEMENTS_PER_ITERATION = 2
-
     def side_effect(*args, **kargs):
         if kargs['params']['paginatorStartElement'] == 0:
             return MockResponse([1, 2], 200)
@@ -79,7 +77,8 @@ def test_ims_class_filtered_entities(mocker):
     mock_multi_get = mocker.patch(
         'inventory_provider.db.ims.requests.get', side_effect=side_effect)
 
-    res = list(ds.get_filtered_entities('Node', 'dummy_param=dummy value'))
+    res = list(ds.get_filtered_entities(
+        'Node', 'dummy_param=dummy value', step_count=2))
     mock_multi_get.assert_any_call(
         'dummy_base/ims/Node/filtered/dummy_param=dummy value',
         headers={'Authorization': 'Bearer dummy_bt'},
@@ -103,14 +102,13 @@ def test_ims_class_get_all_entities(mocker):
     ds = inventory_provider.db.ims.IMS(
         'dummy_base', 'dummy_username', 'dummy_password', 'dummy_bt')
 
-    list(ds.get_all_entities('Node'))
+    list(ds.get_all_entities('Node', step_count=10))
     mock_get.assert_called_once_with(
         'dummy_base/ims/Node/filtered/Id <> 0',
         headers={'Authorization': 'Bearer dummy_bt'},
         params={
             'paginatorStartElement': 0,
-            'paginatorNumberOfElements':
-                inventory_provider.db.ims.IMS.NO_OF_ELEMENTS_PER_ITERATION
+            'paginatorNumberOfElements': 10
         })