def test_objects_know_their_filenames(self):
# Setup
f = SimpleUploadedFile('dummy_file.txt', force_bytes('file_content'))
attachment = AttachmentFactory.create(post=self.post, file=f)
# Run & check
assert attachment.filename == 'dummy_file.txt'
attachment.file.delete()
def test_is_able_to_store_the_state_of_request_files(self):
# Setup
f1 = SimpleUploadedFile('file1.txt', force_bytes('file_content_1'))
f2 = SimpleUploadedFile('file2.txt',
force_bytes('file_content_2_long'))
f2.charset = 'iso-8859-1'
files = {'f1': f1, 'f2': f2}
real_cache = cache.get_backend()
# Run
cache.set('mykey', files)
states = real_cache.get('mykey')
# Check
assert states['f1']['name'] == 'file1.txt'
assert states['f1']['content'] == force_bytes('file_content_1')
assert states['f1']['charset'] is None
assert states['f1']['content_type'] == 'text/plain'
assert states['f1']['size'] == 14
assert states['f2']['name'] == 'file2.txt'
assert states['f2']['content'] == force_bytes('file_content_2_long')
assert states['f2']['charset'] == 'iso-8859-1'
assert states['f2']['content_type'] == 'text/plain'
assert states['f2']['size'] == 19
def test_is_able_to_regenerate_the_request_files_dict(self):
# Setup
original_f1 = SimpleUploadedFile('file1.txt',
force_bytes('file_content_1'))
original_f2 = SimpleUploadedFile(
'file2.txt', force_bytes('file_content_2_long' * 300000))
original_f2.charset = 'iso-8859-1'
original_files = {'f1': original_f1, 'f2': original_f2}
cache.set('mykey', original_files)
# Run
files = cache.get('mykey')
assert 'f1' in files
assert 'f2' in files
f1 = files['f1']
f2 = files['f2']
assert isinstance(f1, InMemoryUploadedFile)
assert f1.name == 'file1.txt'
assert f1.file.read() == force_bytes('file_content_1')
assert isinstance(
f2,
TemporaryUploadedFile) # because of the size of the content of f2
assert f2.name == 'file2.txt'
assert f2.file.read() == force_bytes('file_content_2_long' * 300000)
assert f2.charset == 'iso-8859-1'
