def test_get_mors_with_properties_call(self):
mock_get_mors_with_properties = MagicMock(
return_value=[{'name': 'fake_dc', 'object': MagicMock()}])
with patch('salt.utils.vmware.get_mors_with_properties',
mock_get_mors_with_properties):
vmware.get_datacenters(self.mock_si, datacenter_names=['fake_dc1'])
mock_get_mors_with_properties.assert_called_once_with(
self.mock_si, vim.Datacenter, property_list=['name'])
def test_get_all_datastores(self):
with patch('salt.utils.vmware.get_mors_with_properties',
MagicMock(return_value=self.mock_entries)):
res = vmware.get_datacenters(
self.mock_si, get_all_datacenters=True)
self.assertEqual(res, [self.mock_dc1, self.mock_dc2])
def test_datastore_found(self):
with patch('salt.utils.vmware.get_mors_with_properties',
MagicMock(return_value=self.mock_entries)):
res = vmware.get_datacenters(
self.mock_si, datacenter_names=['fake_dc2'])
self.assertEqual(res, [self.mock_dc2])
def test_no_parameters(self):
with patch('salt.utils.vmware.get_mors_with_properties',
MagicMock(return_value=self.mock_entries)):
res = vmware.get_datacenters(self.mock_si)
self.assertEqual(res, [])
def test_get_mors_with_properties_returns_empty_array(self):
with patch('salt.utils.vmware.get_mors_with_properties',
MagicMock(return_value=[])):
res = vmware.get_datacenters(self.mock_si,
datacenter_names=['fake_dc1'])
self.assertEqual(res, [])
