agent: remove dependency on pytest-mock

This commit is contained in:
Mike Salvatore 2021-02-18 09:59:52 -05:00
parent 86ffaf358f
commit 4efdeeacc3
2 changed files with 2 additions and 6 deletions

View File

@ -13,6 +13,5 @@ pyftpdlib==1.5.6
pymssql<3.0
pypykatz==0.3.12
pysmb==1.2.5
pytest-mock==3.5.1
requests>=2.24
wmi==1.5.1 ; sys_platform == 'win32'

View File

@ -11,11 +11,8 @@ RESULT = False
@pytest.fixture
def post_breach_telem_test_instance(mocker):
mocker.patch(
"infection_monkey.telemetry.post_breach_telem.PostBreachTelem._get_hostname_and_ip",
return_value=(HOSTNAME, IP),
)
def post_breach_telem_test_instance(monkeypatch):
monkeypatch.setattr(PostBreachTelem, "_get_hostname_and_ip", lambda: (HOSTNAME, IP))
return PostBreachTelem(PBA, RESULT)