modified_testcase

test/test_oneview_id_pools_ipv4_range_facts.py

@@ -17,6 +17,7 @@
 ###
 
 import pytest
+import mock
 
 from hpe_test_utils import OneViewBaseTest
 from oneview_module_loader import IdPoolsIpv4RangeFactsModule
@@ -104,8 +105,7 @@ def test_should_get_all_id_pools_ipv4_ranges(self):
         range_3 = DEFAULT_RANGE_TEMPLATE.copy()
         range_4 = DEFAULT_RANGE_TEMPLATE.copy()
         ranges = [range_2, range_3, range_1, range_4]
-        self.resource.get_by_uri().return_value = self.resource
-        self.resource.data = range_1
+        self.resource.get_by_uri().return_value = range_1
         self.mock_ansible_module.params = PARAMS_GET_ALL
 
         IdPoolsIpv4RangeFactsModule().run()
@@ -116,11 +116,13 @@ def test_should_get_all_id_pools_ipv4_ranges(self):
         )
 
     def test_should_get_all_id_pools_ipv4_ranges_from_subnet(self):
-        self.mock_ov_client.id_pools_ipv4_subnets.get_by_uri.return_value = self.resource
+        obj = mock.Mock()
+        obj.data = DEFAULT_SUBNET_TEMPLATE_2
+        self.mock_ov_client.id_pools_ipv4_subnets.get_by_uri.return_value = obj
         range_1 = DEFAULT_RANGE_TEMPLATE.copy()
         range_4 = DEFAULT_RANGE_TEMPLATE.copy()
         ranges = [range_1, range_4]
-        self.resource.get_by_uri().return_value = self.resource
+        self.resource.get_by_uri.return_value = self.resource
         self.resource.data = range_1
         self.mock_ansible_module.params = PARAMS_GET_ALL_FROM_SUBNET
 
@@ -135,7 +137,7 @@ def test_should_get_id_pools_ipv4_range_from_subnet_and_name(self):
         self.mock_ov_client.id_pools_ipv4_subnets.get.return_value = DEFAULT_SUBNET_TEMPLATE_2
 
         range_1 = DEFAULT_RANGE_TEMPLATE.copy()
-        self.resource.get_by_uri().return_value = self.resource
+        self.resource.get_by_uri.return_value = self.resource
         self.resource.data = range_1
         self.mock_ansible_module.params = PARAMS_GET_BY_NAME_AND_SUBNET_URI