Initial commit

2026-07-01 17:56:31 +08:00
commit d78931979d
731 changed files with 311088 additions and 0 deletions
@@ -0,0 +1,148 @@
+from __future__ import annotations
+
+from types import SimpleNamespace
+
+from opc.plugins.office_ui.execution_identity import (
+    canonicalize_execution_identity,
+    execution_identity_from_task,
+)
+
+
+def test_company_identity_clears_stale_custom_org_fields() -> None:
+    identity = canonicalize_execution_identity(
+        exec_mode="company",
+        company_profile="custom",
+        org_id="quantum_harbor",
+        preferred_agent="codex",
+        explicit_exec_mode=True,
+    )
+
+    assert identity.exec_mode == "company"
+    assert identity.company_profile == "corporate"
+    assert identity.org_id == ""
+    assert identity.preferred_agent == "codex"
+
+
+def test_legacy_custom_profile_becomes_custom_org_identity() -> None:
+    identity = canonicalize_execution_identity(
+        company_profile="custom",
+        org_id="quantum_harbor",
+        explicit_exec_mode=False,
+    )
+
+    assert identity.exec_mode == "org"
+    assert identity.company_profile == "custom"
+    assert identity.org_id == "quantum_harbor"
+
+
+def test_custom_exec_mode_is_canonical_org_identity() -> None:
+    identity = canonicalize_execution_identity(
+        exec_mode="custom",
+        company_profile="corporate",
+        org_id="quantum_harbor",
+        explicit_exec_mode=True,
+    )
+
+    assert identity.exec_mode == "org"
+    assert identity.company_profile == "custom"
+    assert identity.org_id == "quantum_harbor"
+
+
+def test_task_identity_clears_company_and_org_fields() -> None:
+    identity = canonicalize_execution_identity(
+        exec_mode="task",
+        company_profile="custom",
+        org_id="quantum_harbor",
+        preferred_agent="claude-code",
+        explicit_exec_mode=True,
+    )
+
+    assert identity.exec_mode == "task"
+    assert identity.company_profile == "corporate"
+    assert identity.org_id == ""
+    assert identity.preferred_agent == "claude_code"
+
+
+def test_task_metadata_identity_prefers_explicit_company_over_stale_custom_profile() -> None:
+    task = SimpleNamespace(
+        metadata={
+            "exec_mode": "company",
+            "company_profile": "custom",
+            "org_id": "quantum_harbor",
+            "preferred_agent": "codex",
+        },
+        org_id="quantum_harbor",
+    )
+
+    identity = execution_identity_from_task(task)
+
+    assert identity.exec_mode == "company"
+    assert identity.company_profile == "corporate"
+    assert identity.org_id == ""
+    assert identity.preferred_agent == "codex"
+
+
+def test_task_metadata_identity_keeps_custom_org_id() -> None:
+    task = SimpleNamespace(
+        metadata={
+            "exec_mode": "org",
+            "company_profile": "custom",
+            "organization_id": "quantum_harbor",
+        },
+        org_id=None,
+    )
+
+    identity = execution_identity_from_task(task)
+
+    assert identity.exec_mode == "org"
+    assert identity.company_profile == "custom"
+    assert identity.org_id == "quantum_harbor"
+
+
+def test_task_org_id_field_is_org_identity_fallback() -> None:
+    task = SimpleNamespace(
+        metadata={},
+        org_id="quantum_harbor",
+    )
+
+    identity = execution_identity_from_task(task)
+
+    assert identity.exec_mode == "org"
+    assert identity.company_profile == "custom"
+    assert identity.org_id == "quantum_harbor"
+
+
+def test_default_org_id_applies_only_when_task_has_no_persisted_identity() -> None:
+    task = SimpleNamespace(metadata={}, org_id=None)
+
+    identity = execution_identity_from_task(
+        task,
+        default_exec_mode="org",
+        default_company_profile="custom",
+        default_org_id="quantum_harbor",
+    )
+
+    assert identity.exec_mode == "org"
+    assert identity.company_profile == "custom"
+    assert identity.org_id == "quantum_harbor"
+
+
+def test_explicit_company_identity_ignores_default_org_id() -> None:
+    task = SimpleNamespace(
+        metadata={
+            "exec_mode": "company",
+            "company_profile": "corporate",
+        },
+        org_id=None,
+    )
+
+    identity = execution_identity_from_task(
+        task,
+        default_exec_mode="org",
+        default_company_profile="custom",
+        default_org_id="quantum_harbor",
+    )
+
+    assert identity.exec_mode == "company"
+    assert identity.company_profile == "corporate"
+    assert identity.org_id == ""