diff --git a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py index f82432dae..92aff10fc 100644 --- a/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py +++ b/monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py @@ -53,7 +53,7 @@ def test_connection_spawns_pipe(pipe_spawner, relay_user_handler, data_socket): connection_handler.handle_new_connection(data_socket) - assert pipe_spawner.spawn_pipe.called + pipe_spawner.spawn_pipe.assert_called_once_with(data_socket) def test_connection_adds_user(pipe_spawner, relay_user_handler, data_socket):