Model/tests/backend/app/documents/test_bulk_download_router.py

"""Tests for the Bulk Document Download trigger endpoint (ADR-0060).

Session, SQS and requester-email seams are dependency-injected; tests run
against the ephemeral Postgres and record message bodies instead of calling AWS.
"""

import json
from collections.abc import Iterator
from dataclasses import dataclass, field
from typing import Any
from uuid import UUID

import pytest
from fastapi import FastAPI
from fastapi.testclient import TestClient
from sqlalchemy import Engine
from sqlmodel import Session

from backend.app.dependencies import validate_token
from backend.app.documents import router as documents_router
from backend.app.documents.router import (
    get_message_sender,
    get_requesting_user_email,
    get_session,
)
from infrastructure.postgres.property_table import PropertyRow
from infrastructure.postgres.task_table import TaskRow
from repositories.tasks.subtask_postgres_repository import SubTaskPostgresRepository

PORTFOLIO_ID = 814
RECIPIENT = "tester@example.com"


@dataclass
class Api:
    client: TestClient
    engine: Engine
    sent_bodies: list[str] = field(default_factory=list)

    def seed_task(self, selection: dict[str, Any]) -> UUID:
        """Create the app-owned task with the FE-written selection config on
        ``task.inputs`` (ADR-0060)."""
        with Session(self.engine) as session:
            row = TaskRow(
                task_source="app:bulk_document_download",
                inputs=json.dumps(selection),
            )
            session.add(row)
            session.commit()
            session.refresh(row)
            return row.id

    def seed_property(self, landlord_property_id: str) -> int:
        with Session(self.engine) as session:
            row = PropertyRow(
                portfolio_id=PORTFOLIO_ID, landlord_property_id=landlord_property_id
            )
            session.add(row)
            session.commit()
            session.refresh(row)
            assert row.id is not None
            return row.id

    def subtask_inputs(self, task_id: UUID) -> list[dict[str, Any]]:
        with Session(self.engine) as session:
            subtasks = SubTaskPostgresRepository(session).list_by_task(task_id)
            return [st.inputs or {} for st in subtasks]


@pytest.fixture
def api(db_engine: Engine) -> Api:
    app = FastAPI()
    app.include_router(documents_router.router, prefix="/v1")
    harness = Api(client=TestClient(app), engine=db_engine)

    def _session() -> Iterator[Session]:
        with Session(db_engine) as session:
            yield session

    app.dependency_overrides[get_session] = _session
    app.dependency_overrides[get_message_sender] = lambda: harness.sent_bodies.append
    app.dependency_overrides[get_requesting_user_email] = lambda: RECIPIENT
    app.dependency_overrides[validate_token] = lambda: "test-token"
    return harness


def _trigger(api: Api, task_id: UUID) -> Any:
    return api.client.post(
        "/v1/documents/bulk-download", json={"task_id": str(task_id)}
    )


def test_pins_the_recipe_and_enqueues_one_message(api: Api) -> None:
    # arrange — two selected properties, pinned as the FE's task.inputs config.
    p1 = api.seed_property("LP1")
    p2 = api.seed_property("LP2")
    task_id = api.seed_task(
        {"portfolio_id": PORTFOLIO_ID, "property_ids": [p1, p2]}
    )

    # act
    response = _trigger(api, task_id)

    # assert — accepted; one sub_task pins the recipe; one message enqueued.
    assert response.status_code == 202
    inputs = api.subtask_inputs(task_id)
    assert len(inputs) == 1
    assert inputs[0]["landlord_property_ids"] == ["LP1", "LP2"]
    assert inputs[0]["recipient_email"] == RECIPIENT
    assert len(api.sent_bodies) == 1
    message = json.loads(api.sent_bodies[0])
    assert message["task_id"] == str(task_id)
    assert "subtask_id" in message


def test_rejects_a_selection_over_the_cap(api: Api, monkeypatch: Any) -> None:
    # arrange — cap of 1, two selected properties.
    monkeypatch.setattr(documents_router, "MAX_PROPERTIES", 1)
    p1 = api.seed_property("LP1")
    p2 = api.seed_property("LP2")
    task_id = api.seed_task(
        {"portfolio_id": PORTFOLIO_ID, "property_ids": [p1, p2]}
    )

    # act
    response = _trigger(api, task_id)

    # assert — refused; nothing created or sent.
    assert response.status_code == 400
    assert api.subtask_inputs(task_id) == []
    assert api.sent_bodies == []


def test_rejects_a_task_that_already_has_a_subtask(api: Api) -> None:
    # arrange — a task already triggered once.
    p1 = api.seed_property("LP1")
    task_id = api.seed_task({"portfolio_id": PORTFOLIO_ID, "property_ids": [p1]})
    assert _trigger(api, task_id).status_code == 202

    # act
    response = _trigger(api, task_id)

    # assert — the double-submit is refused; still one sub_task, one message.
    assert response.status_code == 409
    assert len(api.subtask_inputs(task_id)) == 1
    assert len(api.sent_bodies) == 1


def test_rejects_a_task_with_no_selection_config(api: Api) -> None:
    # arrange — a task whose inputs carry no portfolio_id/selection.
    api.seed_property("LP1")
    task_id = api.seed_task({})

    # act
    response = _trigger(api, task_id)

    # assert
    assert response.status_code == 400
    assert api.sent_bodies == []