funkwhale/api/tests/channels/test_auth.py

import pytest
from rest_framework_jwt.settings import api_settings

from funkwhale_api.common.auth import TokenAuthMiddleware

jwt_payload_handler = api_settings.JWT_PAYLOAD_HANDLER
jwt_encode_handler = api_settings.JWT_ENCODE_HANDLER


@pytest.mark.parametrize("query_string", [b"token=wrong", b""])
def test_header_anonymous(query_string, factories):
    def callback(scope):
        assert scope["user"].is_anonymous

    scope = {"query_string": query_string}
    consumer = TokenAuthMiddleware(callback)
    consumer(scope)


def test_header_correct_token(factories):
    user = factories["users.User"]()
    payload = jwt_payload_handler(user)
    token = jwt_encode_handler(payload)

    def callback(scope):
        assert scope["user"] == user

    scope = {"query_string": "token={}".format(token).encode("utf-8")}
    consumer = TokenAuthMiddleware(callback)
    consumer(scope)