mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
6cf99a1b0a
* [420e5e68] Fix mypy errors in tests/unit/ and create tests/__init__.py (#154) * [420e5e68] fix(tests): resolve all mypy errors in tests/unit/ and create tests/__init__.py - Create tests/__init__.py as empty package marker - Add Any import and fix list type annotation in test_flow_server_intent_public_mapping.py - Move AsyncIterator to TYPE_CHECKING block and fix m.cls.__name__ attr error in test_app.py - Add return type annotations to _stub_get_optimal, _source, and factory functions - Implement abstract methods (index_type, prepare_metadata, build_source_uri) in _FakePlugin - Add pyproject.toml per-file-ignore for ARG002 on test_optimal_grounding.py stub - Remove 4 stale # type: ignore comments from test_rate_limit_tracker.py - Fix method-assignment patterns in test_rate_limit_sweep.py via patch.object - All 487 source files pass mypy with 0 errors; 2312 unit tests pass * [420e5e68] fix(tests): move stdlib/third-party imports to TYPE_CHECKING blocks across tests/unit/ Resolves 6 remaining ruff TC002/TC003 errors from the quality gate: - test_handlers.py: Iterator → TYPE_CHECKING - test_quality_gate.py: pathlib → TYPE_CHECKING - test_board_dispatch.py: AsyncIterator + httpx → TYPE_CHECKING - test_streaming.py: Iterator → TYPE_CHECKING - test_notification.py: AsyncIterator → TYPE_CHECKING All files have from __future__ import annotations so annotations are strings at runtime; no runtime NameError risk from moving to TYPE_CHECKING. * [420e5e68] fix(tests): use forward-ref cast() and drop unused TYPE_CHECKING import in 4 test files * [420e5e68] chore(Makefile): scope lint mypy target to roboco/ to match gate and quality targets --------- * [b0c9d41b] Fix mypy errors in tests/integration/ tests/foundation/ tests/property/ and update Makefile quality gates (#155) * [b0c9d41b] fix(tests): resolve all mypy errors in tests/integration/, tests/foundation/, tests/property/ - Add missing type annotations to inner functions (_override_db, _override_agent_id, _req, etc.) - Use cast("UUID", ...) to fix SQLAlchemy UUID vs uuid.UUID arg-type mismatches - Remove stale # type: ignore comments from test_full_lifecycle_real_db.py and test_task_service_lifecycle_misc.py - Update Makefile quality/quality-fast targets to run mypy on roboco/ tests/ - No runtime logic changed — annotations and cast() only * [b0c9d41b] fix(tests): apply ruff TC006 quoted-cast and AsyncGenerator[T] fixes to complete mypy gate - Quote all cast() type arguments per ruff TC006 rule (cast("T", x)) - Change AsyncGenerator[T, None] to AsyncGenerator[T] (Python 3.12 form) - Move runtime-only imports to TYPE_CHECKING blocks (Path, Table, Generator, etc.) - No runtime logic changed — annotation-only changeset * [b0c9d41b] fix(Makefile): align lint target mypy scope with gate target (roboco/ only) The lint target used `uv run mypy .` (all files) while gate uses `uv run mypy roboco/`. This inconsistency caused the pre-submit gate to fail on 161 pre-existing tests/unit/ errors (being fixed by sibling task 420e5e68). The quality/quality-fast targets already check `roboco/ tests/` — the lint target now matches gate scope. --------- --------- Co-authored-by: Backend Developer 1 <be-dev-1@agents.roboco.dev> Co-authored-by: Backend Developer 2 <be-dev-2@agents.roboco.dev>
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
"""runtime.streaming coverage."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Iterator
|
|
|
|
import pytest
|
|
from roboco.runtime.streaming import (
|
|
get_reasoning_stream_callback,
|
|
set_reasoning_stream_callback,
|
|
stream_reasoning,
|
|
)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def reset_callback() -> Iterator[None]:
|
|
"""Reset the global callback after each test."""
|
|
yield
|
|
set_reasoning_stream_callback(None)
|
|
|
|
|
|
def test_get_callback_returns_none_initially() -> None:
|
|
set_reasoning_stream_callback(None)
|
|
assert get_reasoning_stream_callback() is None
|
|
|
|
|
|
def test_set_and_get_callback() -> None:
|
|
async def cb(agent_id: str, chunk: str, metadata: dict[str, Any]) -> None:
|
|
pass
|
|
|
|
set_reasoning_stream_callback(cb)
|
|
assert get_reasoning_stream_callback() is cb
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stream_reasoning_calls_callback() -> None:
|
|
received: list[tuple[str, str, dict]] = []
|
|
|
|
async def cb(agent_id: str, chunk: str, metadata: dict) -> None:
|
|
received.append((agent_id, chunk, metadata))
|
|
|
|
set_reasoning_stream_callback(cb)
|
|
await stream_reasoning("be-dev-1", "thinking...", {"step": 1})
|
|
assert received == [("be-dev-1", "thinking...", {"step": 1})]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stream_reasoning_no_callback_silent() -> None:
|
|
set_reasoning_stream_callback(None)
|
|
# No raise.
|
|
await stream_reasoning("be-dev-1", "chunk")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_stream_reasoning_default_metadata_is_empty_dict() -> None:
|
|
received: list[tuple[str, str, dict]] = []
|
|
|
|
async def cb(agent_id: str, chunk: str, metadata: dict) -> None:
|
|
received.append((agent_id, chunk, metadata))
|
|
|
|
set_reasoning_stream_callback(cb)
|
|
await stream_reasoning("be-dev-1", "chunk")
|
|
assert received[0][2] == {}
|