flaskbb/tests/fixtures/auth.py

import json

import pytest

from flaskbb.core.tokens import Token


class SimpleTokenSerializer:

    @staticmethod
    def dumps(token):
        return json.dumps({'user_id': token.user_id, 'op': token.operation})

    @staticmethod
    def loads(raw_token):
        loaded = json.loads(raw_token)
        return Token(user_id=loaded['user_id'], operation=loaded['op'])


@pytest.fixture(scope='session')
def token_serializer():
    return SimpleTokenSerializer