def test_delete(self, db_session): file_id = DBFileStorage().create('content here', u'f.jpg', 'image/jpg') fs = DBFileStorage().get(file_id) db_session.add(fs) db_session.flush() assert db_session.query(DBStoredFile.file_id).one()[0] == file_id DBFileStorage().delete(file_id) assert db_session.query(DBStoredFile).count() == 0
def test_replace(self, db_session): file_id = self.make_one() DBFileStorage().replace(file_id, 'second content', u'f2.jpg', 'doc') fs = DBFileStorage().get(file_id) assert fs.filename == u'f2.jpg' assert fs.content_type == 'doc' assert fs.read() == 'second content' DBFileStorage().replace(fs, 'third content', u'f3.jpg', 'xls') assert fs.filename == u'f3.jpg' assert fs.content_type == 'xls' assert fs.read() == 'third content'
def make_one(self, content='content here', filename=u'f.jpg', content_type='image/jpg'): file_id = DBFileStorage().create( content=content, filename=filename, content_type=content_type) return file_id
def make_one( self, content=b"content here", filename="f.jpg", content_type="image/jpg" ): file_id = DBFileStorage().create( content=content, filename=filename, content_type=content_type ) return file_id
def test_replace(self, db_session): file_id = self.make_one() DBFileStorage().replace(file_id, 'second content', u'f2.jpg', 'doc') fs = DBFileStorage().get(file_id) assert fs.filename == u'f2.jpg' assert fs.content_type == 'doc' assert fs.read() == 'second content' DBFileStorage().replace(fs, 'third content', u'f3.jpg', 'xls') assert fs.filename == u'f3.jpg' assert fs.content_type == 'xls' assert fs.read() == 'third content'
def test_replace(self, db_session): file_id = self.make_one() DBFileStorage().replace(file_id, b"second content", "f2.jpg", "doc") fs = DBFileStorage().get(file_id) assert fs.filename == "f2.jpg" assert fs.content_type == "doc" assert fs.read() == b"second content" DBFileStorage().replace(fs, b"third content", "f3.jpg", "xls") assert fs.filename == "f3.jpg" assert fs.content_type == "xls" assert fs.read() == b"third content"
def test_session_integration(self, db_session): from depot.manager import DepotManager DepotManager._default_depot = 'default' DepotManager._depots = {'default': DBFileStorage()} file_id = DepotManager.get().create('content here', u'f.jpg', 'image/jpg') fs = DepotManager.get().get(file_id) db_session.add(fs) import transaction transaction.commit() transaction.begin() db_session.delete(fs) transaction.commit() with pytest.raises(IOError): DepotManager.get().get(file_id)
def test_get(self, db_session): with pytest.raises(IOError): DBFileStorage().get("1") file_id = self.make_one() assert DBFileStorage().get(file_id).data == "content here"
def test_exists(self, db_session): assert DBFileStorage().exists("1") is False file_id = self.make_one() assert DBFileStorage().exists(file_id) is True
def test_list(self): with pytest.raises(NotImplementedError): DBFileStorage().list()