Task 002: add shared domain contracts

2026-05-21 17:45:00 +03:00
parent d42f0c4392
commit a0ae06adfa
31 changed files with 4043 additions and 27 deletions
@@ -0,0 +1,77 @@
+from __future__ import annotations
+
+import sys
+import unittest
+from pathlib import Path
+
+from pydantic import ValidationError
+
+
+BACKEND_ROOT = Path(__file__).resolve().parents[2]
+sys.path.insert(0, str(BACKEND_ROOT))
+
+from src.domain.contracts import (  # noqa: E402
+    ARTICLE_WORKFLOW_STATUSES,
+    ARTICLE_WORKFLOW_TRANSITIONS,
+    ArticleCreateRequest,
+    ArticleWorkflowStatus,
+    Role,
+    UserSummary,
+)
+
+
+class DomainContractTest(unittest.TestCase):
+    def test_role_enum_is_canonical(self) -> None:
+        self.assertEqual(["ADMIN", "EDITOR"], [role.value for role in Role])
+
+        with self.assertRaises(ValidationError):
+            UserSummary.model_validate(
+                {
+                    "id": "00000000-0000-0000-0000-000000000001",
+                    "display_name": "Writer",
+                    "role": "WRITER",
+                }
+            )
+
+    def test_article_create_rejects_missing_required_fields(self) -> None:
+        with self.assertRaises(ValidationError) as error:
+            ArticleCreateRequest.model_validate({})
+
+        missing_fields = {
+            issue["loc"][0]
+            for issue in error.exception.errors()
+            if issue["type"] == "missing"
+        }
+        self.assertEqual({"brief_description", "target_site_id"}, missing_fields)
+
+    def test_workflow_statuses_and_transitions_are_shared_constants(self) -> None:
+        self.assertEqual(
+            [
+                "ARTICLE_BRIEF_CREATED",
+                "BOUNDARY_QUESTIONS_GENERATED",
+                "BOUNDARY_ANSWERS_SUBMITTED",
+                "PLAN_GENERATED",
+                "PLAN_REVIEW_REQUIRED",
+                "PLAN_REVISION_REQUIRED",
+                "RESEARCH_RUNNING",
+                "EVIDENCE_MATRIX_READY",
+                "PARALLEL_PRODUCTION_RUNNING",
+                "DRAFT_ASSEMBLED",
+                "SEO_AND_LANGUAGE_REVIEW_READY",
+                "FINAL_REVIEW_REQUIRED",
+                "FINAL_REVISION_REQUIRED",
+                "PUBLISH_DRY_RUN_REQUIRED",
+                "PUBLISH_COMMIT_READY",
+                "PUBLISH_COMMIT_CREATED",
+            ],
+            [status.value for status in ARTICLE_WORKFLOW_STATUSES],
+        )
+        self.assertEqual(set(ArticleWorkflowStatus), set(ARTICLE_WORKFLOW_TRANSITIONS))
+        for status, next_statuses in ARTICLE_WORKFLOW_TRANSITIONS.items():
+            self.assertIsInstance(status, ArticleWorkflowStatus)
+            for next_status in next_statuses:
+                self.assertIsInstance(next_status, ArticleWorkflowStatus)
+
+
+if __name__ == "__main__":
+    unittest.main()