fix unit test

Author: rbiseck3

test/test_schema.py

@@ -449,22 +449,16 @@ def fn(a: FileData) -> list[FileData]:
                     "identifier": {"type": "string"},
                     "connector_type": {"type": "string"},
                     "source_identifiers": {
-                        "anyOf": [
-                            {
-                                "type": "object",
-                                "properties": {
-                                    "filename": {"type": "string"},
-                                    "fullpath": {"type": "string"},
-                                    "rel_path": {
-                                        "anyOf": [{"type": "string"}, {"type": "null"}],
-                                        "default": None,
-                                    },
-                                },
-                                "required": ["filename", "fullpath"],
+                        "type": "object",
+                        "properties": {
+                            "filename": {"type": "string"},
+                            "fullpath": {"type": "string"},
+                            "rel_path": {
+                                "anyOf": [{"type": "string"}, {"type": "null"}],
+                                "default": None,
                             },
-                            {"type": "null"},
-                        ],
-                        "default": None,
+                        },
+                        "required": ["filename", "fullpath"],
                     },
                     "metadata": {
                         "type": "object",
@@ -533,7 +527,13 @@ def fn(a: FileData) -> list[FileData]:
                         "default": None,
                     },
                 },
-                "required": ["identifier", "connector_type", "metadata", "additional_metadata"],
+                "required": [
+                    "identifier",
+                    "connector_type",
+                    "source_identifiers",
+                    "metadata",
+                    "additional_metadata",
+                ],
             }
         },
     }
@@ -551,22 +551,16 @@ def fn(a: FileData) -> list[FileData]:
                 "identifier": {"type": "string"},
                 "connector_type": {"type": "string"},
                 "source_identifiers": {
-                    "anyOf": [
-                        {
-                            "type": "object",
-                            "properties": {
-                                "filename": {"type": "string"},
-                                "fullpath": {"type": "string"},
-                                "rel_path": {
-                                    "anyOf": [{"type": "string"}, {"type": "null"}],
-                                    "default": None,
-                                },
-                            },
-                            "required": ["filename", "fullpath"],
+                    "type": "object",
+                    "properties": {
+                        "filename": {"type": "string"},
+                        "fullpath": {"type": "string"},
+                        "rel_path": {
+                            "anyOf": [{"type": "string"}, {"type": "null"}],
+                            "default": None,
                         },
-                        {"type": "null"},
-                    ],
-                    "default": None,
+                    },
+                    "required": ["filename", "fullpath"],
                 },
                 "metadata": {
                     "type": "object",
@@ -629,7 +623,13 @@ def fn(a: FileData) -> list[FileData]:
                 },
                 "display_name": {"anyOf": [{"type": "string"}, {"type": "null"}], "default": None},
             },
-            "required": ["identifier", "connector_type", "metadata", "additional_metadata"],
+            "required": [
+                "identifier",
+                "connector_type",
+                "source_identifiers",
+                "metadata",
+                "additional_metadata",
+            ],
         },
     }
 

test/test_utils.py

@@ -4,7 +4,7 @@
 
 import pytest
 from pydantic import BaseModel
-from unstructured_ingest.v2.interfaces import FileData
+from unstructured_ingest.v2.interfaces import FileData, SourceIdentifiers
 from uvicorn.importer import import_from_string
 
 from unstructured_platform_plugins.etl_uvicorn import utils
@@ -110,6 +110,7 @@ def fn(a: A, b: B, c: MyEnum, d: list, e: FileData) -> None:
         identifier="custom_file_data",
         connector_type="mock_connector",
         additional_metadata={"additional": "metadata"},
+        source_identifiers=SourceIdentifiers(filename="file.txt", fullpath="file.txt"),
     )
     inputs = {
         "a": {"b": 4, "c": 5.6},