UT: Fix credentials passed to exploiter test
This commit is contained in:
parent
f93125f671
commit
4fca8ccf5d
|
@ -66,7 +66,7 @@ def hosts_to_exploit(hosts):
|
|||
|
||||
|
||||
def enqueue_hosts(hosts: Iterable[VictimHost]):
|
||||
q = Queue()
|
||||
q: Queue = Queue()
|
||||
for h in hosts:
|
||||
q.put(h)
|
||||
|
||||
|
@ -125,7 +125,7 @@ def test_credentials_passed_to_exploiter(run_exploiters):
|
|||
run_exploiters(mock_puppet, 1)
|
||||
|
||||
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):
|
||||
|
|
Loading…
Reference in New Issue