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

tests/pytests/unit/modules/test_mac_sysctl.py
116 строк3 KB
Pedro Algarvio
Update to isort 5.10.1
01 авг 2022, 21:49
01 авг 2022, 21:494d49013
"""
    :codeauthor: Nicole Thomas <nicole@saltstack.com>
"""

import pytest

import salt.modules.mac_sysctl as mac_sysctl
from salt.exceptions import CommandExecutionError
from tests.support.mock import DEFAULT, MagicMock, mock_open, patch


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


def test_get():
    """
    Tests the return of get function
    """
    mock_cmd = MagicMock(return_value="foo")
    with patch.dict(mac_sysctl.__salt__, {"cmd.run": mock_cmd}):
        assert mac_sysctl.get("kern.ostype") == "foo"


def test_assign_cmd_failed():
    """
    Tests if the assignment was successful or not
    """
    cmd = {
        "pid": 3548,
        "retcode": 1,
        "stderr": "",
        "stdout": "net.inet.icmp.icmplim: 250 -> 50",
    }
    mock_cmd = MagicMock(return_value=cmd)
    with patch.dict(mac_sysctl.__salt__, {"cmd.run_all": mock_cmd}):
        pytest.raises(
            CommandExecutionError, mac_sysctl.assign, "net.inet.icmp.icmplim", 50
        )


def test_assign():
    """
    Tests the return of successful assign function
    """
    cmd = {
        "pid": 3548,
        "retcode": 0,
        "stderr": "",
        "stdout": "net.inet.icmp.icmplim: 250 -> 50",
    }
    ret = {"net.inet.icmp.icmplim": "50"}
    mock_cmd = MagicMock(return_value=cmd)
    with patch.dict(mac_sysctl.__salt__, {"cmd.run_all": mock_cmd}):
        assert mac_sysctl.assign("net.inet.icmp.icmplim", 50) == ret


def test_persist_no_conf_failure():
    """
    Tests adding of config file failure
    """
    read_data = IOError(13, "Permission denied", "/file")
    with patch("salt.utils.files.fopen", mock_open(read_data=read_data)), patch(
        "os.path.isfile", MagicMock(return_value=False)
    ):
        pytest.raises(
            CommandExecutionError,
            mac_sysctl.persist,
            "net.inet.icmp.icmplim",
            50,
            config=None,
        )


def test_persist_no_conf_success():
    """
    Tests successful add of config file when it did not already exist
    """
    config = "/etc/sysctl.conf"
    isfile_mock = MagicMock(side_effect=lambda x: False if x == config else DEFAULT)
    with patch("salt.utils.files.fopen", mock_open()) as m_open, patch(
        "os.path.isfile", isfile_mock
    ):
        mac_sysctl.persist("net.inet.icmp.icmplim", 50, config=config)
        num_handles = len(m_open.filehandles)
        assert num_handles == 1, num_handles
        writes = m_open.write_calls()
        num_writes = len(writes)
        assert num_writes == 1, num_writes
        assert writes[0] == "#\n# Kernel sysctl configuration\n#\n", writes[0]


def test_persist_success():
    """
    Tests successful write to existing sysctl file
    """
    config = "/etc/sysctl.conf"
    to_write = "#\n# Kernel sysctl configuration\n#\n"
    writelines_calls = [
        [
            "#\n",
            "# Kernel sysctl configuration\n",
            "#\n",
            "net.inet.icmp.icmplim=50\n",
        ]
    ]
    isfile_mock = MagicMock(side_effect=lambda x: True if x == config else DEFAULT)
    with patch(
        "salt.utils.files.fopen", mock_open(read_data=to_write)
    ) as m_open, patch("os.path.isfile", isfile_mock):
        mac_sysctl.persist("net.inet.icmp.icmplim", 50, config=config)
        num_handles = len(m_open.filehandles)
        assert num_handles == 1, num_handles
        writes = m_open.writelines_calls()
        assert writes == writelines_calls, writes