flaskbb/tests/unit/tokens/test_serializer.py

from datetime import datetime, timedelta

import pytest
from freezegun import freeze_time

from flaskbb import tokens
from flaskbb.core.tokens import Token, TokenActions, TokenError


def test_can_round_trip_token():
    serializer = tokens.FlaskBBTokenSerializer(
        'hello i am secret', timedelta(seconds=100)
    )
    token = Token(user_id=1, operation=TokenActions.RESET_PASSWORD)
    roundtrip = serializer.loads(serializer.dumps(token))

    assert token == roundtrip


def test_raises_token_error_with_bad_data():
    serializer = tokens.FlaskBBTokenSerializer(
        'hello i am also secret', timedelta(seconds=100)
    )

    with pytest.raises(TokenError) as excinfo:
        serializer.loads('not actually a token')
    assert 'invalid' in str(excinfo.value)


def test_expired_token_raises():
    serializer = tokens.FlaskBBTokenSerializer(
        'i am a secret not', expiry=timedelta(seconds=1)
    )
    dumped_token = serializer.dumps(
        Token(user_id=1, operation=TokenActions.RESET_PASSWORD)
    )

    with freeze_time(datetime.now() + timedelta(days=10)):
        with pytest.raises(TokenError) as excinfo:
            serializer.loads(dumped_token)

    assert 'expired' in str(excinfo.value)