118 lines
3.2 KiB
Python
118 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
|
|
from app.api.tasks import _coerce_task_event_rows, _task_event_payload
|
|
from app.models.activity_events import ActivityEvent
|
|
from app.models.tasks import Task
|
|
|
|
|
|
@dataclass
|
|
class _FakeSqlRow:
|
|
first: object
|
|
second: object
|
|
|
|
def __len__(self) -> int:
|
|
return 2
|
|
|
|
def __getitem__(self, index: int) -> object:
|
|
if index == 0:
|
|
return self.first
|
|
if index == 1:
|
|
return self.second
|
|
raise IndexError(index)
|
|
|
|
|
|
def _make_event() -> ActivityEvent:
|
|
return ActivityEvent(event_type="task.updated")
|
|
|
|
|
|
def _make_task() -> Task:
|
|
return Task(board_id=uuid4(), title="T")
|
|
|
|
|
|
def test_coerce_task_event_rows_accepts_plain_tuple():
|
|
event = _make_event()
|
|
task = _make_task()
|
|
rows = _coerce_task_event_rows([(event, task)])
|
|
assert rows == [(event, task)]
|
|
|
|
|
|
def test_coerce_task_event_rows_accepts_row_like_values():
|
|
event = _make_event()
|
|
task = _make_task()
|
|
rows = _coerce_task_event_rows([_FakeSqlRow(event, task)])
|
|
assert rows == [(event, task)]
|
|
|
|
|
|
def test_coerce_task_event_rows_rejects_invalid_values():
|
|
with pytest.raises(TypeError, match="Expected \\(ActivityEvent, Task \\| None\\) rows"):
|
|
_coerce_task_event_rows([("bad", "row")])
|
|
|
|
|
|
def test_task_event_payload_includes_activity_for_comment_event() -> None:
|
|
task = Task(board_id=uuid4(), title="Ship patch")
|
|
event = ActivityEvent(
|
|
event_type="task.comment",
|
|
message="Looks good.",
|
|
task_id=task.id,
|
|
agent_id=uuid4(),
|
|
)
|
|
|
|
payload = _task_event_payload(
|
|
event,
|
|
task,
|
|
deps_map={},
|
|
dep_status={},
|
|
tag_state_by_task_id={},
|
|
)
|
|
|
|
assert payload["type"] == "task.comment"
|
|
assert payload["activity"] == {
|
|
"id": str(event.id),
|
|
"event_type": "task.comment",
|
|
"message": "Looks good.",
|
|
"agent_id": str(event.agent_id),
|
|
"task_id": str(task.id),
|
|
"created_at": event.created_at.isoformat(),
|
|
}
|
|
comment = payload["comment"]
|
|
assert isinstance(comment, dict)
|
|
assert comment["id"] == str(event.id)
|
|
assert comment["task_id"] == str(task.id)
|
|
assert comment["message"] == "Looks good."
|
|
|
|
|
|
def test_task_event_payload_includes_activity_for_non_comment_event() -> None:
|
|
task = Task(board_id=uuid4(), title="Wire stream events", status="in_progress")
|
|
event = ActivityEvent(
|
|
event_type="task.updated",
|
|
message="Task updated: Wire stream events.",
|
|
task_id=task.id,
|
|
)
|
|
|
|
payload = _task_event_payload(
|
|
event,
|
|
task,
|
|
deps_map={},
|
|
dep_status={},
|
|
tag_state_by_task_id={},
|
|
)
|
|
|
|
assert payload["type"] == "task.updated"
|
|
assert payload["activity"] == {
|
|
"id": str(event.id),
|
|
"event_type": "task.updated",
|
|
"message": "Task updated: Wire stream events.",
|
|
"agent_id": None,
|
|
"task_id": str(task.id),
|
|
"created_at": event.created_at.isoformat(),
|
|
}
|
|
task_payload = payload["task"]
|
|
assert isinstance(task_payload, dict)
|
|
assert task_payload["id"] == str(task.id)
|
|
assert task_payload["is_blocked"] is False
|