mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
|
"""Unit coverage for the shared notification-text helpers.
|
||
|
|
|
||
|
|
``task_display``/``agent_display`` are the producer-side fix so a human
|
||
|
|
reading a notification (panel, bell, Telegram) sees a task title / agent
|
||
|
|
slug instead of a raw UUID — see ``roboco/services/notification_text.py``.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Any
|
||
|
|
from unittest.mock import AsyncMock, MagicMock
|
||
|
|
from uuid import uuid4
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from roboco.foundation.identity import AGENTS
|
||
|
|
from roboco.services.notification_text import agent_display, task_display
|
||
|
|
|
||
|
|
|
||
|
|
class _Task:
|
||
|
|
def __init__(self, title: str | None) -> None:
|
||
|
|
self.title = title
|
||
|
|
|
||
|
|
|
||
|
|
def test_task_display_prefers_title_from_row() -> None:
|
||
|
|
task_id = uuid4()
|
||
|
|
display = task_display(_Task("Fix login bug"), task_id)
|
||
|
|
assert display == f"'Fix login bug' (#{str(task_id)[:8]})"
|
||
|
|
|
||
|
|
|
||
|
|
def test_task_display_accepts_bare_title_string() -> None:
|
||
|
|
task_id = uuid4()
|
||
|
|
display = task_display("Fix login bug", task_id)
|
||
|
|
assert display == f"'Fix login bug' (#{str(task_id)[:8]})"
|
||
|
|
|
||
|
|
|
||
|
|
def test_task_display_truncates_long_titles() -> None:
|
||
|
|
long_title = "x" * 100
|
||
|
|
task_id = uuid4()
|
||
|
|
display = task_display(long_title, task_id)
|
||
|
|
assert display == f"'{'x' * 40}' (#{str(task_id)[:8]})"
|
||
|
|
|
||
|
|
|
||
|
|
def test_task_display_falls_back_to_short_id_when_no_title() -> None:
|
||
|
|
task_id = uuid4()
|
||
|
|
assert task_display(None, task_id) == f"#{str(task_id)[:8]}"
|
||
|
|
assert task_display(_Task(None), task_id) == f"#{str(task_id)[:8]}"
|
||
|
|
assert task_display(_Task(""), task_id) == f"#{str(task_id)[:8]}"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_agent_display_resolves_via_static_map() -> None:
|
||
|
|
"""A fixed-roster agent's UUID resolves to its slug with zero DB I/O."""
|
||
|
|
row = AGENTS["be-dev-1"]
|
||
|
|
assert await agent_display(row.uuid) == "be-dev-1"
|
||
|
|
assert await agent_display(str(row.uuid)) == "be-dev-1"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_agent_display_falls_back_to_db_for_unknown_uuid() -> None:
|
||
|
|
"""A UUID absent from the static map resolves via ``get_agent_slug`` when
|
||
|
|
a db session is supplied."""
|
||
|
|
unknown_uuid = uuid4()
|
||
|
|
db: Any = MagicMock()
|
||
|
|
result = MagicMock()
|
||
|
|
result.scalar_one_or_none.return_value = "fresh-agent"
|
||
|
|
db.execute = AsyncMock(return_value=result)
|
||
|
|
|
||
|
|
assert await agent_display(unknown_uuid, db) == "fresh-agent"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_agent_display_raw_fallback_without_db() -> None:
|
||
|
|
"""No static-map hit + no db session ⇒ the raw value passes through."""
|
||
|
|
unknown_uuid = uuid4()
|
||
|
|
assert await agent_display(unknown_uuid) == str(unknown_uuid)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_agent_display_raw_fallback_for_non_uuid_slug() -> None:
|
||
|
|
"""A plain slug string that isn't in the UUID-keyed map passes through
|
||
|
|
unchanged (e.g. a value that's already a friendly slug)."""
|
||
|
|
assert await agent_display("some-slug") == "some-slug"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_agent_display_passes_through_none() -> None:
|
||
|
|
"""None stays None — "unassigned"/"its owner" wording stays at call sites."""
|
||
|
|
assert await agent_display(None) is None
|