tests: Modify unit tests for AutoNewWindowsUser based on previous commit
changes
This commit is contained in:
parent
8d14ff6385
commit
2f5e6b516a
|
@ -5,7 +5,6 @@ import pytest
|
|||
from infection_monkey.utils.windows.users import AutoNewWindowsUser
|
||||
|
||||
TEST_USER = "test_user"
|
||||
ACTIVE_NO_NET_USER = "/ACTIVE:NO"
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
@ -20,13 +19,21 @@ def subprocess_check_output_spy(monkeypatch):
|
|||
return mock_check_output
|
||||
|
||||
|
||||
def test_new_user_try_delete_windows(subprocess_check_output_spy):
|
||||
new_user = AutoNewWindowsUser(TEST_USER, "password")
|
||||
class StubLogonUser:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
new_user.try_deactivate_user()
|
||||
assert f"net user {TEST_USER} {ACTIVE_NO_NET_USER}" in " ".join(
|
||||
subprocess_check_output_spy.command
|
||||
def Close():
|
||||
return None
|
||||
|
||||
|
||||
def test_new_user_delete_windows(subprocess_check_output_spy, monkeypatch):
|
||||
monkeypatch.setattr(
|
||||
"infection_monkey.utils.windows.users.win32security.LogonUser",
|
||||
lambda _, __, ___, ____, _____: StubLogonUser,
|
||||
)
|
||||
|
||||
new_user.try_delete_user()
|
||||
with (AutoNewWindowsUser(TEST_USER, "password")):
|
||||
pass
|
||||
|
||||
assert f"net user {TEST_USER} /delete" in " ".join(subprocess_check_output_spy.command)
|
||||
|
|
Loading…
Reference in New Issue