def __create_attachment_wrapper(self, **kw): att_init_data = dict(content='dummy_attachment_test', file_name='test_file.txt', description='A dummy attachment.') for attr_name, value in kw.iteritems(): att_init_data[attr_name] = value return AttachmentWrapper(**att_init_data)
def test_get_base64_data_for_upload(self): # Test string test_str = 'This is a string for base64 conversion testing.' self.init_data['content'] = test_str exp_conv = Base64Converter.encode_string(test_str) att = AttachmentWrapper(**self.init_data) self.assert_equal(att.get_base64_data_for_upload(), exp_conv) # Test stream test_stream = StringIO('This is a stream for base64 conversion testing.') exp_conv = Base64Converter.encode_stream(test_stream) self.init_data['content'] = test_stream att = AttachmentWrapper(**self.init_data) self.assert_equal(att.get_base64_data_for_upload(), exp_conv) # Test file map file_map = dict(file1='test stream 1', file2='test stream 2') zip_stream = StringIO() archive = zipfile.ZipFile(zip_stream, 'a', zipfile.ZIP_DEFLATED, False) for fn, content in file_map.iteritems(): archive.writestr(fn, content) for zfile in archive.filelist: zfile.create_system = 0 archive.close() zip_stream.seek(0) exp_conv = Base64Converter.encode_zip_stream(zip_stream) self.init_data['content'] = file_map att = AttachmentWrapper(**self.init_data) self.assert_equal(att.get_base64_data_for_upload(), exp_conv) # Test error raising self.init_data['content'] = 1 att = AttachmentWrapper(**self.init_data) self.assert_raises(TypeError, att.get_base64_data_for_upload)
def test_create_from_trac_data(self): file_name = 'test_file1.txt' description = 'A test file.' size = len(file_name) time = datetime author = 'user1' trac_data = (file_name, description, size, time, author) att = AttachmentWrapper.create_from_trac_data(trac_data) self.init_data['content'] = None self.init_data['time'] = time for attr_name, exp_value in self.init_data.iteritems(): self.assert_equal(getattr(att, attr_name), exp_value)
def test_init(self): att = AttachmentWrapper(**self.init_data) for attr_name, exp_value in self.init_data.iteritems(): self.assert_equal(getattr(att, attr_name), exp_value)
def test_get_base64_data_for_upload(self): # Test string test_str = 'This is a string for base64 conversion testing.' self.init_data['content'] = test_str exp_conv = Base64Converter.encode_string(test_str) att = AttachmentWrapper(**self.init_data) self.assert_equal(att.get_base64_data_for_upload(), exp_conv) # Test stream test_stream = StringIO( 'This is a stream for base64 conversion testing.') exp_conv = Base64Converter.encode_stream(test_stream) self.init_data['content'] = test_stream att = AttachmentWrapper(**self.init_data) self.assert_equal(att.get_base64_data_for_upload(), exp_conv) # Test file map file_map = dict(file1='test stream 1', file2='test stream 2') zip_stream = StringIO() archive = zipfile.ZipFile(zip_stream, 'a', zipfile.ZIP_DEFLATED, False) for fn, content in file_map.iteritems(): archive.writestr(fn, content) for zfile in archive.filelist: zfile.create_system = 0 archive.close() zip_stream.seek(0) exp_conv = Base64Converter.encode_zip_stream(zip_stream) self.init_data['content'] = file_map att = AttachmentWrapper(**self.init_data) self.assert_equal(att.get_base64_data_for_upload(), exp_conv) # Test error raising self.init_data['content'] = 1 att = AttachmentWrapper(**self.init_data) self.assert_raises(TypeError, att.get_base64_data_for_upload)