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

This commit is contained in:
Shreya Malviya 2022-09-16 18:18:40 +05:30 committed by Mike Salvatore
parent d93eb4565b
commit f8134e505e
7 changed files with 7 additions and 7 deletions

View File

@ -4,7 +4,7 @@ import pytest
from pydantic import Field from pydantic import Field
from common.agent_event_serializers import AgentEventSerializerRegistry, IAgentEventSerializer from common.agent_event_serializers import AgentEventSerializerRegistry, IAgentEventSerializer
from common.events import AbstractAgentEvent from common.agent_events import AbstractAgentEvent
class SomeEvent(AbstractAgentEvent): class SomeEvent(AbstractAgentEvent):

View File

@ -10,7 +10,7 @@ from common.agent_event_serializers import (
IAgentEventSerializer, IAgentEventSerializer,
PydanticAgentEventSerializer, PydanticAgentEventSerializer,
) )
from common.events import AbstractAgentEvent from common.agent_events import AbstractAgentEvent
AGENT_ID = UUID("f811ad00-5a68-4437-bd51-7b5cc1768ad5") AGENT_ID = UUID("f811ad00-5a68-4437-bd51-7b5cc1768ad5")

View File

@ -5,8 +5,8 @@ from uuid import UUID
import pytest import pytest
from pubsub.core import Publisher from pubsub.core import Publisher
from common.agent_events import AbstractAgentEvent
from common.event_queue import AgentEventSubscriber, IAgentEventQueue, PyPubSubAgentEventQueue from common.event_queue import AgentEventSubscriber, IAgentEventQueue, PyPubSubAgentEventQueue
from common.events import AbstractAgentEvent
EVENT_TAG_1 = "event tag 1" EVENT_TAG_1 = "event tag 1"
EVENT_TAG_2 = "event tag 2" EVENT_TAG_2 = "event tag 2"

View File

@ -3,9 +3,9 @@ from unittest.mock import MagicMock
import pytest import pytest
from common.agent_events import CredentialsStolenEvent
from common.credentials import Credentials, LMHash, NTHash, Password, Username from common.credentials import Credentials, LMHash, NTHash, Password, Username
from common.event_queue import IAgentEventQueue from common.event_queue import IAgentEventQueue
from common.events import CredentialsStolenEvent
from infection_monkey.credential_collectors import MimikatzCredentialCollector from infection_monkey.credential_collectors import MimikatzCredentialCollector
from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_credential_collector import ( # noqa: E501 from infection_monkey.credential_collectors.mimikatz_collector.mimikatz_credential_collector import ( # noqa: E501
MIMIKATZ_EVENT_TAGS, MIMIKATZ_EVENT_TAGS,

View File

@ -1,8 +1,8 @@
from unittest.mock import MagicMock from unittest.mock import MagicMock
from uuid import UUID from uuid import UUID
from common.agent_events import CredentialsStolenEvent
from common.credentials import Credentials, Password, Username from common.credentials import Credentials, Password, Username
from common.events import CredentialsStolenEvent
from infection_monkey.credential_repository import ( from infection_monkey.credential_repository import (
IPropagationCredentialsRepository, IPropagationCredentialsRepository,
add_credentials_from_event_to_propagation_credentials_repository, add_credentials_from_event_to_propagation_credentials_repository,

View File

@ -10,7 +10,7 @@ from common.agent_event_serializers import (
AgentEventSerializerRegistry, AgentEventSerializerRegistry,
PydanticAgentEventSerializer, PydanticAgentEventSerializer,
) )
from common.events import AbstractAgentEvent from common.agent_events import AbstractAgentEvent
from monkey_island.cc.repository import ( from monkey_island.cc.repository import (
IAgentEventRepository, IAgentEventRepository,
MongoEventRepository, MongoEventRepository,

View File

@ -9,8 +9,8 @@ from common.agent_event_serializers import (
AgentEventSerializerRegistry, AgentEventSerializerRegistry,
PydanticAgentEventSerializer, PydanticAgentEventSerializer,
) )
from common.agent_events import AbstractAgentEvent
from common.event_queue import IAgentEventQueue from common.event_queue import IAgentEventQueue
from common.events import AbstractAgentEvent
from monkey_island.cc.resources import AgentEvents from monkey_island.cc.resources import AgentEvents
EVENTS_URL = AgentEvents.urls[0] EVENTS_URL = AgentEvents.urls[0]