p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

UT: Change send telemetry spy syntax in conftest.py

This commit is contained in:
vakarisz 2022-06-14 12:14:27 +03:00
parent bbcac32172
commit 444b34d548
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/tests/unit_tests/infection_monkey/telemetry/conftest.py
View File

@ -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