forked from forks/microblog.pub
97 lines
2.5 KiB
Python
97 lines
2.5 KiB
Python
import asyncio
|
|
from contextlib import contextmanager
|
|
from typing import Any
|
|
from uuid import uuid4
|
|
|
|
import fastapi
|
|
import httpx
|
|
import respx
|
|
|
|
from app import actor
|
|
from app import httpsig
|
|
from app import models
|
|
from app.actor import LOCAL_ACTOR
|
|
from app.ap_object import RemoteObject
|
|
from app.config import session_serializer
|
|
from app.database import async_session
|
|
from app.main import app
|
|
from tests import factories
|
|
|
|
|
|
@contextmanager
|
|
def mock_httpsig_checker(ra: actor.RemoteActor):
|
|
async def httpsig_checker(
|
|
request: fastapi.Request,
|
|
) -> httpsig.HTTPSigInfo:
|
|
return httpsig.HTTPSigInfo(
|
|
has_valid_signature=True,
|
|
signed_by_ap_actor_id=ra.ap_id,
|
|
)
|
|
|
|
app.dependency_overrides[httpsig.httpsig_checker] = httpsig_checker
|
|
try:
|
|
yield
|
|
finally:
|
|
del app.dependency_overrides[httpsig.httpsig_checker]
|
|
|
|
|
|
def generate_admin_session_cookies() -> dict[str, Any]:
|
|
return {"session": session_serializer.dumps({"is_logged_in": True})}
|
|
|
|
|
|
def setup_remote_actor(respx_mock: respx.MockRouter) -> actor.RemoteActor:
|
|
ra = factories.RemoteActorFactory(
|
|
base_url="https://example.com",
|
|
username="toto",
|
|
public_key="pk",
|
|
)
|
|
respx_mock.get(ra.ap_id).mock(return_value=httpx.Response(200, json=ra.ap_actor))
|
|
return ra
|
|
|
|
|
|
def setup_remote_actor_as_follower(ra: actor.RemoteActor) -> models.Follower:
|
|
actor = factories.ActorFactory.from_remote_actor(ra)
|
|
|
|
follow_id = uuid4().hex
|
|
follow_from_inbox = RemoteObject(
|
|
factories.build_follow_activity(
|
|
from_remote_actor=ra,
|
|
for_remote_actor=LOCAL_ACTOR,
|
|
outbox_public_id=follow_id,
|
|
),
|
|
ra,
|
|
)
|
|
inbox_object = factories.InboxObjectFactory.from_remote_object(
|
|
follow_from_inbox, actor
|
|
)
|
|
|
|
follower = factories.FollowerFactory(
|
|
inbox_object_id=inbox_object.id,
|
|
actor_id=actor.id,
|
|
ap_actor_id=actor.ap_id,
|
|
)
|
|
return follower
|
|
|
|
|
|
def setup_inbox_delete(
|
|
actor: models.Actor, deleted_object_ap_id: str
|
|
) -> models.InboxObject:
|
|
follow_from_inbox = RemoteObject(
|
|
factories.build_delete_activity(
|
|
from_remote_actor=actor,
|
|
deleted_object_ap_id=deleted_object_ap_id,
|
|
),
|
|
actor,
|
|
)
|
|
inbox_object = factories.InboxObjectFactory.from_remote_object(
|
|
follow_from_inbox, actor
|
|
)
|
|
return inbox_object
|
|
|
|
|
|
def run_async(func, *args, **kwargs):
|
|
async def _func():
|
|
async with async_session() as db:
|
|
return await func(db, *args, **kwargs)
|
|
|
|
asyncio.run(_func())
|