def test_decode_zip_file_data(self):
zip_stream = StringIO()
archive = zipfile.ZipFile(zip_stream, 'a', zipfile.ZIP_DEFLATED, False)
archive.writestr('file1', 'test stream 1')
archive.writestr('file2', 'test stream 2')
for zfile in archive.filelist:
zfile.create_system = 0
archive.close()
zip_stream.seek(0)
conv = Base64Converter.encode_zip_stream(zip_stream)
decoded_conv = Base64Converter.decode_to_stream(conv)
ret_archive = zipfile.ZipFile(decoded_conv, 'a', zipfile.ZIP_DEFLATED,
False)
content1 = None
content2 = None
self.assert_equal(len(ret_archive.namelist()), 2)
for file_name in ret_archive.namelist():
if file_name == 'file1':
content1 = ret_archive.read(file_name)
self.assert_equal(content1, 'test stream 1')
self.assert_not_equal(content2, 'test stream 2')
else:
content2 = ret_archive.read(file_name)
self.assert_equal(content2, 'test stream 2')
self.assert_not_equal(content2, 'test stream 1')
def test_decode_stream(self):
test_stream = StringIO('This is a stream for base64 conversion testing.')
conv = Base64Converter.encode_stream(test_stream)
decoded_conv = Base64Converter.decode_to_stream(conv)
decoded_cont = decoded_conv.read()
test_stream.seek(0)
exp_cont = test_stream.read()
self.assert_equal(decoded_cont, exp_cont)
def test_decode_stream(self):
test_stream = StringIO(
'This is a stream for base64 conversion testing.')
conv = Base64Converter.encode_stream(test_stream)
decoded_conv = Base64Converter.decode_to_stream(conv)
decoded_cont = decoded_conv.read()
test_stream.seek(0)
exp_cont = test_stream.read()
self.assert_equal(decoded_cont, exp_cont)
def test_decode_zip_file_data(self):
zip_stream = StringIO()
archive = zipfile.ZipFile(zip_stream, 'a', zipfile.ZIP_DEFLATED, False)
archive.writestr('file1', 'test stream 1')
archive.writestr('file2', 'test stream 2')
for zfile in archive.filelist: zfile.create_system = 0
archive.close()
zip_stream.seek(0)
conv = Base64Converter.encode_zip_stream(zip_stream)
decoded_conv = Base64Converter.decode_to_stream(conv)
ret_archive = zipfile.ZipFile(decoded_conv, 'a', zipfile.ZIP_DEFLATED,
False)
content1 = None
content2 = None
self.assert_equal(len(ret_archive.namelist()), 2)
for file_name in ret_archive.namelist():
if file_name == 'file1':
content1 = ret_archive.read(file_name)
self.assert_equal(content1, 'test stream 1')
self.assert_not_equal(content2, 'test stream 2')
else:
content2 = ret_archive.read(file_name)
self.assert_equal(content2, 'test stream 2')
self.assert_not_equal(content2, 'test stream 1')
