Island: Export PBAFileUpload from __init__

This commit is contained in:
Ilija Lazoroski 2022-08-05 10:00:31 +02:00
parent e4a284b172
commit b874b6a736
3 changed files with 4 additions and 6 deletions

View File

@ -16,6 +16,7 @@ from monkey_island.cc.resources import (
ClearSimulationData, ClearSimulationData,
IPAddresses, IPAddresses,
IslandLog, IslandLog,
PBAFileUpload,
PropagationCredentials, PropagationCredentials,
RemoteRun, RemoteRun,
ResetAgentConfiguration, 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 import Node
from monkey_island.cc.resources.node_states import NodeStates 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_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.ransomware_report import RansomwareReport
from monkey_island.cc.resources.root import Root from monkey_island.cc.resources.root import Root
from monkey_island.cc.resources.security_report import SecurityReport from monkey_island.cc.resources.security_report import SecurityReport

View File

@ -6,3 +6,4 @@ from .reset_agent_configuration import ResetAgentConfiguration
from .propagation_credentials import PropagationCredentials from .propagation_credentials import PropagationCredentials
from .ip_addresses import IPAddresses from .ip_addresses import IPAddresses
from .agent_configuration import AgentConfiguration from .agent_configuration import AgentConfiguration
from .pba_file_upload import PBAFileUpload

View File

@ -9,11 +9,8 @@ from tests.utils import raise_
from common import DIContainer from common import DIContainer
from monkey_island.cc.repository import IAgentConfigurationRepository, IFileRepository from monkey_island.cc.repository import IAgentConfigurationRepository, IFileRepository
from monkey_island.cc.resources.pba_file_upload import ( from monkey_island.cc.resources import PBAFileUpload
LINUX_PBA_TYPE, from monkey_island.cc.resources.pba_file_upload import LINUX_PBA_TYPE, WINDOWS_PBA_TYPE
WINDOWS_PBA_TYPE,
PBAFileUpload,
)
TEST_FILE_CONTENTS = b"m0nk3y" TEST_FILE_CONTENTS = b"m0nk3y"
TEST_FILE = ( TEST_FILE = (