fix: register event schema

Author: atstaeff

eventsourcingdb/handlers/register_event_schema/register_event_schema.py

@@ -1,5 +1,6 @@
 import json
 from http import HTTPStatus
+from typing import Union, Dict, Any
 
 from ...abstract_base_client import AbstractBaseClient
 from ...errors.custom_error import CustomError
@@ -14,7 +15,7 @@
 async def register_event_schema(
     client: AbstractBaseClient,
     event_type: str,
-    json_schema: str,
+    json_schema: Union[str, Dict[str, Any]],
 ) -> None:
     try:
         validate_type(event_type)
@@ -24,10 +25,22 @@ async def register_event_schema(
         ) from validation_error
     except Exception as other_error:
         raise InternalError(str(other_error)) from other_error
+    
+    # Handle both string and dictionary schema formats
+    # If json_schema is a string, parse it to ensure it's valid JSON
+    # If it's already a dict, use it directly
+    schema_obj = json_schema
+    if isinstance(json_schema, str):
+        try:
+            schema_obj = json.loads(json_schema)
+        except json.JSONDecodeError as json_error:
+            raise InvalidParameterError(
+                'json_schema', f'Invalid JSON schema: {str(json_error)}'
+            ) from json_error
 
     request_body = json.dumps({
         'eventType': event_type,
-        'schema': json_schema,
+        'schema': schema_obj,
     })
 
     response: Response

tests/test_write_events.py

@@ -202,7 +202,7 @@ async def test_throws_error_if_event_does_not_match_schema(
 
         await client.register_event_schema(
             "com.super.duper",
-            '{"type":"object","additionalProperties":false}'
+            {"type": "object", "additionalProperties": False}
         )
 
         with pytest.raises(ClientError, match="event candidate does not match schema"):