update unit tests

Author: rbiseck3

test/test_schema.py

@@ -560,18 +560,72 @@ def fn(a: FileData) -> list[FileData]:
                             },
                         },
                         "required": [],
+                        "default": {
+                            "type": "object",
+                            "properties": {
+                                "url": {
+                                    "anyOf": [{"type": "string"}, {"type": "null"}],
+                                    "default": None,
+                                },
+                                "version": {
+                                    "anyOf": [{"type": "string"}, {"type": "null"}],
+                                    "default": None,
+                                },
+                                "record_locator": {
+                                    "anyOf": [
+                                        {
+                                            "type": "object",
+                                            "items": {"key": {"type": "string"}, "value": {}},
+                                        },
+                                        {"type": "null"},
+                                    ],
+                                    "default": None,
+                                },
+                                "date_created": {
+                                    "anyOf": [{"type": "string"}, {"type": "null"}],
+                                    "default": None,
+                                },
+                                "date_modified": {
+                                    "anyOf": [{"type": "string"}, {"type": "null"}],
+                                    "default": None,
+                                },
+                                "date_processed": {
+                                    "anyOf": [{"type": "string"}, {"type": "null"}],
+                                    "default": None,
+                                },
+                                "permissions_data": {
+                                    "anyOf": [
+                                        {
+                                            "type": "array",
+                                            "items": {
+                                                "type": "object",
+                                                "items": {"key": {"type": "string"}, "value": {}},
+                                            },
+                                        },
+                                        {"type": "null"},
+                                    ],
+                                    "default": None,
+                                },
+                                "filesize_bytes": {
+                                    "anyOf": [{"type": "integer"}, {"type": "null"}],
+                                    "default": None,
+                                },
+                            },
+                            "required": [],
+                        },
                     },
                     "additional_metadata": {
                         "type": "object",
                         "items": {"key": {"type": "string"}, "value": {}},
+                        "default": {},
                     },
                     "reprocess": {"type": "boolean", "default": False},
                     "local_download_path": {
                         "anyOf": [{"type": "string"}, {"type": "null"}],
                         "default": None,
                     },
                 },
-                "required": ["identifier", "connector_type", "metadata", "additional_metadata"],
+                "required": ["identifier", "connector_type"],
             }
         },
     }
@@ -657,20 +711,73 @@ def fn(a: FileData) -> list[FileData]:
                         },
                     },
                     "required": [],
+                    "default": {
+                        "type": "object",
+                        "properties": {
+                            "url": {
+                                "anyOf": [{"type": "string"}, {"type": "null"}],
+                                "default": None,
+                            },
+                            "version": {
+                                "anyOf": [{"type": "string"}, {"type": "null"}],
+                                "default": None,
+                            },
+                            "record_locator": {
+                                "anyOf": [
+                                    {
+                                        "type": "object",
+                                        "items": {"key": {"type": "string"}, "value": {}},
+                                    },
+                                    {"type": "null"},
+                                ],
+                                "default": None,
+                            },
+                            "date_created": {
+                                "anyOf": [{"type": "string"}, {"type": "null"}],
+                                "default": None,
+                            },
+                            "date_modified": {
+                                "anyOf": [{"type": "string"}, {"type": "null"}],
+                                "default": None,
+                            },
+                            "date_processed": {
+                                "anyOf": [{"type": "string"}, {"type": "null"}],
+                                "default": None,
+                            },
+                            "permissions_data": {
+                                "anyOf": [
+                                    {
+                                        "type": "array",
+                                        "items": {
+                                            "type": "object",
+                                            "items": {"key": {"type": "string"}, "value": {}},
+                                        },
+                                    },
+                                    {"type": "null"},
+                                ],
+                                "default": None,
+                            },
+                            "filesize_bytes": {
+                                "anyOf": [{"type": "integer"}, {"type": "null"}],
+                                "default": None,
+                            },
+                        },
+                        "required": [],
+                    },
                 },
                 "additional_metadata": {
                     "type": "object",
                     "items": {"key": {"type": "string"}, "value": {}},
+                    "default": {},
                 },
                 "reprocess": {"type": "boolean", "default": False},
                 "local_download_path": {
                     "anyOf": [{"type": "string"}, {"type": "null"}],
                     "default": None,
                 },
             },
-            "required": ["identifier", "connector_type", "metadata", "additional_metadata"],
+            "required": ["identifier", "connector_type"],
         },
     }
-
     assert output_schema == expected_output_schema
     assert is_valid_response_dict(output_schema)

test/test_utils.py

@@ -136,5 +136,5 @@ def fn(a: FileData) -> None:
 
     inputs = {"a": {"not": "the", "right": "values"}}
 
-    with pytest.raises(KeyError):
+    with pytest.raises(ValueError):
         utils.map_inputs(func=fn, raw_inputs=inputs)

unstructured_platform_plugins/etl_uvicorn/utils.py

@@ -112,7 +112,9 @@ def map_inputs(func: Callable, raw_inputs: dict[str, Any]) -> dict[str, Any]:
                 and issubclass(type_data, BaseModel)
             ):
                 field_value = raw_inputs[field_name]
-                raw_inputs[field_name] = type_data.model_validate(field_value.model_dump())
+                if isinstance(field_value, BaseModel):
+                    field_value = field_value.model_dump()
+                raw_inputs[field_name] = type_data.model_validate(field_value)
         except Exception as e:
             raise ValueError(f"failed to map input for field {field_name}: {field_value}") from e
     return raw_inputs