Add endpoint to patch artefact status (#64)

backend/test_observer/controllers/artefacts/artefacts.py

@@ -25,7 +25,7 @@
 from test_observer.data_access.repository import get_artefacts_by_family
 from test_observer.data_access.setup import get_db
 
-from .models import ArtefactBuildDTO, ArtefactDTO
+from .models import ArtefactBuildDTO, ArtefactDTO, ArtefactPatch
 
 router = APIRouter()
 
@@ -55,6 +55,19 @@ def get_artefact(artefact_id: int, db: Session = Depends(get_db)):
     return artefact
 
 
+@router.patch("/{artefact_id}")
+def patch_artefact(
+    artefact_id: int, request: ArtefactPatch, db: Session = Depends(get_db)
+):
+    artefact = db.get(Artefact, artefact_id)
+
+    if artefact is None:
+        raise HTTPException(status_code=404, detail="Artefact not found")
+
+    artefact.status = request.status
+    db.commit()
+
+
 @router.get("/{artefact_id}/builds", response_model=list[ArtefactBuildDTO])
 def get_artefact_builds(artefact_id: int, db: Session = Depends(get_db)):
     """Get latest artefact builds of an artefact together with their test executions"""

backend/test_observer/controllers/artefacts/models.py

@@ -19,7 +19,7 @@
 #        Nadzeya Hutsko <[email protected]>
 from pydantic import AliasPath, BaseModel, ConfigDict, Field
 
-from test_observer.data_access.models_enums import TestExecutionStatus
+from test_observer.data_access.models_enums import ArtefactStatus, TestExecutionStatus
 
 
 class ArtefactDTO(BaseModel):
@@ -60,3 +60,7 @@ class ArtefactBuildDTO(BaseModel):
     architecture: str
     revision: int | None
     test_executions: list[TestExecutionDTO]
+
+
+class ArtefactPatch(BaseModel):
+    status: ArtefactStatus

backend/test_observer/data_access/models_enums.py

@@ -35,6 +35,6 @@ class TestExecutionStatus(Enum):
     NOT_TESTED = "NOT_TESTED"
 
 
-class ArtefactStatus(Enum):
+class ArtefactStatus(str, Enum):
     APPROVED = "APPROVED"
     MARKED_AS_FAILED = "MARKED_AS_FAILED"

backend/tests/controllers/artefacts/test_artefacts.py

@@ -23,6 +23,7 @@
 from sqlalchemy.orm import Session
 
 from test_observer.data_access.models import ArtefactBuild, Environment, TestExecution
+from test_observer.data_access.models_enums import ArtefactStatus
 from tests.helpers import create_artefact
 
 
@@ -129,3 +130,17 @@ def test_get_artefact_builds_only_latest(db_session: Session, test_client: TestC
             "test_executions": [],
         }
     ]
+
+
+def test_artefact_signoff(db_session: Session, test_client: TestClient):
+    artefact = create_artefact(db_session, "candidate")
+
+    response = test_client.patch(
+        f"/v1/artefacts/{artefact.id}",
+        json={"status": ArtefactStatus.APPROVED},
+    )
+
+    db_session.refresh(artefact)
+
+    assert response.status_code == 200
+    assert artefact.status == ArtefactStatus.APPROVED