def populate(): BaseModel.register_sessionmaker(session, force=True) metadata.create_all(engine) status = [Status(u'ok'), Status(u'ko')] session.add_all(status) ok = status[0] for i in xrange(10): tr = Treatment(u'test {}'.format(i), ok) session.add(tr) ko = status[1] for i in xrange(8): tr = Treatment(u'test_ko {}'.format(i), ko) session.add(tr) session.commit()
def test_force_sessionmaker(): BaseModel.register_sessionmaker(u'test') BaseModel.register_sessionmaker(u'plop', force=True)
def test_register_session_maker_already_registered(): BaseModel.register_sessionmaker(u'test') BaseModel.register_sessionmaker(u'plop')
def test_register_function_session(): def build_session(): return u'built_session' BaseModel.register_sessionmaker(build_session) assert BaseModel.session == u'built_session'
def test_register_simple_session(): Session = u'session' BaseModel.register_sessionmaker(Session) assert BaseModel.session == Session