Example #1
0
 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)
Example #2
0
 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)
Example #3
0
 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)
Example #4
0
 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)
Example #5
0
 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)
Example #6
0
 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)