def test_user_password(self):
user_role = Role.query.filter_by(name='User').first()
user = User(email=forgery_py.internet.email_address(),
username=forgery_py.internet.user_name(),
password='******',
avatar_hash=None,
role=user_role,
confirmed=True)
db.session.add(user)
db.session.commit()
with self.assertRaises(AttributeError) as err:
user.password()
self.assertTrue(user.verify_password('old_password'.encode('utf-8')))
user.password = '******'
self.assertFalse(user.verify_password('old_password'.encode('utf-8')))
self.assertTrue(user.verify_password('new_password'.encode('utf-8')))
def test_user_invite(self):
user_role = Role.query.filter_by(name='User').first()
User.generate_fake(count=1)
user1 = User.query.first()
user2 = User(email=forgery_py.internet.email_address(), role=user_role)
db.session.add(user2)
db.session.commit()
valid_token = user2.generate_invite_token(expiration=3600)
invalid_token = user2.generate_confirmation_token(expiration=3600)
valid_invite = user2.accept_invite(token=valid_token,
username='******',
new_password='******')
invalid_invite = user2.accept_invite(invalid_token, 'username2',
'new_password')
invalid_invite2 = user2.accept_invite('', 'username2', 'new_password')
self.assertTrue(valid_invite)
self.assertTrue(user2.username, 'username2')
self.assertTrue(user2.verify_password('new_password'))
self.assertFalse(invalid_invite)
self.assertFalse(invalid_invite2)