def job_success(result, jobid): try: (frontend_scanid, filename, probe) = job_ctrl.info(jobid) log.info("scanid %s jobid:%d probe %s", frontend_scanid, jobid, probe) celery_frontend.scan_result(frontend_scanid, filename, probe, result) job_ctrl.success(jobid) except: log.info("exception", exc_info=True) return
def test_scan_progress(self): scan_id = scan_ctrl.new(self.scanid, self.userid, 10) job_ids = [] for i in xrange(0, 10): for probe in ['probe1', 'probe2']: job_ids.append(job_ctrl.new(scan_id, "file-{0}".format(i), probe, 1)) scan_ctrl.launched(scan_id) shuffle(job_ids) for i, job_id in enumerate(job_ids[:-1]): job_ctrl.success(job_id) (status, progress_details) = scan_ctrl.progress(scan_id) self.assertEqual(status, IrmaScanStatus.label[IrmaScanStatus.launched]) self.assertIsNotNone(progress_details) self.assertEqual(progress_details['successful'], i + 1)
def test_scan_job_success(self): scan_id = scan_ctrl.new(self.scanid, self.userid, 10) job_ids = [] for i in xrange(0, 10): for probe in ['probe1', 'probe2']: job_ids.append(job_ctrl.new(scan_id, "file-{0}".format(i), probe, 1)) scan_ctrl.launched(scan_id) shuffle(job_ids) for job_id in job_ids: job_ctrl.success(job_id) self.assertTrue(scan_ctrl.check_finished(scan_id)) with session_query() as session: scan = Scan.load(scan_id, session) self.assertEqual(scan.status, IrmaScanStatus.processed) self.assertEqual(scan.nb_files, 10) self.assertEqual(scan.nb_jobs, 20)