def test_launch_v1_error(self, m_Scan):
exception = Exception("test")
m_Scan.load_from_ext_id.side_effect = exception
scan_id = "whatever"
probes = ["probe1", "probe2"]
force = False
mimetype_filtering = False
resubmit_files = False
with self.assertRaises(Exception):
api_scans.launch_v1(scan_id=scan_id,
probes=probes,
force=force,
mimetype_filtering=mimetype_filtering,
resubmit_files=resubmit_files)
def test_launch_v1_ok(self, m_Scan, m_check_probe, m_celery_frontend):
scan_id = "whatever"
probes = ["probe1", "probe2"]
force = "False"
mimetype_filtering = "False"
resubmit_files = "False"
m_check_probe.return_value = []
m_scan = Scan("date", "ip")
m_check_probe.assert_called_once_with(None)
m_scan.force = None
m_scan.mimetype_filtering = None
m_scan.resubmit_files = None
m_scan.events = [ScanEvents(IrmaScanStatus.empty, m_scan)]
m_Scan.load_from_ext_id.return_value = m_scan
result = api_scans.launch_v1(scan_id=scan_id,
probes=probes,
force=force,
mimetype_filtering=mimetype_filtering,
resubmit_files=resubmit_files)
m_check_probe.assert_called()
m_Scan.load_from_ext_id.assert_called_once_with(scan_id, self.session)
m_celery_frontend.scan_launch.assert_called_once_with(scan_id)
self.assertIsScan(result)
self.assertEqual(result["force"], force, "force")
self.assertEqual(result["mimetype_filtering"], mimetype_filtering,
"mimetype")
self.assertEqual(result["resubmit_files"], resubmit_files, "resubmit")