90 lines
2.1 KiB
Python
90 lines
2.1 KiB
Python
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)])
|