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