mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
* [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>
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
"""Default branch convention resolves to ``master`` (user-repo convention).
|
|
|
|
Driven by the real Postgres ``db_session`` fixture so the ORM column default
|
|
is exercised by the same SQLAlchemy flush path production uses.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
from typing import TYPE_CHECKING, cast
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
from roboco.db.tables import AgentTable, ProjectTable, TaskTable
|
|
from roboco.models import AgentRole, AgentStatus, Team
|
|
from roboco.services.git import GitService
|
|
from roboco.services.task import TaskService
|
|
|
|
if TYPE_CHECKING:
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
|
|
async def _seed_creator(db_session: AsyncSession) -> AgentTable:
|
|
agent = AgentTable(
|
|
id=uuid4(),
|
|
name="System",
|
|
slug=f"system-{uuid4().hex[:8]}",
|
|
role=AgentRole.SYSTEM,
|
|
team=None,
|
|
status=AgentStatus.ACTIVE,
|
|
model_config={},
|
|
system_prompt="system",
|
|
capabilities=[],
|
|
permissions={},
|
|
metrics={},
|
|
)
|
|
db_session.add(agent)
|
|
await db_session.flush()
|
|
return agent
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_project_default_branch_defaults_to_master(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
"""A project inserted without a ``default_branch`` resolves to ``master``."""
|
|
creator = await _seed_creator(db_session)
|
|
project = ProjectTable(
|
|
id=uuid4(),
|
|
name="No Branch Project",
|
|
slug=f"proj-{uuid4().hex[:8]}",
|
|
git_url="https://github.com/example/no-branch.git",
|
|
assigned_cell=Team.BACKEND,
|
|
created_by=creator.id,
|
|
)
|
|
db_session.add(project)
|
|
await db_session.flush()
|
|
await db_session.refresh(project)
|
|
|
|
assert project.default_branch == "master"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_parent_branch_falls_back_to_master(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
"""The parent-branch resolver falls back to ``master`` when the project
|
|
default is falsy."""
|
|
svc = TaskService(db_session)
|
|
task = SimpleNamespace(id=uuid4(), parent_task_id=None)
|
|
project = SimpleNamespace(default_branch="")
|
|
|
|
branch = await svc._resolve_parent_branch(cast("TaskTable", task), project)
|
|
|
|
assert branch == "master"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_git_project_default_branch_falls_back_to_master(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
"""The git default-branch lookup falls back to ``master`` when the stored
|
|
project default is falsy."""
|
|
creator = await _seed_creator(db_session)
|
|
slug = f"proj-{uuid4().hex[:8]}"
|
|
project = ProjectTable(
|
|
id=uuid4(),
|
|
name="Falsy Branch Project",
|
|
slug=slug,
|
|
git_url="https://github.com/example/falsy-branch.git",
|
|
default_branch="",
|
|
assigned_cell=Team.BACKEND,
|
|
created_by=creator.id,
|
|
)
|
|
db_session.add(project)
|
|
await db_session.flush()
|
|
|
|
svc = GitService(db_session)
|
|
branch = await svc._project_default_branch(slug)
|
|
|
|
assert branch == "master"
|