def setUp(self):
if not Site.__table__.exists():
Site.__table__.create()
site = Site(id=settings.SITE_ID,
domain='example.com',
name=u'example.com')
self.session.add(site)
self.session.commit()
def setUpClass(cls):
super(RegistrationModelTests, cls).setUpClass()
Site.__table__.create()
User.__table__.create()
orm = ORM.get()
cls.session = orm.sessionmaker()
cls.user_info['session'] = cls.session
site = Site(id=settings.SITE_ID,
domain='example.com',
name=u'example.com')
cls.session.add(site)
cls.session.commit()
