ariadne/tests/test_utils.py

38 lines
1.1 KiB
Python

from __future__ import annotations
import re
from ariadne.services.mailu import MailuService
from ariadne.utils.http import extract_bearer_token
class DummyRequest:
def __init__(self, headers):
self.headers = headers
from ariadne.utils.errors import safe_error_detail
from ariadne.utils.passwords import random_password
def test_random_password_length() -> None:
password = random_password(24)
assert len(password) == 24
assert re.match(r"^[A-Za-z0-9]+$", password)
def test_mailu_resolve_email_attribute() -> None:
attrs = {"mailu_email": ["custom@bstein.dev"]}
assert MailuService.resolve_mailu_email("alice", attrs) == "custom@bstein.dev"
def test_mailu_resolve_email_default() -> None:
assert MailuService.resolve_mailu_email("alice", {}) == "alice@bstein.dev"
def test_safe_error_detail_runtime() -> None:
assert safe_error_detail(RuntimeError("boom"), "fallback") == "boom"
def test_extract_bearer_token() -> None:
request = DummyRequest({"Authorization": "Bearer token123"})
assert extract_bearer_token(request) == "token123"