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

tests/pytests/unit/modules/test_debian_service.py
186 строк7 KB
Frode Gundersen
migrate test_debian_service to pytest
30 ноя 2022, 23:31
30 ноя 2022, 23:3120e3bc5
import pytest

import salt.modules.debian_service as debian_service
from tests.support.mock import DEFAULT, MagicMock, patch


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


def test_get_enabled():
    init_d_globs = ["/etc/init.d/S50foo", "/etc/init.d/S90bar"]
    glob_mock = MagicMock(
        side_effect=lambda x: init_d_globs if x == "/etc/rc[S3].d/S*" else DEFAULT
    )
    with patch("glob.glob", glob_mock), patch.object(
        debian_service, "_get_runlevel", MagicMock(return_value="3")
    ):
        ret = debian_service.get_enabled()
        expected = ["bar", "foo"]
        assert ret == expected, ret


def test_get_disabled():
    get_all = MagicMock(return_value=["foo", "bar", "baz"])
    get_enabled = MagicMock(return_value=["bar", "baz"])
    with patch.object(debian_service, "get_all", get_all), patch.object(
        debian_service, "get_enabled", get_enabled
    ):
        ret = debian_service.get_disabled()
        expected = ["foo"]
        assert ret == expected, ret


def test_available():
    get_all = MagicMock(return_value=["foo", "bar", "baz"])
    with patch.object(debian_service, "get_all", get_all):
        assert not debian_service.available("qux")
        assert debian_service.available("foo")


def test_missing():
    get_all = MagicMock(return_value=["foo", "bar", "baz"])
    with patch.object(debian_service, "get_all", get_all):
        assert debian_service.missing("qux")
        assert not debian_service.missing("foo")


def test_get_all():
    get_enabled = MagicMock(return_value=["baz", "hello", "world"])
    init_d_globs = ["/etc/init.d/foo", "/etc/init.d/bar"]
    glob_mock = MagicMock(
        side_effect=lambda x: init_d_globs if x == "/etc/init.d/*" else DEFAULT
    )
    with patch("glob.glob", glob_mock), patch.object(
        debian_service, "get_enabled", get_enabled
    ):
        ret = debian_service.get_all()
        expected = ["bar", "baz", "foo", "hello", "world"]
        assert ret == expected, ret


def test_start():
    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.start("foo")
        # Confirm expected command was run
        mock.assert_called_once_with("service foo start")
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.start("foo")


def test_stop():
    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.stop("foo")
        # Confirm expected command was run
        mock.assert_called_once_with("service foo stop")
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.stop("foo")


def test_restart():
    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.restart("foo")
        # Confirm expected command was run
        mock.assert_called_once_with("service foo restart")
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.restart("foo")


def test_reload_():
    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.reload_("foo")
        # Confirm expected command was run
        mock.assert_called_once_with("service foo reload")
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.reload_("foo")


def test_force_reload():
    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.force_reload("foo")
        # Confirm expected command was run
        mock.assert_called_once_with("service foo force-reload")
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.force_reload("foo")


def test_status():
    mock = MagicMock(return_value="123")
    with patch.dict(debian_service.__salt__, {"status.pid": mock}):
        assert debian_service.status("foo", "foobar")

    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.status("foo")
        # Confirm expected command was run
        mock.assert_called_once_with("service foo status", ignore_retcode=True)
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.enable("foo")

    mock = MagicMock(side_effect=lambda x, **y: 0 if x == "service bar status" else 1)
    get_all = MagicMock(return_value=["foo", "bar", "baz"])
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}), patch.object(
        debian_service, "get_all", get_all
    ):
        ret = debian_service.status("b*")
        expected = {"bar": True, "baz": False}
        assert ret == expected, ret


def test_enable():
    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.enable("foo")
        # Confirm expected command was run
        mock.assert_called_once_with(
            "insserv foo && update-rc.d foo enable", python_shell=True
        )
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.enable("foo")


def test_disable():
    mock = MagicMock(return_value=0)
    with patch.dict(debian_service.__salt__, {"cmd.retcode": mock}):
        # Test successful command (0 retcode)
        assert debian_service.disable("foo")
        # Confirm expected command was run
        mock.assert_called_once_with("update-rc.d foo disable")
        # Test unsuccessful command (nonzero retcode)
        mock.return_value = 1
        assert not debian_service.disable("foo")


def test_enabled():
    mock = MagicMock(return_value=["foo"])
    with patch.object(debian_service, "get_enabled", mock):
        assert debian_service.enabled("foo")
        assert not debian_service.enabled("bar")


def test_disabled():
    mock = MagicMock(return_value=["foo"])
    with patch.object(debian_service, "get_disabled", mock):
        assert debian_service.disabled("foo")
        assert not debian_service.disabled("bar")