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)