tests/pytests/unit/states/test_proxy.py - master - githubmirror/salt

tests/pytests/unit/states/test_proxy.py
203 строки7 KB
Pedro Algarvio
Update to isort 5.10.1
01 авг 2022, 21:49
01 авг 2022, 21:494d49013
import pytest

import salt.states.proxy as proxy
from tests.support.mock import MagicMock, call, patch


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


def test_set_proxy_macos():
    """
    Test to make sure we can set the proxy settings on macOS
    """
    with patch.dict(proxy.__grains__, {"os": "Darwin"}):
        expected = {
            "changes": {
                "new": [
                    {
                        "port": "3128",
                        "server": "192.168.0.1",
                        "service": "http",
                        "user": "frank",
                    },
                    {
                        "port": "3128",
                        "server": "192.168.0.1",
                        "service": "https",
                        "user": "frank",
                    },
                    {
                        "port": "3128",
                        "server": "192.168.0.1",
                        "service": "ftp",
                        "user": "frank",
                    },
                    {"bypass_domains": ["salt.com", "test.com"]},
                ]
            },
            "comment": (
                "http proxy settings updated correctly\nhttps "
                "proxy settings updated correctly\nftp proxy "
                "settings updated correctly\nProxy "
                "bypass domains updated correctly\n"
            ),
            "name": "192.168.0.1",
            "result": True,
        }

        set_proxy_mock = MagicMock(return_value=True)
        patches = {
            "proxy.get_http_proxy": MagicMock(return_value={}),
            "proxy.get_https_proxy": MagicMock(return_value={}),
            "proxy.get_ftp_proxy": MagicMock(return_value={}),
            "proxy.get_proxy_bypass": MagicMock(return_value=[]),
            "proxy.set_http_proxy": set_proxy_mock,
            "proxy.set_https_proxy": set_proxy_mock,
            "proxy.set_ftp_proxy": set_proxy_mock,
            "proxy.set_proxy_bypass": set_proxy_mock,
        }

        with patch.dict(proxy.__salt__, patches):
            out = proxy.managed(
                "192.168.0.1",
                "3128",
                user="frank",
                password="passw0rd",
                bypass_domains=["salt.com", "test.com"],
            )
            out["changes"]["new"][-1]["bypass_domains"] = sorted(
                out["changes"]["new"][-1]["bypass_domains"]
            )

            calls = [
                call("192.168.0.1", "3128", "frank", "passw0rd", "Ethernet"),
                call("192.168.0.1", "3128", "frank", "passw0rd", "Ethernet"),
                call("192.168.0.1", "3128", "frank", "passw0rd", "Ethernet"),
                call(["salt.com", "test.com"], "Ethernet"),
            ]

            set_proxy_mock.assert_has_calls(calls)
            assert out == expected


def test_set_proxy_macos_same():
    """
    Test to make sure we can set the proxy settings on macOS
    """
    with patch.dict(proxy.__grains__, {"os": "Darwin"}):
        expected = {
            "changes": {},
            "comment": (
                "http proxy settings already set.\nhttps "
                "proxy settings already set.\nftp proxy settings"
                " already set.\nProxy bypass domains "
                "are already set correctly.\n"
            ),
            "name": "192.168.0.1",
            "result": True,
        }

        proxy_val = {"enabled": True, "server": "192.168.0.1", "port": "3128"}

        set_proxy_mock = MagicMock()
        patches = {
            "proxy.get_http_proxy": MagicMock(return_value=proxy_val),
            "proxy.get_https_proxy": MagicMock(return_value=proxy_val),
            "proxy.get_ftp_proxy": MagicMock(return_value=proxy_val),
            "proxy.get_proxy_bypass": MagicMock(return_value=["test.com", "salt.com"]),
            "proxy.set_http_proxy": set_proxy_mock,
            "proxy.set_https_proxy": set_proxy_mock,
            "proxy.set_ftp_proxy": set_proxy_mock,
            "proxy.set_proxy_bypass": set_proxy_mock,
        }

        with patch.dict(proxy.__salt__, patches):
            out = proxy.managed(
                "192.168.0.1",
                "3128",
                user="frank",
                password="passw0rd",
                bypass_domains=["salt.com", "test.com"],
            )

            assert not set_proxy_mock.called
            assert out == expected


def test_set_proxy_windows():
    """
    Test to make sure we can set the proxy settings on Windows
    """
    with patch.dict(proxy.__grains__, {"os": "Windows"}):
        expected = {
            "changes": {},
            "comment": "Proxy settings updated correctly",
            "name": "192.168.0.1",
            "result": True,
        }

        set_proxy_mock = MagicMock(return_value=True)
        patches = {
            "proxy.get_proxy_win": MagicMock(return_value={}),
            "proxy.get_proxy_bypass": MagicMock(return_value=[]),
            "proxy.set_proxy_win": set_proxy_mock,
        }

        with patch.dict(proxy.__salt__, patches):
            out = proxy.managed(
                "192.168.0.1",
                "3128",
                user="frank",
                password="passw0rd",
                bypass_domains=["salt.com", "test.com"],
            )

            set_proxy_mock.assert_called_once_with(
                "192.168.0.1",
                "3128",
                ["http", "https", "ftp"],
                ["salt.com", "test.com"],
            )
            assert out == expected


def test_set_proxy_windows_same():
    """
    Test to make sure we can set the proxy settings on Windows
    """
    with patch.dict(proxy.__grains__, {"os": "Windows"}):
        expected = {
            "changes": {},
            "comment": "Proxy settings already correct.",
            "name": "192.168.0.1",
            "result": True,
        }

        proxy_val = {
            "enabled": True,
            "http": {"enabled": True, "server": "192.168.0.1", "port": "3128"},
            "https": {"enabled": True, "server": "192.168.0.1", "port": "3128"},
            "ftp": {"enabled": True, "server": "192.168.0.1", "port": "3128"},
        }

        set_proxy_mock = MagicMock(return_value=True)
        patches = {
            "proxy.get_proxy_win": MagicMock(return_value=proxy_val),
            "proxy.get_proxy_bypass": MagicMock(return_value=["salt.com", "test.com"]),
            "proxy.set_proxy_win": set_proxy_mock,
        }

        with patch.dict(proxy.__salt__, patches):
            out = proxy.managed(
                "192.168.0.1",
                "3128",
                user="frank",
                password="passw0rd",
                bypass_domains=["salt.com", "test.com"],
            )

            assert not set_proxy_mock.called
            assert out == expected