forked from p15670423/monkey
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 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):
|
||||||
|
|
|
@ -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")
|
||||||
|
|
||||||
|
|
|
@ -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"
|
||||||
|
|
|
@ -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,
|
||||||
|
|
|
@ -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,
|
||||||
|
|
|
@ -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,
|
||||||
|
|
|
@ -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]
|
||||||
|
|
Loading…
Reference in New Issue