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

UT: Fix credentials passed to exploiter test

This commit is contained in:
Ilija Lazoroski 2022-09-05 14:52:24 +02:00 committed by Mike Salvatore
parent f93125f671
commit 4fca8ccf5d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py
View File

@ -66,7 +66,7 @@ def hosts_to_exploit(hosts):
def enqueue_hosts(hosts: Iterable[VictimHost]): def enqueue_hosts(hosts: Iterable[VictimHost]):
q = Queue() q: Queue = Queue()
for h in hosts: for h in hosts:
q.put(h) q.put(h)
@ -125,7 +125,7 @@ def test_credentials_passed_to_exploiter(run_exploiters):
run_exploiters(mock_puppet, 1) run_exploiters(mock_puppet, 1)
for call_args in mock_puppet.exploit_host.call_args_list: for call_args in mock_puppet.exploit_host.call_args_list:
assert call_args[0][3].get("credentials") == CREDENTIALS_FOR_PROPAGATION assert call_args[0][4].get("credentials") == CREDENTIALS_FOR_PROPAGATION
def test_stop_after_callback(exploiter_config, callback, scan_completed, stop, hosts_to_exploit): def test_stop_after_callback(exploiter_config, callback, scan_completed, stop, hosts_to_exploit):