def test_not_found_decorator(self, mock_get_com_error_hresult): mock_get_com_error_hresult.side_effect = lambda x: x @_utils.not_found_decorator def f(to_call): to_call() to_call = mock.Mock() to_call.side_effect = exceptions.x_wmi( 'expected error', com_error=_utils._WBEM_E_NOT_FOUND) self.assertRaises(exceptions.NotFound, f, to_call) to_call.side_effect = exceptions.x_wmi() self.assertRaises(exceptions.x_wmi, f, to_call)
def test_get_conn_v2(self): self.liveutils._get_wmi_obj.side_effect = exceptions.x_wmi( com_error=mock.Mock()) self.assertRaises(exceptions.HyperVException, self.liveutils._get_conn_v2, '.') self.liveutils._get_wmi_obj.assert_called_once_with( self.liveutils._wmi_namespace % '.', compatibility_mode=True)
def test_remove_multiple_virt_resources_not_found(self, mock_check_ret): excepinfo = [None] * 5 + [jobutils._utils._WBEM_E_NOT_FOUND] mock_check_ret.side_effect = exceptions.x_wmi( 'expected error', com_error=mock.Mock(excepinfo=excepinfo)) vsms_method = self.jobutils._vs_man_svc.RemoveResourceSettings vsms_method.return_value = (mock.sentinel.job, mock.sentinel.ret_val) mock_virt_res = mock.Mock() self.assertRaises(exceptions.NotFound, self.jobutils.remove_virt_resource, mock_virt_res) vsms_method.assert_called_once_with( ResourceSettings=[mock_virt_res.path_.return_value]) mock_check_ret.assert_called_once_with(mock.sentinel.ret_val, mock.sentinel.job)