mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
|
"""Dependency-update bot per-project opt-in columns (migration 049).
|
||
|
|
|
||
|
|
Migration 049 adds ``projects.dep_update_command`` (varchar null) and
|
||
|
|
``projects.dep_update_paths`` (varchar[] null). 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_dep_update_columns_default_null(db_session: AsyncSession) -> None:
|
||
|
|
project = await _seed_project(db_session)
|
||
|
|
assert project.dep_update_command is None
|
||
|
|
assert project.dep_update_paths is None
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_dep_update_columns_round_trip(db_session: AsyncSession) -> None:
|
||
|
|
project = await _seed_project(db_session)
|
||
|
|
project.dep_update_command = "uv lock --upgrade"
|
||
|
|
project.dep_update_paths = ["uv.lock", "pnpm-lock.yaml"]
|
||
|
|
await db_session.flush()
|
||
|
|
row = (
|
||
|
|
await db_session.execute(
|
||
|
|
select(ProjectTable).where(ProjectTable.id == project.id)
|
||
|
|
)
|
||
|
|
).scalar_one()
|
||
|
|
assert row.dep_update_command == "uv lock --upgrade"
|
||
|
|
assert row.dep_update_paths == ["uv.lock", "pnpm-lock.yaml"]
|