UT: Add test for deactive and delete new auto windows user

This commit is contained in:
Ilija Lazoroski 2021-08-30 16:10:14 +02:00
parent 0635169362
commit 85316bcbb0
1 changed files with 32 additions and 0 deletions

View File

@ -0,0 +1,32 @@
import subprocess
import pytest
from infection_monkey.utils.windows.users import AutoNewWindowsUser
TEST_USER = "test_user"
ACTIVE_NO_NET_USER = "/ACTIVE:NO"
@pytest.fixture
def subprocess_check_output_spy(monkeypatch):
def mock_check_output(command, stderr):
mock_check_output.command = command
mock_check_output.command = ""
monkeypatch.setattr(subprocess, "check_output", mock_check_output)
return mock_check_output
def test_new_user_try_delete_windows(subprocess_check_output_spy):
new_user = AutoNewWindowsUser(TEST_USER, "password")
new_user.try_deactivate_user()
assert f"net user {TEST_USER} {ACTIVE_NO_NET_USER}" in " ".join(
subprocess_check_output_spy.command
)
new_user.try_delete_user()
assert f"net user {TEST_USER} /delete" in " ".join(subprocess_check_output_spy.command)