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)