def test_upload_files(self, m_IrmaSFTPv2, m_isdir):
parent_filename = "parent_file"
filelist = ["file1", "file2"]
m_ftp = MagicMock()
m_IrmaSFTPv2().__enter__.return_value = m_ftp
m_isdir.return_value = False
module.upload_files("frontend", "path", filelist, parent_filename)
m_isdir.assert_has_calls([call('path/file1'),
call('path/file2')])
m_ftp.upload_file.assert_has_calls([call('parent_file_0',
'path/file1'),
call('parent_file_1',
'path/file2')])
def handle_output_files(results, frontend, scanid):
# First check if there is some output files
output_files = results.pop('output_files', None)
if output_files is None:
return
tmpdir = output_files.get('output_dir', None)
file_list = output_files.get('file_list', None)
if tmpdir is None or file_list is None:
return
uploaded_files = ftp_ctrl.upload_files(frontend, tmpdir, file_list, scanid)
log.debug("handle_output_files: uploaded %s", ",".join(file_list))
results['uploaded_files'] = uploaded_files
shutil.rmtree(tmpdir)
return
def handle_output_files(results, frontend, scanid):
# First check if there is some output files
output_files = results.pop('output_files', None)
if output_files is None:
return
tmpdir = output_files.get('output_dir', None)
file_list = output_files.get('file_list', None)
if tmpdir is None or file_list is None:
return
uploaded_files = ftp_ctrl.upload_files(frontend, tmpdir, file_list, scanid)
log.debug("handle_output_files: uploaded %s", ",".join(file_list))
results['uploaded_files'] = uploaded_files
shutil.rmtree(tmpdir)
return
def test_upload_files_not_a_file(self, m_IrmaSFTPv2, m_isdir):
m_isdir.return_value = True
m_ftp = MagicMock()
m_IrmaSFTPv2().__enter__.return_value = m_ftp
module.upload_files("frontend", "path", ["dir"], "parent_file")
m_ftp.upload_file.assert_not_called()