From 4fca8ccf5d3963e39feabf48fcac7ad867a237e8 Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Mon, 5 Sep 2022 14:52:24 +0200 Subject: [PATCH] UT: Fix credentials passed to exploiter test --- .../unit_tests/infection_monkey/master/test_exploiter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py b/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py index d8e5ef6b1..d62b40161 100644 --- a/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py +++ b/monkey/tests/unit_tests/infection_monkey/master/test_exploiter.py @@ -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):