From 2f748516a827a5f9471d7c540d9b5c187422dc0f Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Fri, 16 Sep 2022 12:48:01 +0530 Subject: [PATCH] UT: Change pydantic_event_serializer -> pydantic_agent_event_serializer in test_pydantic_agent_event_serializer.py --- .../test_pydantic_agent_event_serializer.py | 22 ++++++++++--------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/monkey/tests/unit_tests/common/event_serializers/test_pydantic_agent_event_serializer.py b/monkey/tests/unit_tests/common/event_serializers/test_pydantic_agent_event_serializer.py index eaa54eb02..31c3080ee 100644 --- a/monkey/tests/unit_tests/common/event_serializers/test_pydantic_agent_event_serializer.py +++ b/monkey/tests/unit_tests/common/event_serializers/test_pydantic_agent_event_serializer.py @@ -30,7 +30,7 @@ class PydanticEvent(AbstractAgentEvent): @pytest.fixture -def pydantic_event_serializer() -> IAgentEventSerializer: +def pydantic_agent_event_serializer() -> IAgentEventSerializer: return PydanticAgentEventSerializer(PydanticEvent) @@ -38,28 +38,30 @@ def pydantic_event_serializer() -> IAgentEventSerializer: "event", [NotAgentEvent(some_field=1, other_field=2.0), SomeAgentEvent(source=AGENT_ID, bogus=2)], ) -def test_pydantic_event_serializer__serialize_wrong_type(pydantic_event_serializer, event): +def test_pydantic_agent_event_serializer__serialize_wrong_type( + pydantic_agent_event_serializer, event +): with pytest.raises(TypeError): - pydantic_event_serializer.serialize(event) + pydantic_agent_event_serializer.serialize(event) -def test_pydantic_event_serializer__deserialize_wrong_type(pydantic_event_serializer): +def test_pydantic_agent_event_serializer__deserialize_wrong_type(pydantic_agent_event_serializer): with pytest.raises(TypeError): - pydantic_event_serializer.deserialize("bla") + pydantic_agent_event_serializer.deserialize("bla") -def test_pydanitc_event_serializer__de_serialize(pydantic_event_serializer): +def test_pydantic_agent_event_serializer__de_serialize(pydantic_agent_event_serializer): pydantic_event = PydanticEvent(source=AGENT_ID, some_field="some_field") - serialized_event = pydantic_event_serializer.serialize(pydantic_event) - deserialized_object = pydantic_event_serializer.deserialize(serialized_event) + serialized_event = pydantic_agent_event_serializer.serialize(pydantic_event) + deserialized_object = pydantic_agent_event_serializer.deserialize(serialized_event) assert type(serialized_event) != type(deserialized_object) assert deserialized_object == pydantic_event -def test_pydanitc_event_serializer__serialize_inclued_type(pydantic_event_serializer): +def test_pydantic_event_serializer__serialize_inclued_type(pydantic_agent_event_serializer): pydantic_event = PydanticEvent(source=AGENT_ID, some_field="some_field") - serialized_event = pydantic_event_serializer.serialize(pydantic_event) + serialized_event = pydantic_agent_event_serializer.serialize(pydantic_event) assert serialized_event[EVENT_TYPE_FIELD] == PydanticEvent.__name__