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

tests/pytests/unit/modules/test_mac_desktop.py
106 строк3 KB
Pedro Algarvio
Update to isort 5.10.1
01 авг 2022, 21:49
01 авг 2022, 21:494d49013
"""
Unit Tests for the mac_desktop execution module.
"""

import pytest

import salt.modules.mac_desktop as mac_desktop
from salt.exceptions import CommandExecutionError
from tests.support.mock import MagicMock, patch


@pytest.fixture
def configure_loader_modules():
    return {mac_desktop: {}}


def test_get_output_volume():
    """
    Test if it get the output volume (range 0 to 100)
    """
    mock = MagicMock(return_value={"retcode": 0, "stdout": "25"})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        assert mac_desktop.get_output_volume() == "25"


def test_get_output_volume_error():
    """
    Tests that an error is raised when cmd.run_all errors
    """
    mock = MagicMock(return_value={"retcode": 1})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        pytest.raises(CommandExecutionError, mac_desktop.get_output_volume)


def test_set_output_volume():
    """
    Test if it set the volume of sound (range 0 to 100)
    """
    mock = MagicMock(return_value={"retcode": 0})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}), patch(
        "salt.modules.mac_desktop.get_output_volume", MagicMock(return_value="25")
    ):
        assert mac_desktop.set_output_volume("25")


def test_set_output_volume_error():
    """
    Tests that an error is raised when cmd.run_all errors
    """
    mock = MagicMock(return_value={"retcode": 1})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        pytest.raises(CommandExecutionError, mac_desktop.set_output_volume, "25")


def test_screensaver():
    """
    Test if it launch the screensaver
    """
    mock = MagicMock(return_value={"retcode": 0})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        assert mac_desktop.screensaver()


def test_screensaver_error():
    """
    Tests that an error is raised when cmd.run_all errors
    """
    mock = MagicMock(return_value={"retcode": 1})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        pytest.raises(CommandExecutionError, mac_desktop.screensaver)


def test_lock():
    """
    Test if it lock the desktop session
    """
    mock = MagicMock(return_value={"retcode": 0})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        assert mac_desktop.lock()


def test_lock_error():
    """
    Tests that an error is raised when cmd.run_all errors
    """
    mock = MagicMock(return_value={"retcode": 1})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        pytest.raises(CommandExecutionError, mac_desktop.lock)


def test_say():
    """
    Test if it says some words.
    """
    mock = MagicMock(return_value={"retcode": 0})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        assert mac_desktop.say()


def test_say_error():
    """
    Tests that an error is raised when cmd.run_all errors
    """
    mock = MagicMock(return_value={"retcode": 1})
    with patch.dict(mac_desktop.__salt__, {"cmd.run_all": mock}):
        pytest.raises(CommandExecutionError, mac_desktop.say)