示例#1
0
 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)
示例#2
0
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
示例#3
0
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
示例#4
0
文件: scan_tasks.py 项目: yehias/irma
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
示例#5
0
 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)
示例#6
0
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
示例#7
0
 def test008_flush_already_flushed(self):
     self.scan.status = IrmaScanStatus.flushed
     module.flush(self.scan, self.session)
     self.session.delete.assert_not_called()
示例#8
0
 def test_flush_already_flushed(self):
     self.scan.status = IrmaScanStatus.flushed
     module.flush(self.scan, self.session)
     self.session.delete.assert_not_called()