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

tests/pytests/unit/modules/test_win_certutil.py
105 строк4 KB
Pedro Algarvio
Update to isort 5.10.1
01 авг 2022, 21:49
01 авг 2022, 21:494d49013
import pytest

import salt.modules.win_certutil as certutil
from tests.support.mock import MagicMock, patch


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


def test_get_serial():
    """
    Test getting the serial number from a certificate
    """
    expected = "180720d39cd2db3244ba037417241e90"
    mock = MagicMock(
        return_value=(
            "CertInfo\r\n"
            "Cert Serial Number: 180720d39cd2db3244ba037417241e90\r\n"
            "\r\n"
            "OtherStuff"
        )
    )
    with patch.dict(certutil.__salt__, {"cmd.run": mock}), patch.dict(
        certutil.__salt__,
        {"cp.cache_file": MagicMock(return_value="/path/to/cert.cer")},
    ), patch("os.path.exists", MagicMock(return_value=True)):
        out = certutil.get_cert_serial("/path/to/cert.cer")
        mock.assert_called_once_with('certutil.exe -silent -verify "/path/to/cert.cer"')
        assert expected == out


def test_get_serials():
    """
    Test getting all the serial numbers from a store
    """
    expected = [
        "180720d39cd2db3244ba037417241e90",
        "1768ac4e5b72bf1d0df0df118b34b959",
    ]
    mock = MagicMock(
        return_value=(
            "CertInfo\r\n"
            "================ Certificate 0 ================\r\n"
            "Serial Number: 180720d39cd2db3244ba037417241e90\r\n"
            "OtherStuff\r\n"
            "\r\n"
            "================ Certificate 1 ================\r\n"
            "Serial Number: 1768ac4e5b72bf1d0df0df118b34b959\r\n"
            "OtherStuff"
        )
    )
    with patch.dict(certutil.__salt__, {"cmd.run": mock}):
        out = certutil.get_stored_cert_serials("TrustedPublisher")
        mock.assert_called_once_with('certutil.exe -store "TrustedPublisher"')
        assert expected == out


def test_add_store():
    """
    Test adding a certificate to a specific store
    """
    cmd_mock = MagicMock(
        return_value=(
            "CertInfo\r\n"
            "================ Certificate 0 ================\r\n"
            "Serial Number: 180720d39cd2db3244ba037417241e90\r\n"
            "OtherStuff"
        )
    )
    cache_mock = MagicMock(return_value="/tmp/cert.cer")
    with patch.dict(
        certutil.__salt__, {"cmd.run": cmd_mock, "cp.cache_file": cache_mock}
    ), patch("os.path.exists", MagicMock(return_value=True)):
        certutil.add_store("salt://path/to/file", "TrustedPublisher")
        cmd_mock.assert_called_once_with(
            'certutil.exe -addstore TrustedPublisher "/tmp/cert.cer"'
        )
        cache_mock.assert_called_once_with("salt://path/to/file", "base")


def test_del_store():
    """
    Test removing a certificate to a specific store
    """
    with patch("salt.modules.win_certutil.get_cert_serial") as cert_serial_mock:
        cmd_mock = MagicMock(
            return_value=(
                "CertInfo\r\n"
                "================ Certificate 0 ================\r\n"
                "Serial Number: 180720d39cd2db3244ba037417241e90\r\n"
                "OtherStuff"
            )
        )
        cache_mock = MagicMock(return_value="/tmp/cert.cer")
        cert_serial_mock.return_value = "ABCDEF"
        with patch.dict(
            certutil.__salt__, {"cmd.run": cmd_mock, "cp.cache_file": cache_mock}
        ), patch("os.path.exists", MagicMock(return_value=True)):
            certutil.del_store("salt://path/to/file", "TrustedPublisher")
            cmd_mock.assert_called_once_with(
                'certutil.exe -delstore TrustedPublisher "ABCDEF"'
            )
            cache_mock.assert_called_once_with("salt://path/to/file", "base")