def test012_cancel_status_cancelled(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.cancelled
label = IrmaScanStatus.label[scan.status]
expected = "can not cancel scan in {} status".format(label)
with self.assertRaises(IrmaValueError) as context:
module.cancel(scan, session)
self.assertEqual(str(context.exception), expected)
def test012_cancel_status_cancelled(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.cancelled
label = IrmaScanStatus.label[scan.status]
expected = "can not cancel scan in {} status".format(label)
with self.assertRaises(IrmaValueError) as context:
module.cancel(scan, session)
self.assertEqual(str(context.exception), expected)
def test008_cancel_status_launched_brain_error(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.launched
retcode = IrmaReturnCode.error
ret_val = "reason"
self.celery_brain.scan_cancel.return_value = (retcode, ret_val)
with self.assertRaises(IrmaTaskError) as context:
module.cancel(scan, session)
self.assertEqual(str(context.exception), ret_val)
scan.set_status.assert_not_called()
def test008_cancel_status_launched_brain_error(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.launched
retcode = IrmaReturnCode.error
ret_val = "reason"
self.celery_brain.scan_cancel.return_value = (retcode, ret_val)
with self.assertRaises(IrmaTaskError) as context:
module.cancel(scan, session)
self.assertEqual(str(context.exception),
ret_val)
scan.set_status.assert_not_called()
def test008_cancel_status_launched_status_processed(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.launched
retcode = IrmaReturnCode.success
status = IrmaScanStatus.label[IrmaScanStatus.processed]
cancel_res = {'status': status}
self.celery_brain.scan_cancel.return_value = (retcode, cancel_res)
with self.assertRaises(IrmaValueError) as context:
module.cancel(scan, session)
self.assertEqual(str(context.exception),
"can not cancel scan in {0} status".format(status))
scan.set_status.assert_called_once_with(IrmaScanStatus.processed)
def test008_cancel_status_launched_status_processed(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.launched
retcode = IrmaReturnCode.success
status = IrmaScanStatus.label[IrmaScanStatus.processed]
cancel_res = {'status': status}
self.celery_brain.scan_cancel.return_value = (retcode, cancel_res)
with self.assertRaises(IrmaValueError) as context:
module.cancel(scan, session)
self.assertEqual(str(context.exception),
"can not cancel scan in {0} status".format(status))
scan.set_status.assert_called_once_with(IrmaScanStatus.processed)
def cancel(scanid, db):
""" Cancel a scan.
The request should be performed using a POST request method.
"""
try:
log.debug("scanid: %s", scanid)
validate_id(scanid)
scan = Scan.load_from_ext_id(scanid, db)
scan_ctrl.cancel(scan, db)
response.content_type = "application/json; charset=UTF-8"
return scan_schema.dumps(scan).data
except Exception as e:
log.exception(e)
process_error(e)
def cancel(scanid, db):
""" Cancel a scan.
The request should be performed using a POST request method.
"""
try:
log.debug("scanid: %s", scanid)
validate_id(scanid)
scan = Scan.load_from_ext_id(scanid, db)
scan_ctrl.cancel(scan, db)
response.content_type = "application/json; charset=UTF-8"
return scan_schema.dumps(scan).data
except Exception as e:
log.exception(e)
process_error(e)
def test008_cancel_status_launched_ok(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.launched
retcode = IrmaReturnCode.success
cancel_res = {'cancel_details': "details"}
self.celery_brain.scan_cancel.return_value = (retcode, cancel_res)
res = module.cancel(scan, session)
self.assertEqual(res, cancel_res['cancel_details'])
scan.set_status.assert_called_once_with(IrmaScanStatus.cancelled)
def test008_cancel_status_launched_ok(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.launched
retcode = IrmaReturnCode.success
cancel_res = {'cancel_details': "details"}
self.celery_brain.scan_cancel.return_value = (retcode, cancel_res)
res = module.cancel(scan, session)
self.assertEqual(res, cancel_res['cancel_details'])
scan.set_status.assert_called_once_with(IrmaScanStatus.cancelled)
def test006_cancel_status_ready(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.ready
res = module.cancel(scan, session)
self.assertIsNone(res)
scan.set_status.assert_called_once_with(IrmaScanStatus.cancelled)
def test008_cancel_status_launched_status_error(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.error_ftp_upload
res = module.cancel(scan, session)
self.assertIsNone(res)
scan.set_status.assert_not_called()
def test006_cancel_status_ready(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.ready
res = module.cancel(scan, session)
self.assertIsNone(res)
scan.set_status.assert_called_once_with(IrmaScanStatus.cancelled)
def test008_cancel_status_launched_status_error(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.error_ftp_upload
res = module.cancel(scan, session)
self.assertIsNone(res)
scan.set_status.assert_not_called()