def test_get_user_from_confirmation_token_bad_or_wrong_token(self): """get_user_from... should raise exception with bad/wrong token.""" s = Serializer(current_app.config['SECRET_KEY']) wrongtoken = s.dumps({'whee': 'beep!'}) with pytest.raises(KeyError): get_user_from_confirmation_token(wrongtoken) with pytest.raises(ValueError): get_user_from_confirmation_token('badtoken')
def test_get_user_from_confirmation_token(self, db): """get_user_from... raises exeption if user already confirmed.""" user = make_dummy_user() user.id = 42 user.confirmed = True db.session.add(user) db.session.commit() token = user.generate_account_confirmation_token() with pytest.raises(ValueError): get_user_from_confirmation_token(token)
def test_get_user_from_confirmation_token_with_valid_token(self, db): """get_user_from_confirmation_token should work given a valid token.""" user1 = make_dummy_user() db.session.add(user1) db.session.commit() token = user1.generate_account_confirmation_token() user2 = get_user_from_confirmation_token(token) assert user1.id == user2.id