refactor: enhance row validation in activity and tasks APIs for better type safety
This commit is contained in:
89
backend/tests/test_activity_api_rows.py
Normal file
89
backend/tests/test_activity_api_rows.py
Normal file
@@ -0,0 +1,89 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
from uuid import uuid4
|
||||
|
||||
import pytest
|
||||
|
||||
from app.api.activity import _coerce_task_comment_rows
|
||||
from app.models.activity_events import ActivityEvent
|
||||
from app.models.agents import Agent
|
||||
from app.models.boards import Board
|
||||
from app.models.tasks import Task
|
||||
|
||||
|
||||
@dataclass
|
||||
class _FakeSqlRow4:
|
||||
first: object
|
||||
second: object
|
||||
third: object
|
||||
fourth: object
|
||||
|
||||
def __len__(self) -> int:
|
||||
return 4
|
||||
|
||||
def __getitem__(self, index: int) -> object:
|
||||
if index == 0:
|
||||
return self.first
|
||||
if index == 1:
|
||||
return self.second
|
||||
if index == 2:
|
||||
return self.third
|
||||
if index == 3:
|
||||
return self.fourth
|
||||
raise IndexError(index)
|
||||
|
||||
|
||||
def _make_event() -> ActivityEvent:
|
||||
return ActivityEvent(event_type="task.comment", message="hello")
|
||||
|
||||
|
||||
def _make_board() -> Board:
|
||||
return Board(
|
||||
organization_id=uuid4(),
|
||||
name="B",
|
||||
slug="b",
|
||||
)
|
||||
|
||||
|
||||
def _make_task(board_id) -> Task:
|
||||
return Task(board_id=board_id, title="T")
|
||||
|
||||
|
||||
def _make_agent(board_id) -> Agent:
|
||||
return Agent(
|
||||
board_id=board_id,
|
||||
gateway_id=uuid4(),
|
||||
name="A",
|
||||
)
|
||||
|
||||
|
||||
def test_coerce_task_comment_rows_accepts_plain_tuple():
|
||||
board = _make_board()
|
||||
task = _make_task(board.id)
|
||||
event = _make_event()
|
||||
agent = _make_agent(board.id)
|
||||
|
||||
rows = _coerce_task_comment_rows([(event, task, board, agent)])
|
||||
assert rows == [(event, task, board, agent)]
|
||||
|
||||
|
||||
def test_coerce_task_comment_rows_accepts_row_like_values():
|
||||
board = _make_board()
|
||||
task = _make_task(board.id)
|
||||
event = _make_event()
|
||||
row = _FakeSqlRow4(event, task, board, None)
|
||||
|
||||
rows = _coerce_task_comment_rows([row])
|
||||
assert rows == [(event, task, board, None)]
|
||||
|
||||
|
||||
def test_coerce_task_comment_rows_rejects_invalid_values():
|
||||
board = _make_board()
|
||||
task = _make_task(board.id)
|
||||
|
||||
with pytest.raises(
|
||||
TypeError,
|
||||
match="Expected \\(ActivityEvent, Task, Board, Agent \\| None\\) rows",
|
||||
):
|
||||
_coerce_task_comment_rows([(uuid4(), task, board, None)])
|
||||
53
backend/tests/test_tasks_api_rows.py
Normal file
53
backend/tests/test_tasks_api_rows.py
Normal file
@@ -0,0 +1,53 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
from uuid import uuid4
|
||||
|
||||
import pytest
|
||||
|
||||
from app.api.tasks import _coerce_task_event_rows
|
||||
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")])
|
||||
Reference in New Issue
Block a user