def install(app, *args):
engine = m.create_tables(app)
db = SQLAlchemyFixture(env=m, style=NamedDataStyle(), engine=engine)
data = db.data(*args)
data.setup()
db.dispose()
return data
def load_fixtures(self, *args):
# TODO: Check https://github.com/mitsuhiko/flask-sqlalchemy/pull/89
# and update version of Flask-Sqlalchemy
from api import models
from fixture import SQLAlchemyFixture
from fixture.style import NamedDataStyle
db = SQLAlchemyFixture(env=models,
style=NamedDataStyle(),
engine=self.engine)
data = db.data(*args)
data.setup()
db.dispose()
def test_fixture_can_be_disposed():
if sa_major < 0.5:
from sqlalchemy.exceptions import InvalidRequestError
else:
from sqlalchemy.exc import InvalidRequestError
engine = create_engine(conf.LITE_DSN)
metadata.bind = engine
metadata.create_all()
Session = get_transactional_session()
session = Session()
fixture = SQLAlchemyFixture(
env={'CategoryData':Category},
engine=metadata.bind
)
class CategoryData(DataSet):
class cars:
name = 'cars'
class free_stuff:
name = 'get free stuff'
clear_mappers()
mapper(Category, categories)
data = fixture.data(CategoryData)
data.setup()
data.teardown()
fixture.dispose()
# cannot use fixture anymore :
try:
data.setup()
except InvalidRequestError:
pass
else:
assert False, "data.setup() did not raise InvalidRequestError after connection was disposed"
# a new instance of everything is needed :
metadata.create_all()
fixture = SQLAlchemyFixture(
env={'CategoryData':Category},
engine=metadata.bind
)
data = fixture.data(CategoryData)
data.setup()
data.teardown()
def test_fixture_can_be_disposed():
from sqlalchemy.exceptions import InvalidRequestError
engine = create_engine(conf.LITE_DSN)
metadata.bind = engine
metadata.create_all()
Session = get_transactional_session()
session = Session()
fixture = SQLAlchemyFixture(
env={'CategoryData':Category},
engine=metadata.bind
)
class CategoryData(DataSet):
class cars:
name = 'cars'
class free_stuff:
name = 'get free stuff'
clear_mappers()
mapper(Category, categories)
data = fixture.data(CategoryData)
data.setup()
data.teardown()
fixture.dispose()
# cannot use fixture anymore :
try:
data.setup()
except InvalidRequestError:
pass
else:
assert False, "data.setup() did not raise InvalidRequestError after connection was disposed"
# a new instance of everything is needed :
metadata.create_all()
fixture = SQLAlchemyFixture(
env={'CategoryData':Category},
engine=metadata.bind
)
data = fixture.data(CategoryData)
data.setup()
data.teardown()
def install(app, *args):
dbfixture = SQLAlchemyFixture(env=models, style=NamedDataStyle(),
engine=db.engine)
data = dbfixture.data(*args)
data.setup()
dbfixture.dispose()
def install(app, *args):
engine = create_engine(app.config['SQLALCHEMY_DATABASE_URI'])
db = SQLAlchemyFixture(env=models, style=NamedDataStyle(), engine=engine)
data = db.data(*args)
data.setup()
db.dispose()
def install(app, *args):
engine = create_engine(app.config['SQLALCHEMY_DATABASE_URI'])
db = SQLAlchemyFixture(env=models, style=NamedDataStyle(), engine=engine)
data = db.data(*args)
data.setup()
db.dispose()
