diff --git a/monkey/tests/unit_tests/infection_monkey/telemetry/conftest.py b/monkey/tests/unit_tests/infection_monkey/telemetry/conftest.py index 1bcc2e423..ce790f2d9 100644 --- a/monkey/tests/unit_tests/infection_monkey/telemetry/conftest.py +++ b/monkey/tests/unit_tests/infection_monkey/telemetry/conftest.py @@ -1,3 +1,5 @@ +from unittest.mock import MagicMock + import pytest from infection_monkey.control import ControlClient @@ -13,5 +15,5 @@ def spy_send_telemetry(monkeypatch): _spy_send_telemetry.data = None control_client = ControlClient("localhost:5000") ControlClient.control_client_object = control_client - monkeypatch.setattr(ControlClient.control_client_object, "send_telemetry", _spy_send_telemetry) + ControlClient.control_client_object.send_telemetry = MagicMock(side_effect=_spy_send_telemetry) return _spy_send_telemetry