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

UT: Modify UTs to conform to modified ExploiterResultData

This commit is contained in:
Shreya Malviya 2022-02-21 13:27:11 +05:30
parent add9c3a4fe
commit ae856383a9
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
monkey/tests/unit_tests/infection_monkey/master/test_propagator.py
View File

@ -100,6 +100,10 @@ dot_3_services = {
}, },
} }
os_windows = "windows"
os_linux = "linux"
@pytest.fixture @pytest.fixture
def mock_ip_scanner(): def mock_ip_scanner():
@ -184,34 +188,38 @@ class MockExploiter:
results_callback( results_callback(
"PowerShellExploiter", "PowerShellExploiter",
host, host,
ExploiterResultData(True, {}, {}, None), ExploiterResultData(True, True, os_windows, {}, {}, None),
) )
results_callback( results_callback(
"SSHExploiter", "SSHExploiter",
host, host,
ExploiterResultData(False, {}, {}, "SSH FAILED for .1"), ExploiterResultData(False, False, os_linux, {}, {}, "SSH FAILED for .1"),
) )
elif host.ip_addr.endswith(".2"): elif host.ip_addr.endswith(".2"):
results_callback( results_callback(
"PowerShellExploiter", "PowerShellExploiter",
host, host,
ExploiterResultData(False, {}, {}, "POWERSHELL FAILED for .2"), ExploiterResultData(
False, False, os_windows, {}, {}, "POWERSHELL FAILED for .2"
),
) )
results_callback( results_callback(
"SSHExploiter", "SSHExploiter",
host, host,
ExploiterResultData(False, {}, {}, "SSH FAILED for .2"), ExploiterResultData(False, False, os_linux, {}, {}, "SSH FAILED for .2"),
) )
elif host.ip_addr.endswith(".3"): elif host.ip_addr.endswith(".3"):
results_callback( results_callback(
"PowerShellExploiter", "PowerShellExploiter",
host, host,
ExploiterResultData(False, {}, {}, "POWERSHELL FAILED for .3"), ExploiterResultData(
False, False, os_windows, {}, {}, "POWERSHELL FAILED for .3"
),
) )
results_callback( results_callback(
"SSHExploiter", "SSHExploiter",
host, host,
ExploiterResultData(True, {}, {}, None), ExploiterResultData(True, True, os_linux, {}, {}, None),
) )