tests/pytests/unit/modules/test_win_system.py - master - githubmirror/salt

tests/pytests/unit/modules/test_win_system.py
99 строк2 KB
Pedro Algarvio
Update pre-commit hook versions
29 фев 2024, 15:30
29 фев 2024, 15:30f454911
"""
    :codeauthor: Gareth J. Greenaway <gareth@saltstack.com>
"""

import pytest

import salt.modules.cmdmod as cmdmod
import salt.modules.win_system as win_system
from tests.support.mock import MagicMock, patch

pytestmark = [
    pytest.mark.windows_whitelisted,
    pytest.mark.skip_unless_on_windows,
]


@pytest.fixture
def configure_loader_modules():
    return {win_system: {"__salt__": {"cmd.run": cmdmod.run}}}


def test_lock():
    """
    Test locking workstation
    """
    with patch("ctypes.windll.user32.LockWorkStation", MagicMock()):
        ret = win_system.lock()
        assert ret


def test_get_computer_name():
    """
    Test getting the computer name
    """
    ret = win_system.get_computer_name()
    assert ret


def test_set_computer_name():
    """
    Test getting the computer name
    """
    with patch("ctypes.windll.kernel32.SetComputerNameExW", MagicMock()):
        ret = win_system.set_computer_name("test_hostname")
        assert ret


def test_get_hostname():
    """
    Test getting the computer name
    """
    ret = win_system.get_hostname()
    assert ret


def test_set_system_time():
    """
    Test setting the system time
    """
    with patch.object(win_system, "set_system_date_time", MagicMock()):
        ret = win_system.set_system_time("12:05 AM")
        assert ret


def test_get_system_time():
    """
    Test setting the system time
    """
    ret = win_system.get_system_time()
    assert ret


def test_set_system_date_time():
    """
    Test setting the system time
    """
    get_local_time = (2022, 6, 3, 11, 0, 0, 0)
    with patch("win32api.GetLocalTime", MagicMock(return_value=get_local_time)), patch(
        "ctypes.windll.kernel32.SetLocalTime", MagicMock()
    ) as set_local_time:
        ret = win_system.set_system_time("12:05 AM")
        assert ret


def test_get_system_date():
    """
    Test setting the system date
    """
    ret = win_system.get_system_date()
    assert ret


def test_set_system_date():
    """
    Test setting the system time
    """
    with patch.object(win_system, "set_system_date_time", MagicMock()):
        ret = win_system.set_system_date("03-28-13")
        assert ret