UT: Fix imports common.events -> common.agent_events
This commit is contained in:
parent
d93eb4565b
commit
f8134e505e
|
@ -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):
|
||||
|
|
|
@ -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")
|
||||
|
||||
|
|
|
@ -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"
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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]
|
||||
|
|
Loading…
Reference in New Issue