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:
vakarisz 2022-09-23 12:39:08 +03:00
parent d48e8b3f3e
commit 6ced730b53
5 changed files with 5 additions and 5 deletions

View File

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

View File

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

View File

@ -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")

View File

@ -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():

View File

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