p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Tests: Use default parameters in build_ransomware_payload() fixture 

This allows ransomware payloads with different mocks to be built on a
per-test basis with minimal effort and maximal code reuse.
This commit is contained in:
Mike Salvatore 2021-11-30 10:03:02 -05:00
parent bedc8d4f84
commit f87802678b
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
monkey/tests/unit_tests/infection_monkey/ransomware/test_ransomware_payload.py
View File

@ -21,12 +21,17 @@ def ransomware_payload(build_ransomware_payload, ransomware_payload_config):
def build_ransomware_payload( def build_ransomware_payload(
mock_file_encryptor, mock_file_selector, mock_leave_readme, telemetry_messenger_spy mock_file_encryptor, mock_file_selector, mock_leave_readme, telemetry_messenger_spy
): ):
def inner(config): def inner(
config,
file_encryptor=mock_file_encryptor,
file_selector=mock_file_selector,
leave_readme=mock_leave_readme,
):
return RansomwarePayload( return RansomwarePayload(
config, config,
mock_file_encryptor, file_encryptor,
mock_file_selector, file_selector,
mock_leave_readme, leave_readme,
telemetry_messenger_spy, telemetry_messenger_spy,
) )
@ -121,19 +126,15 @@ def test_telemetry_success(ransomware_payload, telemetry_messenger_spy):
def test_telemetry_failure( def test_telemetry_failure(
monkeypatch, ransomware_payload_config, mock_leave_readme, telemetry_messenger_spy build_ransomware_payload, ransomware_payload_config, telemetry_messenger_spy
): ):
file_not_exists = "/file/not/exist" file_not_exists = "/file/not/exist"
ransomware_payload = RansomwarePayload( mfe = MagicMock(
ransomware_payload_config, side_effect=FileNotFoundError(f"[Errno 2] No such file or directory: '{file_not_exists}'")
MagicMock(
side_effect=FileNotFoundError(
f"[Errno 2] No such file or directory: '{file_not_exists}'"
) )
), mfs = MagicMock(return_value=[PurePosixPath(file_not_exists)])
MagicMock(return_value=[PurePosixPath(file_not_exists)]), ransomware_payload = build_ransomware_payload(
mock_leave_readme, config=ransomware_payload_config, file_encryptor=mfe, file_selector=mfs
telemetry_messenger_spy,
) )
ransomware_payload.run_payload() ransomware_payload.run_payload()