mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
|
"""Per-project environment ladder column (migration 073).
|
||
|
|
|
||
|
|
Migration 073 adds ``projects.environments`` (JSONB 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,
|
||
|
|
mirroring ``test_migration_dep_update.py``.
|
||
|
|
"""
|
||
|
|
|
||
|
|
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="L-Proj",
|
||
|
|
slug=f"l-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_environments_defaults_null(db_session: AsyncSession) -> None:
|
||
|
|
project = await _seed_project(db_session)
|
||
|
|
assert project.environments is None
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_environments_round_trip(db_session: AsyncSession) -> None:
|
||
|
|
project = await _seed_project(db_session)
|
||
|
|
project.environments = [
|
||
|
|
{"name": "head", "branch": "dev"},
|
||
|
|
{"name": "prod", "branch": "master"},
|
||
|
|
]
|
||
|
|
await db_session.flush()
|
||
|
|
row = (
|
||
|
|
await db_session.execute(
|
||
|
|
select(ProjectTable).where(ProjectTable.id == project.id)
|
||
|
|
)
|
||
|
|
).scalar_one()
|
||
|
|
assert row.environments == [
|
||
|
|
{"name": "head", "branch": "dev"},
|
||
|
|
{"name": "prod", "branch": "master"},
|
||
|
|
]
|