LCORE-2400: Unit test for dump_schema function

Author: tisnik

tests/unit/utils/test_schema_dumper.py

@@ -1,8 +1,10 @@
 """Unit tests for utils/schema_dumper module."""
 
+from json import load
+from pathlib import Path
 from typing import Any
 
-from utils.schema_dumper import recursive_update
+from utils.schema_dumper import dump_schema, recursive_update
 
 
 def test_update_empty_input() -> None:
@@ -409,3 +411,110 @@ def test_anyof_with_additional_fields_more_items() -> None:
 
     # non-empty dict with known content should be returned
     assert result == expected
+
+
+def test_dump_schema(tmpdir: Path) -> None:
+    """Test that schema can be dump into a JSON file.
+
+    An example of schema dump:
+    {
+        "openapi": "3.0.0",
+        "info": {
+            "title": "Lightspeed Core Stack",
+            "version": "0.3.0"
+        },
+        "components": {
+            "schemas": {
+                "A2AStateConfiguration": {
+                    "additionalProperties": false,
+                    "description": "xyzzy",
+                    "properties": {
+                        "sqlite": {
+                            "anyOf": [
+                                {
+                                    "$ref": "#/components/schemas/SQLiteDatabaseConfiguration"
+                                },
+                                {
+                                    "type": "null"
+                                }
+                            ],
+                            "default": null,
+                            "description": "SQLite database configuration for A2A state storage.",
+                            "title": "SQLite configuration"
+                        },
+                    ...
+                }
+                ...
+                ...
+                ...
+        },
+        "paths": {}
+    }
+    """
+    filename = tmpdir / "foo.json"
+    dump_schema(str(filename))
+
+    with open(filename, "r", encoding="utf-8") as fin:
+        # schema should be stored in JSON format
+        content = load(fin)
+        assert content is not None
+
+        # top-level keys test
+        keys = ("openapi", "info", "components", "paths")
+        for key in keys:
+            assert key in content
+
+        # components should be top-level node
+        components = content["components"]
+        assert components is not None
+
+        # schemas should be a node stored inside components node
+        assert "schemas" in components
+        schemas = components["schemas"]
+        assert schemas is not None
+
+        # list of schemas expected in a dump
+        expected_schemas = (
+            "A2AStateConfiguration",
+            "APIKeyTokenConfiguration",
+            "AccessRule",
+            "Action",
+            "ApprovalFilter",
+            "ApprovalsConfiguration",
+            "AuthenticationConfiguration",
+            "AuthorizationConfiguration",
+            "AzureEntraIdConfiguration",
+            "ByokRag",
+            "CORSConfiguration",
+            "CompactionConfiguration",
+            "Configuration",
+            "ConversationHistoryConfiguration",
+            "CustomProfile",
+            "Customization",
+            "DatabaseConfiguration",
+            "InMemoryCacheConfig",
+            "InferenceConfiguration",
+            "JsonPathOperator",
+            "JwkConfiguration",
+            "JwtConfiguration",
+            "JwtRoleRule",
+            "LlamaStackConfiguration",
+            "ModelContextProtocolServer",
+            "OkpConfiguration",
+            "PostgreSQLDatabaseConfiguration",
+            "QuotaHandlersConfiguration",
+            "QuotaLimiterConfiguration",
+            "QuotaSchedulerConfiguration",
+            "RHIdentityConfiguration",
+            "RagConfiguration",
+            "RerankerConfiguration",
+            "RlsapiV1Configuration",
+            "SQLiteDatabaseConfiguration",
+            "ServiceConfiguration",
+            "SkillsConfiguration",
+            "SplunkConfiguration",
+            "TLSConfiguration",
+            "UserDataCollection",
+        )
+        for expected_schema in expected_schemas:
+            assert expected_schema in schemas