UT: Fix imports common.events -> common.agent_events

guardicore · Sep 16, 2022 · d1654a4 · d1654a4
1 parent 4e3aac2
commit d1654a4
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/...y/tests/unit_tests/common/agent_event_serializers/test_agent_event_serializer_registry.py b/...y/tests/unit_tests/common/agent_event_serializers/test_agent_event_serializer_registry.py
@@ -4,7 +4,7 @@
 from pydantic import Field
 
 from common.agent_event_serializers import AgentEventSerializerRegistry, IAgentEventSerializer
-from common.events import AbstractAgentEvent
+from common.agent_events import AbstractAgentEvent
 
 
 class SomeEvent(AbstractAgentEvent):

diff --git a/...y/tests/unit_tests/common/agent_event_serializers/test_pydantic_agent_event_serializer.py b/...y/tests/unit_tests/common/agent_event_serializers/test_pydantic_agent_event_serializer.py
@@ -10,7 +10,7 @@
     IAgentEventSerializer,
     PydanticAgentEventSerializer,
 )
-from common.events import AbstractAgentEvent
+from common.agent_events import AbstractAgentEvent
 
 AGENT_ID = UUID("f811ad00-5a68-4437-bd51-7b5cc1768ad5")
 

diff --git a/monkey/tests/unit_tests/common/event_queue/test_pypubsub_agent_event_queue.py b/monkey/tests/unit_tests/common/event_queue/test_pypubsub_agent_event_queue.py
@@ -5,8 +5,8 @@
 import pytest
 from pubsub.core import Publisher
 
+from common.agent_events import AbstractAgentEvent
 from common.event_queue import AgentEventSubscriber, IAgentEventQueue, PyPubSubAgentEventQueue
-from common.events import AbstractAgentEvent
 
 EVENT_TAG_1 = "event tag 1"
 EVENT_TAG_2 = "event tag 2"

diff --git a/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py b/monkey/tests/unit_tests/infection_monkey/credential_collectors/test_mimikatz_collector.py
@@ -3,9 +3,9 @@
 
 import pytest
 
+from common.agent_events import CredentialsStolenEvent
 from common.credentials import Credentials, LMHash, NTHash, Password, Username
 from common.event_queue import IAgentEventQueue
-from common.events import CredentialsStolenEvent
 from infection_monkey.credential_collectors import MimikatzCredentialCollector
 from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_credential_collector import (  # noqa: E501
     MIMIKATZ_EVENT_TAGS,

diff --git a/monkey/tests/unit_tests/infection_monkey/credential_store/test_add_credentials_from_event.py b/monkey/tests/unit_tests/infection_monkey/credential_store/test_add_credentials_from_event.py
@@ -1,8 +1,8 @@
 from unittest.mock import MagicMock
 from uuid import UUID
 
+from common.agent_events import CredentialsStolenEvent
 from common.credentials import Credentials, Password, Username
-from common.events import CredentialsStolenEvent
 from infection_monkey.credential_repository import (
     IPropagationCredentialsRepository,
     add_credentials_from_event_to_propagation_credentials_repository,

diff --git a/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_event_repository.py b/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_event_repository.py
@@ -10,7 +10,7 @@
     AgentEventSerializerRegistry,
     PydanticAgentEventSerializer,
 )
-from common.events import AbstractAgentEvent
+from common.agent_events import AbstractAgentEvent
 from monkey_island.cc.repository import (
     IAgentEventRepository,
     MongoEventRepository,

diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/test_events.py b/monkey/tests/unit_tests/monkey_island/cc/resources/test_events.py
@@ -9,8 +9,8 @@
     AgentEventSerializerRegistry,
     PydanticAgentEventSerializer,
 )
+from common.agent_events import AbstractAgentEvent
 from common.event_queue import IAgentEventQueue
-from common.events import AbstractAgentEvent
 from monkey_island.cc.resources import AgentEvents
 
 EVENTS_URL = AgentEvents.urls[0]