"""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"},
    ]