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')