mirror of
https://github.com/jointakahe/takahe.git
synced 2024-12-01 19:51:00 +00:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
import pytest
|
||
|
|
from asgiref.sync import async_to_sync
|
||
|
|
from django.test.client import RequestFactory
|
||
|
|
from pytest_httpx import HTTPXMock
|
||
|
|
|
||
|
|
from core.signatures import HttpSignature
|
||
|
|
from users.models import SystemActor
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_system_actor_signed(config_system, httpx_mock: HTTPXMock):
|
||
|
|
"""
|
||
|
|
Tests that the system actor signs requests properly
|
||
|
|
"""
|
||
|
|
system_actor = SystemActor()
|
||
|
|
system_actor.generate_keys()
|
||
|
|
# Send a fake outbound request
|
||
|
|
httpx_mock.add_response()
|
||
|
|
async_to_sync(system_actor.signed_request)(
|
||
|
|
method="get",
|
||
|
|
uri="http://example.com/test-actor",
|
||
|
|
)
|
||
|
|
# Retrieve it and construct a fake request object
|
||
|
|
outbound_request = httpx_mock.get_request()
|
||
|
|
fake_request = RequestFactory().get(
|
||
|
|
path="/test-actor",
|
||
|
|
HTTP_HOST="example.com",
|
||
|
|
HTTP_DATE=outbound_request.headers["date"],
|
||
|
|
HTTP_SIGNATURE=outbound_request.headers["signature"],
|
||
|
|
HTTP_ACCEPT=outbound_request.headers["accept"],
|
||
|
|
)
|
||
|
|
# Verify that
|
||
|
|
HttpSignature.verify_request(fake_request, system_actor.public_key)
|