mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
|
"""Multi-repo CI-watch per-project opt-in columns (migration 048).
|
||
|
|
|
||
|
|
Migration 048 adds ``projects.ci_watch_enabled`` (bool, NOT NULL default false —
|
||
|
|
an unopted project is unwatched) and ``projects.ci_watch_workflow`` (varchar
|
||
|
|
null — scope the CI signal to one workflow file). The real upgrade/downgrade
|
||
|
|
chain is verified separately against a throwaway Postgres; these assertions
|
||
|
|
guard the resulting schema shape and a value round-trip.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import TYPE_CHECKING
|
||
|
|
from uuid import uuid4
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from roboco.db.tables import AgentTable, ProjectTable
|
||
|
|
from roboco.models import AgentRole, AgentStatus, Team
|
||
|
|
from sqlalchemy import select
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
|
||
|
|
async def _seed_project(db_session: AsyncSession) -> ProjectTable:
|
||
|
|
agent = AgentTable(
|
||
|
|
id=uuid4(),
|
||
|
|
name="Dev",
|
||
|
|
slug=f"be-dev-{uuid4().hex[:8]}",
|
||
|
|
role=AgentRole.DEVELOPER,
|
||
|
|
team=Team.BACKEND,
|
||
|
|
status=AgentStatus.ACTIVE,
|
||
|
|
model_config={},
|
||
|
|
system_prompt="dev",
|
||
|
|
capabilities=[],
|
||
|
|
permissions={},
|
||
|
|
metrics={},
|
||
|
|
)
|
||
|
|
db_session.add(agent)
|
||
|
|
await db_session.flush()
|
||
|
|
project = ProjectTable(
|
||
|
|
id=uuid4(),
|
||
|
|
name="B-Proj",
|
||
|
|
slug=f"b-proj-{uuid4().hex[:8]}",
|
||
|
|
git_url="https://example.com/r.git",
|
||
|
|
assigned_cell=Team.BACKEND,
|
||
|
|
created_by=agent.id,
|
||
|
|
)
|
||
|
|
db_session.add(project)
|
||
|
|
await db_session.flush()
|
||
|
|
return project
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_ci_watch_columns_default_off(db_session: AsyncSession) -> None:
|
||
|
|
project = await _seed_project(db_session)
|
||
|
|
assert project.ci_watch_enabled is False
|
||
|
|
assert project.ci_watch_workflow is None
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_ci_watch_columns_round_trip(db_session: AsyncSession) -> None:
|
||
|
|
project = await _seed_project(db_session)
|
||
|
|
project.ci_watch_enabled = True
|
||
|
|
project.ci_watch_workflow = "ci.yml"
|
||
|
|
await db_session.flush()
|
||
|
|
row = (
|
||
|
|
await db_session.execute(
|
||
|
|
select(ProjectTable).where(ProjectTable.id == project.id)
|
||
|
|
)
|
||
|
|
).scalar_one()
|
||
|
|
assert row.ci_watch_enabled is True
|
||
|
|
assert row.ci_watch_workflow == "ci.yml"
|