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 common.agent_event_serializers import AgentEventSerializerRegistry, IAgentEventSerializer
from common.events import AbstractAgentEvent
from common.agent_events import AbstractAgentEvent
class SomeEvent(AbstractAgentEvent):

View File

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

View File

@ -5,8 +5,8 @@ from uuid import UUID
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"

View File

@ -3,9 +3,9 @@ from unittest.mock import MagicMock
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,

View File

@ -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,

View File

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

View File

@ -9,8 +9,8 @@ from common.agent_event_serializers import (
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]