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

UT: Use AgentEventSerializerRegistry in place of EventSerializerRegistry

This commit is contained in:
Shreya Malviya 2022-09-16 12:28:47 +05:30 committed by Mike Salvatore
parent 4a23f5a95b
commit 2568d52730
3 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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, IAgentEventSerializer from common.event_serializers import AgentEventSerializerRegistry, IAgentEventSerializer
from common.events import AbstractAgentEvent from common.events import AbstractAgentEvent
@ -25,7 +25,7 @@ OTHER_SERIALIZER = MagicMock(spec=IAgentEventSerializer)
@pytest.fixture @pytest.fixture
def event_serializer_registry(): def event_serializer_registry():
event_serializer_registry = EventSerializerRegistry() event_serializer_registry = AgentEventSerializerRegistry()
event_serializer_registry[SomeEvent] = SOME_SERIALIZER event_serializer_registry[SomeEvent] = SOME_SERIALIZER
event_serializer_registry[OtherEvent] = OTHER_SERIALIZER event_serializer_registry[OtherEvent] = OTHER_SERIALIZER

6
monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_event_repository.py
View File

@ -6,7 +6,7 @@ import mongomock
import pytest import pytest
from pydantic import Field from pydantic import Field
from common.event_serializers import EventSerializerRegistry, PydanticEventSerializer from common.event_serializers import AgentEventSerializerRegistry, PydanticEventSerializer
from common.events import AbstractAgentEvent from common.events import AbstractAgentEvent
from monkey_island.cc.repository import ( from monkey_island.cc.repository import (
IEventRepository, IEventRepository,
@ -34,8 +34,8 @@ EVENTS: List[AbstractAgentEvent] = [
@pytest.fixture @pytest.fixture
def event_serializer_registry() -> EventSerializerRegistry: def event_serializer_registry() -> AgentEventSerializerRegistry:
registry = EventSerializerRegistry() registry = AgentEventSerializerRegistry()
registry[FakeAgentEvent] = PydanticEventSerializer(FakeAgentEvent) registry[FakeAgentEvent] = PydanticEventSerializer(FakeAgentEvent)
registry[FakeAgentItemEvent] = PydanticEventSerializer(FakeAgentItemEvent) registry[FakeAgentItemEvent] = PydanticEventSerializer(FakeAgentItemEvent)
return registry return registry

8
monkey/tests/unit_tests/monkey_island/cc/resources/test_events.py
View File

@ -6,7 +6,7 @@ import pytest
from tests.common import StubDIContainer from tests.common import StubDIContainer
from common.event_queue import IAgentEventQueue from common.event_queue import IAgentEventQueue
from common.event_serializers import EventSerializerRegistry, PydanticEventSerializer from common.event_serializers import AgentEventSerializerRegistry, PydanticEventSerializer
from common.events import AbstractAgentEvent from common.events import AbstractAgentEvent
from monkey_island.cc.resources import Events from monkey_island.cc.resources import Events
@ -88,8 +88,8 @@ def mock_agent_event_queue():
@pytest.fixture @pytest.fixture
def event_serializer_registry() -> EventSerializerRegistry: def event_serializer_registry() -> AgentEventSerializerRegistry:
event_serializer_registry = EventSerializerRegistry() event_serializer_registry = AgentEventSerializerRegistry()
event_serializer_registry[SomeAgentEvent] = PydanticEventSerializer(SomeAgentEvent) event_serializer_registry[SomeAgentEvent] = PydanticEventSerializer(SomeAgentEvent)
event_serializer_registry[OtherAgentEvent] = PydanticEventSerializer(OtherAgentEvent) event_serializer_registry[OtherAgentEvent] = PydanticEventSerializer(OtherAgentEvent)
event_serializer_registry[DifferentAgentEvent] = PydanticEventSerializer(DifferentAgentEvent) event_serializer_registry[DifferentAgentEvent] = PydanticEventSerializer(DifferentAgentEvent)
@ -102,7 +102,7 @@ def flask_client(build_flask_client, mock_agent_event_queue, event_serializer_re
container = StubDIContainer() container = StubDIContainer()
container.register_instance(IAgentEventQueue, mock_agent_event_queue) container.register_instance(IAgentEventQueue, mock_agent_event_queue)
container.register_instance(EventSerializerRegistry, event_serializer_registry) container.register_instance(AgentEventSerializerRegistry, event_serializer_registry)
with build_flask_client(container) as flask_client: with build_flask_client(container) as flask_client:
yield flask_client yield flask_client