def setUp(self):
from fixture import SQLAlchemyFixture, NamedDataStyle
import sqlalchemy
from sqlalchemy.orm import (
mapper, relation, clear_mappers, sessionmaker, scoped_session)
from sqlalchemy import create_engine
metadata.bind = create_engine(conf.LITE_DSN)
metadata.create_all()
class options:
dsn = conf.LITE_DSN
env = ['fixture.examples.db.sqlalchemy_examples']
self.options = options
self.generator = DataSetGenerator(self.options, template=StubTemplate())
ScopedSession = scoped_session(sessionmaker(autoflush=False, transactional=True))
ScopedSession.mapper(Category, categories, save_on_init=False)
ScopedSession.mapper(Product, products, properties={
'category': relation(Category),
}, save_on_init=False)
ScopedSession.mapper(Offer, offers, properties={
'category': relation(Category, backref='products'),
'product': relation(Product)
}, save_on_init=False)
self.fixture = SQLAlchemyFixture(
env=sqlalchemy_examples,
style=NamedDataStyle(),
engine=metadata.bind)
self.data = self.fixture.data(self.CategoryData)
self.data.setup()
self.hnd = self.generator.get_handler(
"%s.Category" % (Category.__module__),
obj=Category,
connection=metadata.bind)
self.hnd.begin()
def setUp(self):
class options:
dsn = conf.LITE_DSN
env = ['fixture.examples.db.sqlalchemy_examples']
self.generator = DataSetGenerator(options, template=StubTemplate())
