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'