From 6ced730b53db0cb561b9a07ef200b92fbca1de38 Mon Sep 17 00:00:00 2001
From: vakarisz <vakarisz@yahoo.com>
Date: Fri, 23 Sep 2022 12:39:08 +0300
Subject: [PATCH] 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
---
 .../payload/ransomware/test_integrated_ransomware.py            | 2 +-
 .../network/relay/test_relay_connection_handler.py              | 2 +-
 .../infection_monkey/network/relay/test_relay_user_handler.py   | 2 +-
 .../infection_monkey/network/relay/test_sockets_pipe.py         | 2 +-
 .../unit_tests/infection_monkey/telemetry/test_exploit_telem.py | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/monkey/tests/integration_tests/infection_monkey/payload/ransomware/test_integrated_ransomware.py b/monkey/tests/integration_tests/infection_monkey/payload/ransomware/test_integrated_ransomware.py
index ced8cf9fd..496793492 100644
--- a/monkey/tests/integration_tests/infection_monkey/payload/ransomware/test_integrated_ransomware.py
+++ b/monkey/tests/integration_tests/infection_monkey/payload/ransomware/test_integrated_ransomware.py
@@ -4,7 +4,7 @@ from unittest.mock import MagicMock
 import pytest
 
 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
diff --git a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py
index 48f8b6bf5..1ed65d1e0 100644
--- a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py
+++ b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py
@@ -4,7 +4,7 @@ from unittest.mock import MagicMock
 
 import pytest
 
-from monkey.infection_monkey.network.relay import (
+from infection_monkey.network.relay import (
     RELAY_CONTROL_MESSAGE_REMOVE_FROM_WAITLIST,
     RelayConnectionHandler,
     RelayUserHandler,
diff --git a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_user_handler.py b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_user_handler.py
index 6f3ecb8fa..2f314b85a 100644
--- a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_user_handler.py
+++ b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_user_handler.py
@@ -3,7 +3,7 @@ from time import sleep
 
 import pytest
 
-from monkey.infection_monkey.network.relay import RelayUserHandler
+from infection_monkey.network.relay import RelayUserHandler
 
 USER_ADDRESS = IPv4Address("0.0.0.0")
 
diff --git a/monkey/tests/unit_tests/infection_monkey/network/relay/test_sockets_pipe.py b/monkey/tests/unit_tests/infection_monkey/network/relay/test_sockets_pipe.py
index 0a98c0247..30b554967 100644
--- a/monkey/tests/unit_tests/infection_monkey/network/relay/test_sockets_pipe.py
+++ b/monkey/tests/unit_tests/infection_monkey/network/relay/test_sockets_pipe.py
@@ -1,6 +1,6 @@
 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():
diff --git a/monkey/tests/unit_tests/infection_monkey/telemetry/test_exploit_telem.py b/monkey/tests/unit_tests/infection_monkey/telemetry/test_exploit_telem.py
index c38c1d130..06e9c183d 100644
--- a/monkey/tests/unit_tests/infection_monkey/telemetry/test_exploit_telem.py
+++ b/monkey/tests/unit_tests/infection_monkey/telemetry/test_exploit_telem.py
@@ -3,9 +3,9 @@ import json
 import pytest
 
 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.telemetry.exploit_telem import ExploitTelem
-from monkey.infection_monkey.i_puppet.i_puppet import ExploiterResultData
 
 DOMAIN_NAME = "domain-name"
 IP = "0.0.0.0"