From 9e9160304cbba52c2224c022cb42f612406823f7 Mon Sep 17 00:00:00 2001
From: Ilija Lazoroski <ilija.la@live.com>
Date: Fri, 9 Sep 2022 17:11:01 +0200
Subject: [PATCH] UT: Add unit tests for PydanticEventSerializer

---
 .../test_pydantic_event_serializer.py         | 50 +++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 monkey/tests/unit_tests/common/event_serializers/test_pydantic_event_serializer.py

diff --git a/monkey/tests/unit_tests/common/event_serializers/test_pydantic_event_serializer.py b/monkey/tests/unit_tests/common/event_serializers/test_pydantic_event_serializer.py
new file mode 100644
index 000000000..ce38f2680
--- /dev/null
+++ b/monkey/tests/unit_tests/common/event_serializers/test_pydantic_event_serializer.py
@@ -0,0 +1,50 @@
+from abc import ABC
+from dataclasses import dataclass, field
+
+import pytest
+from pydantic import ValidationError
+
+from common.base_models import InfectionMonkeyBaseModel
+from common.event_serializers import PydanticEventSerializer
+from common.events import AbstractAgentEvent
+
+
+@dataclass(frozen=True)
+class NotAgentEvent(ABC):
+    some_field: int
+    other_field: float
+
+
+@dataclass(frozen=True)
+class SomeAgentEvent(AbstractAgentEvent):
+    bogus: int = field(default_factory=int)
+
+
+class PydanticEvent(InfectionMonkeyBaseModel):
+    some_field: str
+
+
+@pytest.fixture
+def pydantic_event_serializer():
+    return PydanticEventSerializer(PydanticEvent)
+
+
+@pytest.mark.parametrize("event", [NotAgentEvent(1, 2.0), SomeAgentEvent(2)])
+def test_pydantic_event_serializer__serialize_wrong_type(pydantic_event_serializer, event):
+    with pytest.raises(TypeError):
+        pydantic_event_serializer.serialize(event)
+
+
+def test_pydantic_event_serializer__deserialize_wrong_type(pydantic_event_serializer):
+    with pytest.raises(ValidationError):
+        pydantic_event_serializer.deserialize("bla")
+
+
+def test_pydanitc_event_serializer__de_serialize(pydantic_event_serializer):
+    pydantic_event = PydanticEvent(some_field="some_field")
+
+    serialized_event = pydantic_event_serializer.serialize(pydantic_event)
+    deserialized_object = pydantic_event_serializer.deserialize(serialized_event)
+
+    assert type(serialized_event) != type(deserialized_object)
+    assert deserialized_object == pydantic_event