def test_no_datacenters_returned(self):
with patch('salt.utils.vmware.get_datacenters',
MagicMock(return_value=[])):
with self.assertRaises(VMwareObjectRetrievalError) as excinfo:
vmware.get_datacenter(self.mock_si, 'fake_dc1')
self.assertEqual('Datacenter \'fake_dc1\' was not found',
excinfo.exception.strerror)
def test_get_datacenters_call(self):
mock_get_datacenters = MagicMock(return_value=[MagicMock()])
with patch('salt.utils.vmware.get_datacenters',
mock_get_datacenters):
vmware.get_datacenter(self.mock_si, 'fake_dc1')
mock_get_datacenters.assert_called_once_with(
self.mock_si, datacenter_names=['fake_dc1'])
def test_datastore_not_found(self):
with patch('salt.utils.vmware.get_mors_with_properties',
MagicMock(return_value=self.mock_entries)):
with self.assertRaises(VMwareObjectRetrievalError) as excinfo:
vmware.get_datacenter(self.mock_si, 'fake_dc')
self.assertEqual(excinfo.exception.strerror,
'Datacenter \'fake_dc\' was not found')
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_datacenter(self.mock_si, 'fake_dc')
mock_get_mors_with_properties.assert_called_once_with(
self.mock_si, vim.Datacenter, property_list=['name'])
def test_get_datacenter_return(self):
with patch('salt.utils.vmware.get_datacenters',
MagicMock(return_value=[self.mock_dc])):
res = vmware.get_datacenter(self.mock_si, 'fake_dc1')
self.assertEqual(res, self.mock_dc)
def test_datastore_found(self):
with patch('salt.utils.vmware.get_mors_with_properties',
MagicMock(return_value=self.mock_entries)):
res = vmware.get_datacenter(self.mock_si, 'fake_dc2')
self.assertEqual(res, self.mock_dc2)
