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_count(): assert Status.count() == 2 assert Status.count(name=u'ok') == 1 assert Status.count(name=u'lol') == 0
def test_load(): status = Status.load({'name': u'plop'}) assert status.name == u'plop' status = Status.load({'id': 1}) assert status.id == 1 assert status.name == u'ok'