Publish curated ITWorx-ModelForge source from 990401737f96c8d46a9f7b0d84994516cadcb6a0

backend/tests/test_external_creative_governance.py

@@ -0,0 +1,136 @@
+from __future__ import annotations
+
+import uuid
+from pathlib import Path
+
+import pytest
+from pydantic import ValidationError
+
+from modelforge_api.domain.external_creative import (
+    ExternalActionAttestation,
+    ExternalAllocationCreate,
+    ExternalAllocationReconcile,
+    ExternalProfileObservation,
+    ExternalResourceApproval,
+)
+from modelforge_api.services.external_creative import PRIORITY, _digest
+
+AUDIO_WORKLOADS = (
+    "speech_generation",
+    "voice_clone",
+    "music_generation",
+    "sound_generation",
+    "foley_generation",
+    "transcription",
+    "audio_enhancement",
+    "lip_sync",
+)
+
+
+def test_priority_order_keeps_production_before_interactive_and_background() -> None:
+    assert PRIORITY["production"] < PRIORITY["interactive"] < PRIORITY["background"]
+    assert PRIORITY["background"] < PRIORITY["benchmark"]
+
+
+def test_resource_binding_requires_bounded_positive_profile() -> None:
+    binding = ExternalResourceApproval(
+        project_id=uuid.uuid4(),
+        resource_ref="wan.video@2.2",
+        workload="video",
+        provider_resource_id="configured-comfyui",
+        configured_vram_bytes=12 * 1024**3,
+    )
+    assert binding.approved is True
+    with pytest.raises(ValidationError):
+        binding.model_copy(update={"configured_vram_bytes": 0}).model_validate(
+            binding.model_copy(update={"configured_vram_bytes": 0}).model_dump()
+        )
+
+
+@pytest.mark.parametrize("workload", AUDIO_WORKLOADS)
+def test_audio_workloads_are_explicitly_admitted_by_the_contract(workload: str) -> None:
+    binding = ExternalResourceApproval(
+        project_id=uuid.uuid4(),
+        resource_ref="candidate.audio@exact-revision",
+        workload=workload,
+        provider_resource_id="configured-audio-provider",
+        approved=False,
+        configured_vram_bytes=4 * 1024**3,
+    )
+    assert binding.workload == workload
+    assert binding.approved is False
+
+
+def test_unknown_audio_workload_is_rejected() -> None:
+    with pytest.raises(ValidationError):
+        ExternalResourceApproval(
+            project_id=uuid.uuid4(),
+            resource_ref="candidate.audio@exact-revision",
+            workload="generic_audio",
+            provider_resource_id="configured-audio-provider",
+            approved=False,
+            configured_vram_bytes=4 * 1024**3,
+        )
+
+
+def test_workload_is_part_of_the_idempotent_allocation_identity() -> None:
+    common = {
+        "request_id": "director-shot-42",
+        "idempotency_key": "director-shot-42-sound",
+        "installation_id": "private-muse",
+        "resource_ref": "candidate.audio@exact-revision",
+        "provider_resource_id": "configured-audio-provider",
+        "required_vram_bytes": 4 * 1024**3,
+    }
+    speech = ExternalAllocationCreate(**common, workload="speech_generation")
+    clone = ExternalAllocationCreate(**common, workload="voice_clone")
+    assert _digest(speech) != _digest(clone)
+
+
+def test_attestation_and_reconciliation_are_closed_enums() -> None:
+    attestation = ExternalActionAttestation(
+        action="release_idle_comfy",
+        observed_before=10,
+        observed_after=4,
+        completed_at="2026-09-09T12:00:00Z",
+    )
+    assert attestation.observed_before - attestation.observed_after == 6
+    assert ExternalAllocationReconcile(action="confirm_released", reason="provider verified idle")
+    with pytest.raises(ValidationError):
+        ExternalAllocationReconcile(action="force", reason="provider verified idle")
+
+
+def test_profile_observation_is_bounded_and_allocation_bound() -> None:
+    observation = ExternalProfileObservation(
+        allocation_id=uuid.uuid4(),
+        resource_ref="flux.image@9b",
+        provider_resource_id="configured-comfyui",
+        workload="image",
+        observed_peak_vram_bytes=14 * 1024**3,
+        reserved_vram_bytes=16 * 1024**3,
+        duration_ms=120_000,
+        outcome="succeeded",
+    )
+    assert observation.observed_peak_vram_bytes <= observation.reserved_vram_bytes
+
+
+def test_migration_retains_each_profile_observation() -> None:
+    repository_root = Path(__file__).resolve().parents[2]
+    migration = (
+        repository_root / "backend/alembic/versions/20260909_0031_external_creative_governance.py"
+    ).read_text()
+    assert '"external_creative_profile_observations"' in migration
+    assert '"uq_external_creative_profile_observation_allocation"' in migration
+    assert 'down_revision = "20260909_0030"' in migration
+
+
+def test_audio_workload_migration_widens_every_persisted_route_component() -> None:
+    repository_root = Path(__file__).resolve().parents[2]
+    migration = (
+        repository_root / "backend/alembic/versions/20260909_0032_audio_creative_workloads.py"
+    ).read_text()
+    assert 'down_revision = "20260909_0031"' in migration
+    assert '"external_creative_allocations"' in migration
+    assert '"external_creative_resources"' in migration
+    assert '"external_creative_profile_observations"' in migration
+    assert "type_=sa.String(32)" in migration