From b312c509ce5bffd60a52178cf7323884450bb6af Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Fri, 1 Apr 2022 18:11:55 +0530 Subject: [PATCH] UT: Fix tests for new user creation --- .../tests/unit_tests/infection_monkey/utils/linux/test_users.py | 2 +- .../infection_monkey/utils/windows/test_windows_users.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/utils/linux/test_users.py b/monkey/tests/unit_tests/infection_monkey/utils/linux/test_users.py index 8b0408c0a..2c079309f 100644 --- a/monkey/tests/unit_tests/infection_monkey/utils/linux/test_users.py +++ b/monkey/tests/unit_tests/infection_monkey/utils/linux/test_users.py @@ -9,7 +9,7 @@ TEST_USER = "test_user" @pytest.fixture def subprocess_check_output_spy(monkeypatch): - def mock_check_output(command, stderr): + def mock_check_output(command, stderr, timeout): mock_check_output.command = command mock_check_output.command = "" diff --git a/monkey/tests/unit_tests/infection_monkey/utils/windows/test_windows_users.py b/monkey/tests/unit_tests/infection_monkey/utils/windows/test_windows_users.py index cb5e0746b..09762fe57 100644 --- a/monkey/tests/unit_tests/infection_monkey/utils/windows/test_windows_users.py +++ b/monkey/tests/unit_tests/infection_monkey/utils/windows/test_windows_users.py @@ -10,7 +10,7 @@ TEST_USER = "test_user" @pytest.fixture def subprocess_check_output_spy(monkeypatch): - def mock_check_output(command, stderr): + def mock_check_output(command, stderr, timeout): mock_check_output.command = command mock_check_output.command = ""