p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

UT: Use IAgentEventSerializer in place of IEventSerializer

This commit is contained in:
Shreya Malviya 2022-09-16 12:19:24 +05:30 committed by Mike Salvatore
parent 0e2b165c15
commit 619dc68c0e
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/tests/unit_tests/common/event_serializers/test_event_serializer_registry.py
View File

@ -3,7 +3,7 @@ from unittest.mock import MagicMock
import pytest import pytest
from pydantic import Field from pydantic import Field
from common.event_serializers import EventSerializerRegistry, IEventSerializer from common.event_serializers import EventSerializerRegistry, IAgentEventSerializer
from common.events import AbstractAgentEvent from common.events import AbstractAgentEvent
@ -19,8 +19,8 @@ class NoneEvent(AbstractAgentEvent):
none_param: float = Field(default=1.0) none_param: float = Field(default=1.0)
SOME_SERIALIZER = MagicMock(spec=IEventSerializer) SOME_SERIALIZER = MagicMock(spec=IAgentEventSerializer)
OTHER_SERIALIZER = MagicMock(spec=IEventSerializer) OTHER_SERIALIZER = MagicMock(spec=IAgentEventSerializer)
@pytest.fixture @pytest.fixture
@ -50,7 +50,7 @@ def test_event_serializer_registry_set_unsupported_type(event_serializer_registr
def test_event_serializer_registry_set_unsupported_type_key(event_serializer_registry): def test_event_serializer_registry_set_unsupported_type_key(event_serializer_registry):
with pytest.raises(TypeError): with pytest.raises(TypeError):
event_serializer_registry["BogusKey"] = MagicMock(spec=IEventSerializer) event_serializer_registry["BogusKey"] = MagicMock(spec=IAgentEventSerializer)
def test_event_serializer_registry_get_unsuported_type(event_serializer_registry): def test_event_serializer_registry_get_unsuported_type(event_serializer_registry):

8
monkey/tests/unit_tests/common/event_serializers/test_pydantic_event_serializer.py
View File

@ -5,7 +5,11 @@ from uuid import UUID
import pytest import pytest
from pydantic import Field from pydantic import Field
from common.event_serializers import EVENT_TYPE_FIELD, IEventSerializer, PydanticEventSerializer from common.event_serializers import (
EVENT_TYPE_FIELD,
IAgentEventSerializer,
PydanticEventSerializer,
)
from common.events import AbstractAgentEvent from common.events import AbstractAgentEvent
AGENT_ID = UUID("f811ad00-5a68-4437-bd51-7b5cc1768ad5") AGENT_ID = UUID("f811ad00-5a68-4437-bd51-7b5cc1768ad5")
@ -26,7 +30,7 @@ class PydanticEvent(AbstractAgentEvent):
@pytest.fixture @pytest.fixture
def pydantic_event_serializer() -> IEventSerializer: def pydantic_event_serializer() -> IAgentEventSerializer:
return PydanticEventSerializer(PydanticEvent) return PydanticEventSerializer(PydanticEvent)