def test09_FtpUpload(self):
e = FtpExporter()
e.host = 'localhost'
e.port = TEST_PORT
e.username = '******'
e.password = '******'
# copy some files to export directory
export_folder = e.exportDirectory()
try:
os.makedirs(export_folder)
except:
self.assertTrue(False, 'could not create export directory')
out_file = os.path.join(export_folder, 'index.html')
with open(out_file, 'w') as i:
i.write('test')
e.postProcess(out_file)
expected_index_file = os.path.join(FTP_USER_FOLDER, 'public_html',
'index.html')
self.assertTrue(os.path.exists(expected_index_file))
content = open(expected_index_file, 'r').readlines()
self.assertEqual(content, ['test'])
# try overwriting existing file
with open(out_file, 'w') as i:
i.write('test2')
e.postProcess(out_file)
self.assertTrue(expected_index_file)
content = open(expected_index_file, 'r').readlines()
self.assertEqual(content, ['test2'])
def test08_FtpExporterTempFolder(self):
"""Test FTP exporter generation of temp folder"""
e = FtpExporter()
e.host = None
self.assertTrue(e.exportDirectory())
prev_folder = e.exportDirectory()
e.postProcess('')
# a new export folder should be generated to avoid outdated files
self.assertNotEqual(e.exportDirectory(), prev_folder)
def test10_FtpUploadSubfolder(self):
e = FtpExporter()
e.host = 'localhost'
e.port = TEST_PORT
e.username = '******'
e.password = '******'
# copy some files to export directory
export_folder = e.exportDirectory()
try:
os.makedirs(export_folder)
except:
self.assertTrue(False, 'could not create export directory')
out_file = os.path.join(export_folder, 'index.html')
with open(out_file, 'w') as i:
i.write('test')
sub_folder = os.path.join(export_folder, 'sub')
try:
os.makedirs(sub_folder)
except:
self.assertTrue(False, 'could not create export directory')
sub_folder_out_file = os.path.join(sub_folder, 'index.html')
with open(sub_folder_out_file, 'w') as i:
i.write('test2')
result = WriterResult()
result.index_file = out_file
result.folder = export_folder
e.postProcess(result)
expected_index_file = os.path.join(FTP_USER_FOLDER, 'public_html',
'index.html')
self.assertTrue(os.path.exists(expected_index_file))
content = open(expected_index_file, 'r').readlines()
self.assertEqual(content, ['test'])
expected_sub_folder_index_file = os.path.join(FTP_USER_FOLDER,
'public_html', 'sub',
'index.html')
self.assertTrue(os.path.exists(expected_sub_folder_index_file))
content = open(expected_sub_folder_index_file, 'r').readlines()
self.assertEqual(content, ['test2'])