def test_create_staff(self): st = User.create_staff(u'staff', '*****@*****.**', 'staff', session=self.session) self.assertFalse(st.is_superuser) self.assertTrue(st.is_active) self.assertTrue(st.is_staff) st = User.create_staff(u'staff2', '*****@*****.**', 'staff2', session=self.session) self.assertFalse(st.is_superuser) self.assertTrue(st.is_active) self.assertTrue(st.is_staff)
def setUpClass(cls): super(AuthViewsTestCase, cls).setUpClass() cls.session = orm.sessionmaker() User.__table__.create() cls.user = User.create_user(u'testclient', '*****@*****.**', session=cls.session) cls.user.set_password('password') cls.staff = User.create_staff(u'staff', '*****@*****.**', '123', session=cls.session) User.create_user(u'testclient2', '*****@*****.**', session=cls.session)