def is_finished(scanid):
with session_transaction() as session:
scan = Scan.load_from_ext_id(scanid, session=session)
if scan.finished() and scan.status != IrmaScanStatus.finished:
scan.set_status(IrmaScanStatus.finished)
session.commit()
# launch flush celery task on brain
log.debug("scanid: %s calling scan_flush", scan.external_id)
celery_brain.scan_flush(scan.external_id)
def test007_scan_flush(self, m_async_call):
arg = "test"
result = module.scan_flush(arg)
m_async_call.assert_called_once_with(self.m_brain_app,
BRAIN_SCAN_TASKS,
"scan_flush",
args=[arg])
self.assertEqual(result, m_async_call())
def test007_scan_flush(self, m_async_call):
arg = "test"
result = module.scan_flush(arg)
m_async_call.assert_called_once_with(self.m_brain_app,
BRAIN_SCAN_TASKS,
"scan_flush",
args=[arg])
self.assertEqual(result, m_async_call())
def test007_scan_flush(self):
arg = "test"
expected = ((self.m_brain_app, "brain.tasks", "scan_flush"),
{"args": [arg]})
with patch("frontend.controllers.braintasks.async_call") as mock:
result = module.scan_flush(arg)
self.assertTrue(mock.called)
self.assertEqual(mock.call_args, expected)
self.assertEqual(result, mock())
