def test_user_token(self):
user = models.User(name="tester", email="*****@*****.**")
user.put()
user2 = models.User(name="tester2", email="*****@*****.**")
user2.put()
token = models.User.create_signup_token(user.get_id())
self.assertTrue(models.User.validate_signup_token(user.get_id(), token))
self.assertFalse(models.User.validate_resend_token(user.get_id(), token))
self.assertFalse(models.User.validate_signup_token(user2.get_id(), token))
def save_user_to_database(**kwargs):
"""
Create new user record in database from validated data.
:param kwargs:
:return:
"""
user = m.User(**kwargs)
m.db.session.add(user)
m.db.session.commit()
return user
def user(self, **kwargs):
user = m.User(id=kwargs.get('id', fake.id()))
user.username = kwargs.get('username', fake.name())
user.email = kwargs.get('email', fake.email())
user.role = kwargs.get('role', fake.user_role())
user.password = kwargs.get('password', fake.text(10))
m.db.session.add(user)
m.db.session.flush()
return user
def save_user_to_database(**kwargs):
user = models.User(**kwargs)
models.db.session.add(user)
return user
