p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

UT: Add missing type hints to test_repository_service

This commit is contained in:
Mike Salvatore 2022-07-13 11:09:46 -04:00
parent 28b01a0d0e
commit 37885b786f
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/tests/unit_tests/monkey_island/cc/services/test_repository_service.py
View File

@ -13,7 +13,7 @@ WINDOWS_FILENAME = "windows_pba_file.ps1"
@pytest.fixture
def agent_configuration(default_agent_configuration) -> AgentConfiguration:
def agent_configuration(default_agent_configuration: AgentConfiguration) -> AgentConfiguration:
custom_pbas = replace(
default_agent_configuration.custom_pbas,
linux_filename=LINUX_FILENAME,
@ -23,7 +23,9 @@ def agent_configuration(default_agent_configuration) -> AgentConfiguration:
@pytest.fixture
def agent_configuration_repository(agent_configuration) -> IAgentConfigurationRepository:
def agent_configuration_repository(
agent_configuration: AgentConfiguration,
) -> IAgentConfigurationRepository:
agent_configuration_repository = InMemoryAgentConfigurationRepository()
agent_configuration_repository.store_configuration(agent_configuration)
@ -31,7 +33,7 @@ def agent_configuration_repository(agent_configuration) -> IAgentConfigurationRe
@pytest.fixture
def mock_file_repository():
def mock_file_repository() -> IFileRepository:
return MagicMock(spec=IFileRepository)