roboco/tests/unit/models/test_project_sandbox_services.py

"""Project.sandbox_services / ProjectUpdate.sandbox_services validation.

Only "postgres" and "redis" are recognized sandbox services (mirrors the
provisioner's VALID_SANDBOX_SERVICES) — an unknown value must be rejected with
a clear message rather than silently accepted and later failing at provision
time inside a container spawn.
"""

from __future__ import annotations

from uuid import uuid4

import pytest
from pydantic import ValidationError
from roboco.models.base import Team
from roboco.models.project import Project, ProjectUpdate


def _project(sandbox_services: list[str] | None = None) -> Project:
    return Project(
        name="P",
        slug="p",
        git_url="https://example.com/r.git",
        assigned_cell=Team.BACKEND,
        created_by=uuid4(),
        sandbox_services=sandbox_services,
    )


def test_project_accepts_valid_sandbox_services() -> None:
    project = _project(sandbox_services=["postgres", "redis"])
    assert project.sandbox_services == ["postgres", "redis"]


def test_project_normalizes_sandbox_services_order_and_dupes() -> None:
    project = _project(sandbox_services=["redis", "postgres", "redis"])
    assert project.sandbox_services == ["postgres", "redis"]


def test_project_defaults_sandbox_services_to_none() -> None:
    project = _project()
    assert project.sandbox_services is None


def test_project_rejects_unknown_sandbox_service() -> None:
    with pytest.raises(ValidationError):
        _project(sandbox_services=["mysql"])


def test_project_update_accepts_valid_sandbox_services() -> None:
    update = ProjectUpdate(sandbox_services=["postgres"])
    assert update.sandbox_services == ["postgres"]


def test_project_update_rejects_unknown_sandbox_service() -> None:
    with pytest.raises(ValidationError):
        ProjectUpdate(sandbox_services=["mysql"])


def test_project_update_accepts_empty_list() -> None:
    update = ProjectUpdate(sandbox_services=[])
    assert update.sandbox_services == []