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)
