ssh-audit/test/test_printconfig.py

import pytest
from ssh_audit.ssh_audit import process_commandline


# pylint: disable=attribute-defined-outside-init
class TestAuditConf:
    @pytest.fixture(autouse=True)
    def init(self, ssh_audit):
        self.OutputBuffer = ssh_audit.OutputBuffer()
        self.process_commandline = process_commandline

    @staticmethod
    def _test_conf(conf, **kwargs):
        options = {
            'print_config': ''
        }
        for k, v in kwargs.items():
            options[k] = v
        assert conf.print_config == options['print_config']

    def test_printconfig_conf_process_commandline(self):
        # pylint: disable=too-many-statements
        c = lambda x: self.process_commandline(self.OutputBuffer, x.split())  # noqa
        with pytest.raises(SystemExit):
            conf = c('')
        with pytest.raises(SystemExit):
            conf = c('--print-config')
            self._test_conf(conf)

        for vendor in ["Amazon", "Debian", "Rocky", "Mint", "Ubuntu", "NoOS"]:
            vendor = vendor
            for os_ver in ["2404", "2204", "2004", "1804", "2023", "22", "21", "20", "9", "Bookworm", "Bullseye", "NoVersion"]:
                os_ver = os_ver
                for cs_type in ["Client", "Server", "Mistake"]:
                    cs_type = cs_type
                    with pytest.raises(SystemExit):
                        conf = c(f'--print-config {vendor} {os_ver} {cs_type}')
                        self._test_conf(conf)