From 1991669402bdeaee563182ec6052c09f688fbde0 Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Fri, 5 Aug 2022 11:04:58 +0200 Subject: [PATCH] Island: Export {LINUX, WINDOWS}_PBA_TYPE from __init__ --- monkey/monkey_island/cc/resources/__init__.py | 2 +- .../monkey_island/cc/resources/test_pba_file_upload.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/monkey/monkey_island/cc/resources/__init__.py b/monkey/monkey_island/cc/resources/__init__.py index 3368869af..d04190b7c 100644 --- a/monkey/monkey_island/cc/resources/__init__.py +++ b/monkey/monkey_island/cc/resources/__init__.py @@ -6,5 +6,5 @@ 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 +from .pba_file_upload import PBAFileUpload, LINUX_PBA_TYPE, WINDOWS_PBA_TYPE from .pba_file_download import PBAFileDownload 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 397a5f6ab..a40eb60f8 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,8 +9,7 @@ from tests.utils import raise_ from common import DIContainer from monkey_island.cc.repository import IAgentConfigurationRepository, IFileRepository -from monkey_island.cc.resources import PBAFileUpload -from monkey_island.cc.resources.pba_file_upload import LINUX_PBA_TYPE, WINDOWS_PBA_TYPE +from monkey_island.cc.resources import LINUX_PBA_TYPE, WINDOWS_PBA_TYPE, PBAFileUpload TEST_FILE_CONTENTS = b"m0nk3y" TEST_FILE = (