def setUp(self):
self.session = Session()
self.session.query(User).delete()
self.email = '*****@*****.**'
reg_controller = RegistrationController(self.session)
self.password = '******'
token = reg_controller.register(self.email, self.password)
reg_controller.confirm(self.email, token)
self.controller = AuthenticationController(self.session)
class RegistrationTest(unittest.TestCase):
def setUp(self):
self.session = Session()
self.session.query(User).delete()
self.email = '*****@*****.**'
self.controller = RegistrationController(self.session)
self.token = self.controller.register(self.email, 'qwerty')
def test_registration(self):
self.assertEqual(self.session.query(User).filter(and_(User.email == self.email, User.confirmed_at == None)).count(), 1)
def test_confirmation(self):
self.controller.confirm(self.email, self.token)
self.assertEqual(self.session.query(User).filter(and_(User.email == self.email, User.confirmed_at != None)).count(), 1)
def setUp(self):
self.session = Session()
self.session.query(User).delete()
self.email = '*****@*****.**'
self.controller = RegistrationController(self.session)
self.token = self.controller.register(self.email, 'qwerty')
