def test_cancel_status_processed(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.processed
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 test_cancel_status_processed(self):
scan, session = MagicMock(), MagicMock()
scan.status = IrmaScanStatus.processed
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 test_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 cancel(scan_id: uuid):
""" Cancel a scan.
The request should be performed using a POST request method.
"""
log.debug("scan %s: cancel", scan_id)
session = db.session
scan = Scan.load_from_ext_id(scan_id, session)
scan_ctrl.cancel(scan, session)
return scan_schema.dump(scan).data
def cancel(scan_id: uuid):
""" Cancel a scan.
The request should be performed using a POST request method.
"""
log.debug("scan %s: cancel", scan_id)
session = db.session
scan = Scan.load_from_ext_id(scan_id, session)
scan_ctrl.cancel(scan, session)
return scan_schema.dump(scan).data
def test_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 test_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 test_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 test_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 test_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 test_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 test_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 test_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 test_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)
