diff options
Diffstat (limited to 'oslo_vmware/tests/test_pbm.py')
-rw-r--r-- | oslo_vmware/tests/test_pbm.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/oslo_vmware/tests/test_pbm.py b/oslo_vmware/tests/test_pbm.py index 2a3b29f..e4dcd10 100644 --- a/oslo_vmware/tests/test_pbm.py +++ b/oslo_vmware/tests/test_pbm.py @@ -23,6 +23,7 @@ import urllib.request as urllib from oslo_vmware import pbm from oslo_vmware.tests import base +from oslo_vmware import vim_util class PBMUtilityTest(base.TestCase): @@ -108,8 +109,7 @@ class PBMUtilityTest(base.TestCase): profile=profile_id) def _create_datastore(self, value): - ds = mock.Mock() - ds.value = value + ds = vim_util.get_moref(value, 'Datastore') return ds def test_convert_datastores_to_hubs(self): @@ -144,7 +144,8 @@ class PBMUtilityTest(base.TestCase): filtered_ds = pbm.filter_datastores_by_hubs(hubs, datastores) self.assertEqual(len(hubs), len(filtered_ds)) - filtered_ds_values = [ds.value for ds in filtered_ds] + filtered_ds_values = [vim_util.get_moref_value(ds) + for ds in filtered_ds] self.assertEqual(set(hub_ids), set(filtered_ds_values)) def test_get_pbm_wsdl_location(self): @@ -180,7 +181,7 @@ class PBMUtilityTest(base.TestCase): session.invoke_api.return_value = profile_id value = 'vm-1' - vm = mock.Mock(value=value) + vm = vim_util.get_moref(value, 'VirtualMachine') ret = pbm.get_profiles(session, vm) self.assertEqual(profile_id, ret) |