def test_get_hostname(self): mock_get_managed_object_name = MagicMock(return_value='fake_hostname') with patch('salt.utils.vmware.get_managed_object_name', mock_get_managed_object_name): vsan.get_host_vsan_system(self.mock_si, self.mock_host_ref) mock_get_managed_object_name.assert_called_once_with( self.mock_host_ref)
def test_hostname_argument(self): mock_get_managed_object_name = MagicMock(return_value='fake_hostname') with patch('salt.utils.vmware.get_managed_object_name', MagicMock(return_value='fake_hostname')): vsan.get_host_vsan_system(self.mock_si, self.mock_host_ref, hostname='passedin_hostname') self.assertEqual(mock_get_managed_object_name.call_count, 0)
def test_traversal_spec(self): mock_traversal_spec = MagicMock(return_value=self.mock_traversal_spec) with patch( 'salt.utils.vmware.vmodl.query.PropertyCollector.TraversalSpec', mock_traversal_spec): vsan.get_host_vsan_system(self.mock_si, self.mock_host_ref) mock_traversal_spec.assert_called_once_with( path='configManager.vsanSystem', type=vim.HostSystem, skip=False)
def test_empty_mors_result(self): mock_get_mors = MagicMock(return_value=None) with patch('salt.utils.vmware.get_mors_with_properties', mock_get_mors): with self.assertRaises(VMwareObjectRetrievalError) as excinfo: vsan.get_host_vsan_system(self.mock_si, self.mock_host_ref) self.assertEqual(excinfo.exception.strerror, 'Host\'s \'fake_hostname\' VSAN system was ' 'not retrieved')
def test_get_mors_with_properties(self): mock_get_mors = \ MagicMock(return_value=[{'object': self.mock_vsan_system}]) with patch('salt.utils.vmware.get_mors_with_properties', mock_get_mors): vsan.get_host_vsan_system(self.mock_si, self.mock_host_ref) mock_get_mors.assert_called_once_with( self.mock_si, vim.HostVsanSystem, property_list=['config.enabled'], container_ref=self.mock_host_ref, traversal_spec=self.mock_traversal_spec)
def test_valid_mors_result(self): res = vsan.get_host_vsan_system(self.mock_si, self.mock_host_ref) self.assertEqual(res, self.mock_vsan_system)