From 51318f71897cd538452acef471114368ac5fec6c Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Fri, 5 Aug 2022 10:05:57 +0200 Subject: [PATCH] Island: Export PBAFileDownload from __init__ --- monkey/monkey_island/cc/app.py | 2 +- monkey/monkey_island/cc/resources/__init__.py | 1 + .../monkey_island/cc/resources/test_pba_file_download.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/monkey/monkey_island/cc/app.py b/monkey/monkey_island/cc/app.py index 397a035c9..c837328fc 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, + PBAFileDownload, PBAFileUpload, PropagationCredentials, RemoteRun, @@ -40,7 +41,6 @@ from monkey_island.cc.resources.monkey import Monkey 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.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 94fce4539..3368869af 100644 --- a/monkey/monkey_island/cc/resources/__init__.py +++ b/monkey/monkey_island/cc/resources/__init__.py @@ -7,3 +7,4 @@ from .propagation_credentials import PropagationCredentials from .ip_addresses import IPAddresses from .agent_configuration import AgentConfiguration from .pba_file_upload import PBAFileUpload +from .pba_file_download import PBAFileDownload diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_download.py b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_download.py index 31ae0309a..cae3dccf1 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_download.py +++ b/monkey/tests/unit_tests/monkey_island/cc/resources/test_pba_file_download.py @@ -4,7 +4,7 @@ from tests.monkey_island import FILE_CONTENTS, FILE_NAME, MockFileRepository from tests.unit_tests.monkey_island.conftest import get_url_for_resource from monkey_island.cc.repository import IFileRepository -from monkey_island.cc.resources.pba_file_download import PBAFileDownload +from monkey_island.cc.resources import PBAFileDownload @pytest.fixture