re-apply test changes

Author: jordan-homan

_test_contract/platform_api/test_jobs.py

@@ -165,3 +165,60 @@ def test_cancel_job(httpx_mock, platform_client: UnstructuredClient, platform_ap
     request = requests[0]
     assert request.method == "POST"
     assert request.url == url
+
+
+def test_create_job(httpx_mock, platform_client: UnstructuredClient, platform_api_url: str):
+    import json
+
+    url = f"{platform_api_url}/api/v1/jobs/"
+
+    httpx_mock.add_response(
+        method="POST",
+        status_code=200,
+        headers={"Content-Type": "application/json"},
+        json={
+            "created_at": "2025-06-22T11:37:21.648Z",
+            "id": "fcdc4994-eea5-425c-91fa-e03f2bd8030d",
+            "status": "SCHEDULED",
+            "runtime": None,
+            "workflow_id": "16b80fee-64dc-472d-8f26-1d7729b6423d",
+            "workflow_name": "job-fcdc4994",
+            "input_file_ids": ["upload-test-file-123"],
+            "output_node_files": [
+                {
+                    "node_id": "93fc2ce8-e7c8-424f-a6aa-41460fc5d35d",
+                    "file_id": "upload-test-file-123",
+                    "node_type": "partition",
+                    "node_subtype": "unstructured_api",
+                }
+            ],
+            "job_type": "template",
+        },
+        url=url,
+    )
+
+    # request_data should be a JSON string containing the job creation data
+    request_data = json.dumps({
+        "template_id": "hi_res_partition",
+    })
+
+    create_job_response = platform_client.jobs.create_job(
+        request=operations.CreateJobRequest(
+            body_create_job=shared.BodyCreateJob(
+                request_data=request_data,
+            )
+        )
+    )
+    assert create_job_response.status_code == 200
+
+    requests = httpx_mock.get_requests()
+    assert len(requests) == 1
+    request = requests[0]
+    assert request.method == "POST"
+    assert request.url == url
+
+    job = create_job_response.job_information
+    assert job.id == "fcdc4994-eea5-425c-91fa-e03f2bd8030d"
+    assert job.status == "SCHEDULED"
+    assert job.job_type == "template"
+    assert job.created_at == datetime.fromisoformat("2025-06-22T11:37:21.648+00:00")

_test_contract/platform_api/test_templates.py

@@ -0,0 +1,141 @@
+from datetime import datetime
+
+import pytest
+
+from unstructured_client import UnstructuredClient
+from unstructured_client.models import operations
+from unstructured_client.models.errors import SDKError
+
+
+def test_list_templates(httpx_mock, platform_client: UnstructuredClient, platform_api_url: str):
+    url = f"{platform_api_url}/api/v1/templates/"
+
+    httpx_mock.add_response(
+        method="GET",
+        headers={"Content-Type": "application/json"},
+        json=[
+            {
+                "id": "hi_res_partition",
+                "name": "High Resolution Partition",
+                "description": "Partition documents with high resolution strategy",
+                "version": "1.0.0",
+                "nodes": [
+                    {
+                        "id": "93fc2ce8-e7c8-424f-a6aa-41460fc5d35d",
+                        "name": "partition step",
+                        "type": "partition",
+                        "subtype": "unstructured_api",
+                    }
+                ],
+                "edges": [],
+            },
+            {
+                "id": "hi_res_and_enrichment",
+                "name": "High Resolution and Enrichment",
+                "description": "Partition with enrichment",
+                "version": "1.0.0",
+                "nodes": [],
+                "edges": [],
+            },
+        ],
+        url=url,
+    )
+
+    templates_response = platform_client.templates.list_templates(
+        request=operations.ListTemplatesRequest()
+    )
+    assert templates_response.status_code == 200
+
+    requests = httpx_mock.get_requests()
+    assert len(requests) == 1
+    request = requests[0]
+    assert request.method == "GET"
+    assert request.url == url
+
+    assert "templates" in templates_response.response_list_templates
+    templates = templates_response.response_list_templates
+    assert len(templates) == 2
+    assert templates[0]["id"] == "hi_res_partition"
+    assert templates[0]["name"] == "High Resolution Partition"
+    assert templates[1]["id"] == "hi_res_and_enrichment"
+
+
+def test_get_template(httpx_mock, platform_client: UnstructuredClient, platform_api_url: str):
+    url = f"{platform_api_url}/api/v1/templates/hi_res_partition"
+
+    httpx_mock.add_response(
+        method="GET",
+        headers={"Content-Type": "application/json"},
+        json={
+            "id": "hi_res_partition",
+            "name": "High Resolution Partition",
+            "description": "Partition documents with high resolution strategy",
+            "version": "1.0.0",
+            "nodes": [
+                {
+                    "id": "93fc2ce8-e7c8-424f-a6aa-41460fc5d35d",
+                    "name": "partition step",
+                    "type": "partition",
+                    "subtype": "unstructured_api",
+                    "settings": {
+                        "strategy": "fast",
+                        "include_page_breaks": False,
+                    },
+                }
+            ],
+            "edges": [
+                {
+                    "source_id": "00000000-0000-0000-0000-000000000001-downloader",
+                    "destination_id": "93fc2ce8-e7c8-424f-a6aa-41460fc5d35d",
+                }
+            ],
+        },
+        url=url,
+    )
+
+    template_response = platform_client.templates.get_template(
+        request=operations.GetTemplateRequest(template_id="hi_res_partition")
+    )
+    assert template_response.status_code == 200
+
+    requests = httpx_mock.get_requests()
+    assert len(requests) == 1
+    request = requests[0]
+    assert request.method == "GET"
+    assert request.url == url
+
+    assert "template" in template_response.response_get_template
+    template = template_response.response_get_template
+    assert template["id"] == "hi_res_partition"
+    assert template["name"] == "High Resolution Partition"
+    assert len(template["nodes"]) == 1
+    assert template["nodes"][0]["id"] == "93fc2ce8-e7c8-424f-a6aa-41460fc5d35d"
+
+
+def test_get_template_not_found(
+    httpx_mock, platform_client: UnstructuredClient, platform_api_url: str
+):
+    url = f"{platform_api_url}/api/v1/templates/nonexistent_template"
+
+    httpx_mock.add_response(
+        method="GET",
+        status_code=404,
+        headers={"Content-Type": "application/json"},
+        json={"detail": "Template nonexistent_template not found"},
+        url=url,
+    )
+
+    with pytest.raises(SDKError) as e:
+        platform_client.templates.get_template(
+            request=operations.GetTemplateRequest(template_id="nonexistent_template")
+        )
+
+    assert e.value.status_code == 404
+    assert "API error occurred" in e.value.message
+
+    requests = httpx_mock.get_requests()
+    assert len(requests) == 1
+    request = requests[0]
+    assert request.method == "GET"
+    assert request.url == url
+