forked from p15670423/monkey
UT: Remove monkey. part from imports in unit tests
No part of the codebase import with this longer syntax of `monkey.infection_monkey.something`, because `infection_monkey.something` is shorter and enough
This commit is contained in:
parent
d48e8b3f3e
commit
6ced730b53
|
@ -4,7 +4,7 @@ from unittest.mock import MagicMock
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
import infection_monkey.payload.ransomware.ransomware_builder as ransomware_builder
|
import infection_monkey.payload.ransomware.ransomware_builder as ransomware_builder
|
||||||
from monkey.common.agent_configuration.default_agent_configuration import RANSOMWARE_OPTIONS
|
from common.agent_configuration.default_agent_configuration import RANSOMWARE_OPTIONS
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
|
|
|
@ -4,7 +4,7 @@ from unittest.mock import MagicMock
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from monkey.infection_monkey.network.relay import (
|
from infection_monkey.network.relay import (
|
||||||
RELAY_CONTROL_MESSAGE_REMOVE_FROM_WAITLIST,
|
RELAY_CONTROL_MESSAGE_REMOVE_FROM_WAITLIST,
|
||||||
RelayConnectionHandler,
|
RelayConnectionHandler,
|
||||||
RelayUserHandler,
|
RelayUserHandler,
|
||||||
|
|
|
@ -3,7 +3,7 @@ from time import sleep
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from monkey.infection_monkey.network.relay import RelayUserHandler
|
from infection_monkey.network.relay import RelayUserHandler
|
||||||
|
|
||||||
USER_ADDRESS = IPv4Address("0.0.0.0")
|
USER_ADDRESS = IPv4Address("0.0.0.0")
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
from unittest.mock import MagicMock
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
from monkey.infection_monkey.network.relay import SocketsPipe
|
from infection_monkey.network.relay import SocketsPipe
|
||||||
|
|
||||||
|
|
||||||
def test_sockets_pipe__name_increments():
|
def test_sockets_pipe__name_increments():
|
||||||
|
|
|
@ -3,9 +3,9 @@ import json
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from infection_monkey.exploit.sshexec import SSHExploiter
|
from infection_monkey.exploit.sshexec import SSHExploiter
|
||||||
|
from infection_monkey.i_puppet.i_puppet import ExploiterResultData
|
||||||
from infection_monkey.model.host import VictimHost
|
from infection_monkey.model.host import VictimHost
|
||||||
from infection_monkey.telemetry.exploit_telem import ExploitTelem
|
from infection_monkey.telemetry.exploit_telem import ExploitTelem
|
||||||
from monkey.infection_monkey.i_puppet.i_puppet import ExploiterResultData
|
|
||||||
|
|
||||||
DOMAIN_NAME = "domain-name"
|
DOMAIN_NAME = "domain-name"
|
||||||
IP = "0.0.0.0"
|
IP = "0.0.0.0"
|
||||||
|
|
Loading…
Reference in New Issue