Python: Add Purview Middleware (#1142)

* [Py Purview] Purview Python Initial Commit * [Py Purview] Purview Python Minor Fixes * [Py Purview] Purview Python Comment Fixesish * [Py Purview] Purview Python Agent Middleware Done * [Py Purview] Purview Python Agent Middleware Done * [Py Purview] Purview Python Lint Errors * [Py Purview] Purview Python Final Hopefully * [Py Purview] Purview Python Final Hopefully * [Py Purview] Purview Python Fix ReadMe * [Py Purview] Purview Python Fix MyPy * [Py Purview] Purview Python Minor Updates on comments * [Py Purview] Purview Python Fix Build Error --------- Co-authored-by: Dmytro Struk <13853051+dmytrostruk@users.noreply.github.com>
2026-06-16 21:04:09 +08:00 · 2025-10-16 21:46:04 +00:00
parent 76ae0a62ac
commit 59da578902
24 changed files with 3707 additions and 5 deletions
@@ -0,0 +1,68 @@
+# Copyright (c) Microsoft. All rights reserved.
+"""Shared pytest fixtures for Purview tests."""
+
+import pytest
+
+from agent_framework_purview._models import (
+    Activity,
+    ActivityMetadata,
+    ContentToProcess,
+    DeviceMetadata,
+    IntegratedAppMetadata,
+    OperatingSystemSpecifications,
+    PolicyLocation,
+    ProcessContentRequest,
+    ProcessConversationMetadata,
+    ProtectedAppMetadata,
+    PurviewTextContent,
+)
+
+
+@pytest.fixture
+def content_to_process_factory():
+    """Factory fixture to create ContentToProcess objects with test data."""
+
+    def _create_content(text: str = "Test") -> ContentToProcess:
+        text_content = PurviewTextContent(data=text)
+        metadata = ProcessConversationMetadata(
+            identifier="msg-1",
+            content=text_content,
+            name="Test",
+            is_truncated=False,
+        )
+        activity_meta = ActivityMetadata(activity=Activity.UPLOAD_TEXT)
+        device_meta = DeviceMetadata(
+            operating_system_specifications=OperatingSystemSpecifications(
+                operating_system_platform="Windows", operating_system_version="10"
+            )
+        )
+        integrated_app = IntegratedAppMetadata(name="App", version="1.0")
+        location = PolicyLocation(data_type="microsoft.graph.policyLocationApplication", value="app-id")
+        protected_app = ProtectedAppMetadata(name="Protected", version="1.0", application_location=location)
+
+        return ContentToProcess(
+            content_entries=[metadata],
+            activity_metadata=activity_meta,
+            device_metadata=device_meta,
+            integrated_app_metadata=integrated_app,
+            protected_app_metadata=protected_app,
+        )
+
+    return _create_content
+
+
+@pytest.fixture
+def process_content_request_factory(content_to_process_factory):
+    """Factory fixture to create ProcessContentRequest objects with test data."""
+
+    def _create_request(
+        text: str = "Test", user_id: str = "user-123", tenant_id: str = "tenant-456"
+    ) -> ProcessContentRequest:
+        content = content_to_process_factory(text)
+        return ProcessContentRequest(
+            content_to_process=content,
+            user_id=user_id,
+            tenant_id=tenant_id,
+        )
+
+    return _create_request