p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Add test for settting agent proxy

This commit is contained in:
Ilija Lazoroski 2021-10-07 10:50:41 +02:00
parent a11d1d5f1e
commit a8182cbb3d
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
monkey/tests/unit_tests/infection_monkey/test_control.py Normal file
View File

@ -0,0 +1,18 @@
import pytest
from monkey.infection_monkey.control import ControlClient
PROXY_FOUND = ("8.8.8.8", "45455")
@pytest.mark.parametrize("is_windows_os", [True, False])
def test_control_set_proxies(monkeypatch, is_windows_os):
monkeypatch.setattr("monkey.infection_monkey.control.is_windows_os", lambda: is_windows_os)
control_client = ControlClient()
control_client.set_proxies(PROXY_FOUND)
if is_windows_os:
assert control_client.proxies["https"].startswith("http://")
else:
assert control_client.proxies["https"].startswith(PROXY_FOUND[0])