def test_wait_for_tasks_call(self):
mock_wait_for_tasks = MagicMock()
with patch('salt.utils.vsan.vsanapiutils.WaitForTasks',
mock_wait_for_tasks):
vsan._wait_for_tasks(self.mock_tasks, self.mock_si)
mock_wait_for_tasks.assert_called_once_with(self.mock_tasks,
self.mock_si)
def test_wait_for_tasks_raises_vmodl_runtime_error(self):
exc = vmodl.RuntimeFault()
exc.msg = 'VimRuntime msg'
with patch('salt.utils.vsan.vsanapiutils.WaitForTasks',
MagicMock(side_effect=exc)):
with self.assertRaises(VMwareRuntimeError) as excinfo:
vsan._wait_for_tasks(self.mock_tasks, self.mock_si)
self.assertEqual(excinfo.exception.strerror, 'VimRuntime msg')
def test_wait_for_tasks_raises_vim_fault(self):
exc = vim.fault.VimFault()
exc.msg = 'VimFault msg'
with patch('salt.utils.vsan.vsanapiutils.WaitForTasks',
MagicMock(side_effect=exc)):
with self.assertRaises(VMwareApiError) as excinfo:
vsan._wait_for_tasks(self.mock_tasks, self.mock_si)
self.assertEqual(excinfo.exception.strerror, 'VimFault msg')
def test_wait_for_tasks_raises_no_permission(self):
exc = vim.fault.NoPermission()
exc.privilegeId = 'Fake privilege'
with patch('salt.utils.vsan.vsanapiutils.WaitForTasks',
MagicMock(side_effect=exc)):
with self.assertRaises(VMwareApiError) as excinfo:
vsan._wait_for_tasks(self.mock_tasks, self.mock_si)
self.assertEqual(excinfo.exception.strerror,
'Not enough permissions. Required privilege: '
'Fake privilege')
