chore: import upstream snapshot with attribution

python/tests/unit/agents/runtime/test_message_serialization.py

@@ -0,0 +1,142 @@
+# Copyright (c) Microsoft. All rights reserved.
+
+from dataclasses import dataclass
+
+import pytest
+from pydantic import BaseModel
+
+from semantic_kernel.agents.runtime.core.serialization import (
+    JSON_DATA_CONTENT_TYPE,
+    DataclassJsonMessageSerializer,
+    MessageSerializer,
+    SerializationRegistry,
+    try_get_known_serializers_for_type,
+)
+
+
+class PydanticMessage(BaseModel):
+    message: str
+
+
+class NestingPydanticMessage(BaseModel):
+    message: str
+    nested: PydanticMessage
+
+
+@dataclass
+class DataclassMessage:
+    message: str
+
+
+@dataclass
+class NestingDataclassMessage:
+    message: str
+    nested: DataclassMessage
+
+
+@dataclass
+class NestingPydanticDataclassMessage:
+    message: str
+    nested: PydanticMessage
+
+
+def test_pydantic() -> None:
+    serde = SerializationRegistry()
+    serde.add_serializer(try_get_known_serializers_for_type(PydanticMessage))
+
+    message = PydanticMessage(message="hello")
+    name = serde.type_name(message)
+    json = serde.serialize(message, type_name=name, data_content_type=JSON_DATA_CONTENT_TYPE)
+    assert name == "PydanticMessage"
+    assert json == b'{"message":"hello"}'
+    deserialized = serde.deserialize(json, type_name=name, data_content_type=JSON_DATA_CONTENT_TYPE)
+    assert deserialized == message
+
+
+def test_nested_pydantic() -> None:
+    serde = SerializationRegistry()
+    serde.add_serializer(try_get_known_serializers_for_type(NestingPydanticMessage))
+
+    message = NestingPydanticMessage(message="hello", nested=PydanticMessage(message="world"))
+    name = serde.type_name(message)
+    json = serde.serialize(message, type_name=name, data_content_type=JSON_DATA_CONTENT_TYPE)
+    assert json == b'{"message":"hello","nested":{"message":"world"}}'
+    deserialized = serde.deserialize(json, type_name=name, data_content_type=JSON_DATA_CONTENT_TYPE)
+    assert deserialized == message
+
+
+def test_dataclass() -> None:
+    serde = SerializationRegistry()
+    serde.add_serializer(try_get_known_serializers_for_type(DataclassMessage))
+
+    message = DataclassMessage(message="hello")
+    name = serde.type_name(message)
+    json = serde.serialize(message, type_name=name, data_content_type=JSON_DATA_CONTENT_TYPE)
+    assert json == b'{"message": "hello"}'
+    deserialized = serde.deserialize(json, type_name=name, data_content_type=JSON_DATA_CONTENT_TYPE)
+    assert deserialized == message
+
+
+def test_nesting_dataclass_dataclass() -> None:
+    serde = SerializationRegistry()
+    with pytest.raises(ValueError):
+        serde.add_serializer(try_get_known_serializers_for_type(NestingDataclassMessage))
+
+
+@dataclass
+class DataclassNestedUnionSyntaxOldMessage:
+    message: str | int
+
+
+@dataclass
+class DataclassNestedUnionSyntaxNewMessage:
+    message: str | int
+
+
+@pytest.mark.parametrize("cls", [DataclassNestedUnionSyntaxOldMessage, DataclassNestedUnionSyntaxNewMessage])
+def test_nesting_union_old_syntax_dataclass(
+    cls: type[DataclassNestedUnionSyntaxOldMessage | DataclassNestedUnionSyntaxNewMessage],
+) -> None:
+    with pytest.raises(ValueError):
+        _serializer = DataclassJsonMessageSerializer(cls)
+
+
+def test_nesting_dataclass_pydantic() -> None:
+    serde = SerializationRegistry()
+    with pytest.raises(ValueError):
+        serde.add_serializer(try_get_known_serializers_for_type(NestingPydanticDataclassMessage))
+
+
+def test_invalid_type() -> None:
+    serde = SerializationRegistry()
+    try:
+        serde.add_serializer(try_get_known_serializers_for_type(str))
+    except ValueError as e:
+        assert str(e) == "Unsupported type <class 'str'>"
+
+
+def test_custom_type() -> None:
+    serde = SerializationRegistry()
+
+    class CustomStringTypeSerializer(MessageSerializer[str]):
+        @property
+        def data_content_type(self) -> str:
+            return "str"
+
+        @property
+        def type_name(self) -> str:
+            return "custom_str"
+
+        def deserialize(self, payload: bytes) -> str:
+            message = payload.decode("utf-8")
+            return message[1:-1]
+
+        def serialize(self, message: str) -> bytes:
+            return f'"{message}"'.encode()
+
+    serde.add_serializer(CustomStringTypeSerializer())
+    message = "hello"
+    json = serde.serialize(message, type_name="custom_str", data_content_type="str")
+    assert json == b'"hello"'
+    deserialized = serde.deserialize(json, type_name="custom_str", data_content_type="str")
+    assert deserialized == message