def test_flush(self, m_ftp_ctrl): self.assertNotEqual(self.scan.status, IrmaScanStatus.flushed) j1, j2 = MagicMock(), MagicMock() self.scan.jobs = [j1, j2] module.flush(self.scan, self.session) m_ftp_ctrl.flush.assert_called_once_with(self.ftpuser, self.scan.files) self.session.delete.assert_any_call(j1) self.session.delete.assert_any_call(j2) self.assertEqual(self.scan.status, IrmaScanStatus.flushed)
def scan_flush(scan_id): try: log.info("scan %s: flush requested", scan_id) with session_transaction() as session: user = User.get_by_rmqvhost(session) scan = Scan.get_scan(scan_id, user.id, session) scan_ctrl.flush(scan, session) except Exception as e: log.exception(e) return
def scan_flush(frontend_scan_id): try: log.info("scan_id %s: scan flush requested", frontend_scan_id) with session_transaction() as session: user = User.get_by_rmqvhost(session) scan = Scan.get_scan(frontend_scan_id, user.id, session) scan_ctrl.flush(scan, session) except Exception as e: log.exception(e) return
def scan_flush(scan_id): try: log.info("scan %s: flush requested", scan_id) with session_transaction() as session: user = User.get_by_rmqvhost(session) scan = Scan.get_scan(scan_id, user.id, session) scan_ctrl.flush(scan, session) except Exception as e: log.exception(type(e).__name__ + " : " + str(e)) return
def test_flush(self, m_ftp_ctrl): self.assertNotEqual(self.scan.status, IrmaScanStatus.flushed) j1, j2 = MagicMock(), MagicMock() self.scan.jobs = [j1, j2] module.flush(self.scan, self.session) m_ftp_ctrl.flush.assert_called_once_with(self.ftpuser, self.scan.files) self.session.delete.assert_any_call(j1) self.session.delete.assert_any_call(j2) self.assertEqual(self.scan.status, IrmaScanStatus.flushed)
def scan_flush(frontend_scanid): try: log.info("scan_id %s scan flush requested", frontend_scanid) rmqvhost = config.get_frontend_rmqvhost() user_id = user_ctrl.get_userid(rmqvhost) scan_id = scan_ctrl.get_scan_id(frontend_scanid, user_id) scan_ctrl.flush(scan_id) ftpuser = user_ctrl.get_ftpuser(user_id) ftp_ctrl.flush_dir(ftpuser, frontend_scanid) except: log.info("exception", exc_info=True) return
def test008_flush_already_flushed(self): self.scan.status = IrmaScanStatus.flushed module.flush(self.scan, self.session) self.session.delete.assert_not_called()
def test_flush_already_flushed(self): self.scan.status = IrmaScanStatus.flushed module.flush(self.scan, self.session) self.session.delete.assert_not_called()