conftest.py 文件源码

def patch_click(monkeypatch):
    """ Fixture that monkeypatches click printing functions

        Patches functions like click.echo to capture all output in
        a mock to use functions like mock.assert_called_once_with().

        In the test you can directly import and use click functions
        as a Mock. For example:

            import click
            click.assert_called_with()
    """
    echo_mock = mock.Mock()
    confirm_mock = mock.Mock()
    monkeypatch.setattr(click, "echo", echo_mock)
    monkeypatch.setattr(click, "confirm", confirm_mock)
    return None