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)