def test_get_file_not_found(self, app_manager): st = mm.by_name() file_url = url_for('mm.get_file', mm='media', filename='not.found') response = app_manager.test_client().get(file_url) assert response.status_code == 404
def test_get_file_no_storage(self, app_manager): st = mm.by_name() file_url = url_for('mm.get_file', mm='fake', filename='file.test') response = app_manager.test_client().get(file_url) assert response.status_code == 404
def test_save_from_filestorage_with_filename(self, app_manager, utils): st = mm.by_name() f = utils.filestorage('test.png', 'test') filename = st.save(f, st.generate_name(f)) assert st.exists(filename) st.delete(filename)
def test_basic_postprocess_watermark(self, app_manager, image, utils): st = mm.by_name() with open(image, 'rb') as fp: f = utils.filestorage('horse.jpg', fp) filename = st.save(f) assert st.exists(filename) st.delete(filename)
def test_save_from_file(self, app_manager, image, utils): st = mm.by_name() f = utils.file(Image.open(image)) filename = st.save(f, 'test.jpg') assert st.exists(filename) st.delete(filename)
def test_save_from_filestorage(self, app_manager, image, utils): st = mm.by_name() with open(image, 'rb') as fp: f = utils.filestorage('flask.jpg', fp) filename = st.save(f) assert st.exists(filename) st.delete(filename)
def test_save_from_file(self, app_manager, utils): st = mm.by_name() f = utils.file(b'test') filename = st.save(f, 'test.png') assert st.exists(filename) st.delete(filename)
def test_basic_watermark(self, app_manager, image, utils): st = mm.by_name() with open(image, 'rb') as fp: f = utils.filestorage('flask.jpg', fp) filename = st.save(f, postprocess=POSTPROCESS_PARAMS) assert st.exists(filename) st.delete(filename)
def test_get_file(self, app_manager): st = mm.by_name() file_url = url_for('mm.get_file', mm='media', filename='file.test') response = app_manager.test_client().get(file_url, follow_redirects=True) print("respinse data:", response.data) #assert response.status_code == 200 assert response.data == ''.encode('utf-8')
def test_save_from_filestorage_original_name(self, app_manager, utils): st = mm.by_name() f = utils.filestorage('test.png', 'test') filename = st.save(f) assert filename == 'test.png' assert st.exists(filename) st.delete(filename)
def test_save_thumbnail(self, app_manager, image, utils): st = mm.by_name() f = utils.file(Image.open(image)) filename = st.save(f, 'test.jpg') assert st.exists(filename) assert st.exists(st.generate_thumbnail_name(filename)) st.delete(filename)
def test_save_from_filestorage_with_filename_original_name( self, app_manager, image, utils): st = mm.by_name() with open(image, 'rb') as fp: f = utils.filestorage('flask.jpg', fp) filename = st.save(f, 'cicakutya.jpg', generate_name=False) assert filename == 'cicakutya.jpg' assert st.exists(filename) st.delete(filename)
def test_size_thumbnail(self, app_manager, image, utils): st = mm.by_name() f = utils.file(Image.open(image)) filename = st.save(f, 'test.jpg') assert st.exists(st.generate_thumbnail_name(filename)) thumb = Image.open( os.path.join('tests', 'test', st.generate_thumbnail_name(filename))) assert thumb.size[0] == THUMB_WIDTH assert thumb.size[1] == THUMB_HEIGHT st.delete(filename)
def test_compress_files_filestorage(self, app_manager, utils): st = mm.by_name() f1 = utils.filestorage('test.png', 'test') f2 = utils.filestorage('test.png', 'test') filename1 = st.save(f1, st.generate_name(f1)) filename2 = st.save(f2, st.generate_name(f2)) archive = st.archive_files(st.generate_name('compressed.zip'), [filename1, filename2]) assert st.exists(archive) st.delete(filename1) st.delete(filename2) st.delete(archive)
def test_compress_files_filestorage_read(self, app_manager, utils): # Create archive st = mm.by_name() f1 = utils.filestorage('test.png', 'test') f2 = utils.filestorage('test.png', 'test') filename1 = st.save(f1, st.generate_name(f1)) filename2 = st.save(f2, st.generate_name(f2)) archive = st.archive_files(st.generate_name('compressed.zip'), [filename1, filename2]) assert st.exists(archive) # Read back file with st.open(archive, mode='rb') as f: byte_stream = io.BytesIO(f.read()) byte_stream.seek(0) st.delete(filename1) st.delete(filename2) st.delete(archive)
def test_by_name_default(self, app, init_mm): init_mm.init_app(app) assert init_mm.by_name() == mm.by_name( ) == app.extensions['mediamanager']['media']
def test_path(self, app_manager): st = mm.by_name() assert st.path('file.test') == os.path.abspath( os.path.join(os.path.dirname(__file__), 'test', 'file.test'))
def test_write_not_overwrite(self, app_manager): st = mm.by_name() with pytest.raises(FileExistsError): st.write('file.test', '')
def test_write_overwrite(self, app_manager): st = mm.by_name() st.write('file.test', '', overwrite=True)
def test_read_not_found(self, app_manager): st = mm.by_name() with pytest.raises(FileNotFoundError): st.read('file.not.found')
def test_read(self, app_manager): st = mm.by_name() assert st.read('file.test') == b''
def test_open_read_not_found(self, app_manager): st = mm.by_name() with pytest.raises(FileNotFoundError): with st.open('file.not.found') as f: pass
def test_file_open_as_binary(self, app_manager): st = mm.by_name() with st.open('file.test', mode='rb') as f: byte_stream = io.BytesIO(f.read()) byte_stream.seek(0) assert byte_stream.read() == b''
def test_file_open(self, app_manager): st = mm.by_name() with st.open('file.test') as f: assert f.read() == ""
def test_exists(self, app_manager): st = mm.by_name() assert st.exists('file.test') assert not st.exists('notexists')
def test_open_write_new_file(self, app_manager): st = mm.by_name() with st.open('file.test', 'w') as f: f.write('')
def test_root(self, app_manager): st = mm.by_name() assert st.storage.root == os.path.abspath( os.path.join(os.path.dirname(__file__), 'test'))
def test_url(self, app_manager): st = mm.by_name() expected_url = url_for('mm.get_file', mm=st.name, filename='test.txt') assert st.url('test.txt') == expected_url