flake8/tests/unit/test_application.py
2020-05-11 18:34:15 -07:00

100 lines
2.9 KiB
Python

"""Tests for the Application class."""
import argparse
import sys
import mock
import pytest
from flake8.main import application as app
def options(**kwargs):
"""Generate argparse.Namespace for our Application."""
kwargs.setdefault('verbose', 0)
kwargs.setdefault('output_file', None)
kwargs.setdefault('count', False)
kwargs.setdefault('exit_zero', False)
return argparse.Namespace(**kwargs)
@pytest.fixture
def application():
"""Create an application."""
return app.Application()
@pytest.mark.parametrize(
'result_count, catastrophic, exit_zero, value', [
(0, False, False, False),
(0, True, False, True),
(2, False, False, True),
(2, True, False, True),
(0, True, True, True),
(2, False, True, False),
(2, True, True, True),
]
)
def test_exit_does_raise(result_count, catastrophic, exit_zero, value,
application):
"""Verify Application.exit doesn't raise SystemExit."""
application.result_count = result_count
application.catastrophic_failure = catastrophic
application.options = options(exit_zero=exit_zero)
with pytest.raises(SystemExit) as excinfo:
application.exit()
assert excinfo.value.args[0] is value
def test_warns_on_unknown_formatter_plugin_name(application):
"""Verify we log a warning with an unfound plugin."""
default = mock.Mock()
execute = default.execute
application.formatting_plugins = {
'default': default,
}
with mock.patch.object(app.LOG, 'warning') as warning:
assert execute is application.formatter_for('fake-plugin-name')
assert warning.called is True
assert warning.call_count == 1
def test_returns_specified_plugin(application):
"""Verify we get the plugin we want."""
desired = mock.Mock()
execute = desired.execute
application.formatting_plugins = {
'default': mock.Mock(),
'desired': desired,
}
with mock.patch.object(app.LOG, 'warning') as warning:
assert execute is application.formatter_for('desired')
assert warning.called is False
def test_prelim_opts_args(application):
"""Verify we get sensible prelim opts and args."""
opts = application.parse_preliminary_options(
['--foo', '--verbose', 'src', 'setup.py', '--statistics', '--version'])
assert opts.verbose
def test_prelim_opts_ignore_help(application):
"""Verify -h/--help is not handled."""
# normally argparse would `SystemExit` on `--help`
application.parse_preliminary_options(['--help', '-h'])
def test_prelim_opts_handles_empty(application):
"""Verify empty argv lists are handled correctly."""
# this would set `.verbose` but we validate that it does not
irrelevant_args = ['--verbose']
with mock.patch.object(sys, 'argv', irrelevant_args):
opts = application.parse_preliminary_options([])
assert not opts.verbose