100% Coverage

tests/unit/api/test_schemas_provider.py

@@ -0,0 +1,24 @@
+"""Coverage for roboco.api.schemas.provider helpers."""
+
+from __future__ import annotations
+
+from types import SimpleNamespace
+from uuid import uuid4
+
+from roboco.api.schemas.provider import assignment_to_response
+from roboco.models.base import AssignmentScope, ModelProvider
+
+
+def test_assignment_to_response_round_trip() -> None:
+    provider = SimpleNamespace(type=ModelProvider.ANTHROPIC)
+    row = SimpleNamespace(
+        id=uuid4(),
+        scope=AssignmentScope.GLOBAL,
+        scope_value=None,
+        provider=provider,
+        model_name="opus",
+    )
+    response = assignment_to_response(row)
+    assert response.scope == AssignmentScope.GLOBAL
+    assert response.provider_type == ModelProvider.ANTHROPIC
+    assert response.model_name == "opus"