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, [])