Function mock_config

testing/test_assertion.py:27–61

(
    verbose: int = 0,
    assertion_override: int | None = None,
    assertion_text_diff_style: str = util.ASSERTION_TEXT_DIFF_STYLE_NDIFF,
)

def mock_config(
    verbose: int = 0,
    assertion_override: int | None = None,
    assertion_text_diff_style: str = util.ASSERTION_TEXT_DIFF_STYLE_NDIFF,
):
    class TerminalWriter:
        def _highlight(self, source, lexer="python"):
            return source

    class PluginManager:
        def has_plugin(self, name: str) -> bool:
            return True

    class Config:
        pluginmanager = PluginManager()

        def get_terminal_writer(self):
            return TerminalWriter()

        def get_verbosity(self, verbosity_type: str | None = None) -> int:
            if verbosity_type is None:
                return verbose
            if verbosity_type == _Config.VERBOSITY_ASSERTIONS:
                if assertion_override is not None:
                    return assertion_override
                return verbose

            raise KeyError(f"Not mocked out: {verbosity_type}")

        def getini(self, name: str) -> str:
            if name == util.ASSERTION_TEXT_DIFF_STYLE_INI:
                return assertion_text_diff_style
            raise KeyError(f"Not mocked out: {name}")

    return Config()


class TestMockConfig: