diff --git a/monkey/monkey_island/cc/app.py b/monkey/monkey_island/cc/app.py index 786d92146..397a035c9 100644 --- a/monkey/monkey_island/cc/app.py +++ b/monkey/monkey_island/cc/app.py @@ -16,6 +16,7 @@ from monkey_island.cc.resources import ( ClearSimulationData, IPAddresses, IslandLog, + PBAFileUpload, PropagationCredentials, RemoteRun, ResetAgentConfiguration, @@ -40,7 +41,6 @@ from monkey_island.cc.resources.netmap import NetMap from monkey_island.cc.resources.node import Node from monkey_island.cc.resources.node_states import NodeStates from monkey_island.cc.resources.pba_file_download import PBAFileDownload -from monkey_island.cc.resources.pba_file_upload import PBAFileUpload from monkey_island.cc.resources.ransomware_report import RansomwareReport from monkey_island.cc.resources.root import Root from monkey_island.cc.resources.security_report import SecurityReport diff --git a/monkey/monkey_island/cc/resources/__init__.py b/monkey/monkey_island/cc/resources/__init__.py index a345595a3..94fce4539 100644 --- a/monkey/monkey_island/cc/resources/__init__.py +++ b/monkey/monkey_island/cc/resources/__init__.py @@ -6,3 +6,4 @@ from .reset_agent_configuration import ResetAgentConfiguration from .propagation_credentials import PropagationCredentials from .ip_addresses import IPAddresses from .agent_configuration import AgentConfiguration +from .pba_file_upload import PBAFileUpload diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py index 2c4f54df1..397a5f6ab 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py +++ b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_upload.py @@ -9,11 +9,8 @@ from tests.utils import raise_ from common import DIContainer from monkey_island.cc.repository import IAgentConfigurationRepository, IFileRepository -from monkey_island.cc.resources.pba_file_upload import ( - LINUX_PBA_TYPE, - WINDOWS_PBA_TYPE, - PBAFileUpload, -) +from monkey_island.cc.resources import PBAFileUpload +from monkey_island.cc.resources.pba_file_upload import LINUX_PBA_TYPE, WINDOWS_PBA_TYPE TEST_FILE_CONTENTS = b"m0nk3y" TEST_FILE = (