def test034_launch_asynchronous_ftp_error(self,
m_session_transaction,
m_Scan,
m_ftp_ctrl,
m_add_empty_result):
m_scan, m_session = MagicMock(), MagicMock()
fw1, fw2 = MagicMock(), MagicMock()
file1, file2 = MagicMock(), MagicMock()
pathf1, pathf2 = 'path-file1', 'path-file2'
file1.path = pathf1
file2.path = pathf2
fw1.file.sha256 = "sha256file1"
fw1.file.mimetype = "mimetypefile1"
fw2.file.sha256 = "sha256file2"
fw2.file.mimetype = "mimetypefile2"
m_scan.files_web = [fw1, fw2]
m_scan.files = [file1, file2]
probe1, probe2 = "Probe1", "Probe2"
probelist = [probe1, probe2]
m_scan.get_probe_list.return_value = probelist
m_add_empty_result.return_value = probelist
m_session_transaction().__enter__.return_value = m_session
m_scan.status = IrmaScanStatus.ready
m_scan.mimetype_filtering = False
m_Scan.load_from_ext_id.return_value = m_scan
scanid = "scanid"
m_ftp_ctrl.upload_scan.side_effect = IrmaFtpError()
module.launch_asynchronous(scanid)
expected = IrmaScanStatus.error_ftp_upload
m_scan.set_status.assert_called_once_with(expected)
def test034_launch_asynchronous_ftp_error(self,
m_session_transaction,
m_Scan,
m_ftp_ctrl,
m_add_empty_result):
m_scan, m_session = MagicMock(), MagicMock()
fw1, fw2 = MagicMock(), MagicMock()
file1, file2 = MagicMock(), MagicMock()
pathf1, pathf2 = 'path-file1', 'path-file2'
file1.path = pathf1
file2.path = pathf2
fw1.file.sha256 = "sha256file1"
fw1.file.mimetype = "mimetypefile1"
fw2.file.sha256 = "sha256file2"
fw2.file.mimetype = "mimetypefile2"
m_scan.files_web = [fw1, fw2]
m_scan.files = [file1, file2]
probe1, probe2 = "Probe1", "Probe2"
probelist = [probe1, probe2]
m_scan.get_probe_list.return_value = probelist
m_add_empty_result.return_value = probelist
m_session_transaction().__enter__.return_value = m_session
m_scan.status = IrmaScanStatus.ready
m_scan.mimetype_filtering = False
m_Scan.load_from_ext_id.return_value = m_scan
scanid = "scanid"
m_ftp_ctrl.upload_scan.side_effect = IrmaFtpError()
module.launch_asynchronous(scanid)
expected = IrmaScanStatus.error_ftp_upload
m_scan.set_status.assert_called_once_with(expected)
def scan_launch(scanid):
try:
log.debug("scanid: %s", scanid)
scan_ctrl.launch_asynchronous(scanid)
except IrmaDatabaseError as e:
log.exception(e)
raise scan_launch.retry(countdown=2, max_retries=3, exc=e)
def scan_launch(scanid):
try:
log.debug("scanid: %s", scanid)
scan_ctrl.launch_asynchronous(scanid)
except IrmaDatabaseError as e:
log.exception(e)
raise scan_launch.retry(countdown=2, max_retries=3, exc=e)
def test032_launch_asynchronous_nothing_to_do(self, m_session_transaction,
m_Scan):
m_session, m_scan = MagicMock(), MagicMock()
m_session_transaction().__enter__.return_value = m_session
m_scan.status = IrmaScanStatus.ready
m_Scan.load_from_ext_id.return_value = m_scan
module.launch_asynchronous("whatever")
m_scan.set_status.assert_called_once_with(IrmaScanStatus.finished)
def test032_launch_asynchronous_nothing_to_do(self,
m_session_transaction,
m_Scan):
m_session, m_scan = MagicMock(), MagicMock()
m_session_transaction().__enter__.return_value = m_session
m_scan.status = IrmaScanStatus.ready
m_Scan.load_from_ext_id.return_value = m_scan
module.launch_asynchronous("whatever")
m_scan.set_status.assert_called_once_with(IrmaScanStatus.finished)
def scan_launch(scanid):
try:
scan_ctrl.launch_asynchronous(scanid)
except IrmaDatabaseError as e:
print "Exception has occurred:{0}".format(e)
raise scan_launch.retry(countdown=2, max_retries=3, exc=e)