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())