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)