def test_baseline(capsys, tmp_path: Path):
code_path = tmp_path / 'example.py'
code_path.write_text('a\nb\n')
with chdir(tmp_path):
result = main(['baseline', str(code_path)])
assert result == (0, '')
captured = capsys.readouterr()
assert captured.err == ''
hashes = captured.out.strip().split()
assert len(hashes) == 2
line_path = tmp_path / 'baseline.txt'
line_path.write_text(hashes[0])
with chdir(tmp_path):
result = main([
'lint',
'--baseline',
str(line_path),
'--format',
'default',
str(code_path),
])
assert result == (1, '')
captured = capsys.readouterr()
assert captured.err == ''
assert captured.out.strip() == "{}:2:1: F821 undefined name 'b'".format(
str(code_path))
def test_version(capsys):
result = main(['--version'])
assert result == (0, '')
captured = capsys.readouterr()
assert captured.err == ''
assert 'FlakeHell' in captured.out
assert 'Flake8' in captured.out
def test_exceptions_stdin(capsys, tmp_path: Path):
# write config
text = """
[tool.flakehell.plugins]
pyflakes = ["+*"]
[tool.flakehell.exceptions."example.py"]
pyflakes = ["-F401"]
"""
(tmp_path / 'pyproject.toml').write_text(dedent(text))
# make fake `stdin` with source file content
# code_path = tmp_path / 'example.py'
# code_path.write_text('import sys\na')
# call the command with passing matching `--stdin-display-name`
cmd = [
'lint', '--format', 'default', '--stdin-display-name', 'example.py',
'--', '-'
]
with chdir(tmp_path):
# with code_path.open('r') as stream:
result = main(cmd)
assert result == (1, '')
captured = capsys.readouterr()
assert captured.err == ''
assert captured.out.strip() == "example.py:2:1: F821 undefined name 'a'"
def test_help(flag, capsys):
result = main([flag])
assert result == (0, '')
captured = capsys.readouterr()
assert captured.err == ''
for name in ('baseline', 'code', 'codes', 'lint', 'missed', 'plugins'):
assert name in captured.out
def test_ignore_file_by_top_level_noqa(capsys, tmp_path: Path):
(tmp_path / 'example1.py').write_text('import sys\n')
(tmp_path / 'example2.py').write_text('# flake8: noqa\nimport sys\n')
with chdir(tmp_path):
result = main(['lint', '--format', 'default'])
assert result == (1, '')
captured = capsys.readouterr()
assert captured.err == ''
exp = "./example1.py:1:1: F401 'sys' imported but unused"
assert captured.out.strip() == exp
def test_exclude_file(capsys, tmp_path: Path):
(tmp_path / 'checked.py').write_text('import sys\n')
(tmp_path / 'ignored').mkdir()
(tmp_path / 'ignored' / 'first.py').write_text('import sys\n')
(tmp_path / 'ignored' / 'second.py').write_text('invalid syntax!')
with chdir(tmp_path):
result = main(['lint', '--format', 'default', '--exclude', 'ignored'])
assert result == (1, '')
captured = capsys.readouterr()
assert captured.err == ''
exp = """
./checked.py:1:1: F401 'sys' imported but unused
"""
assert captured.out.strip() == dedent(exp).strip()
def test_lint_help(capsys):
result = main(['lint', '--help'])
assert result == (0, '')
captured = capsys.readouterr()
assert captured.err == ''
# flake8 options
assert '-h, --help' in captured.out
assert '--builtins' in captured.out
assert '--isort-show-traceback' in captured.out
# ignored flake8 options
assert '--per-file-ignores' not in captured.out
assert '--enable-extensions' not in captured.out
# flakehell options
assert '--baseline' in captured.out
def test_diff(capsys, tmp_path: Path):
text = """
[tool.flakehell.plugins]
pyflakes = ["+*"]
[tool.flakehell.exceptions."tests/"]
pyflakes = ["-F401"]
"""
(tmp_path / 'pyproject.toml').write_text(dedent(text))
(tmp_path / 'example.py').write_text('import sys\na')
(tmp_path / 'tests').mkdir()
(tmp_path / 'tests' / 'test_example.py').write_text('import sys\na')
diff = """
--- a/tests/test_example.py
+++ b/tests/test_example.py
@@ -1,1 +1,2 @@ class FlakeHellCheckersManager(Manager):
- .
+ import sys
+ a
--- a/example.py
+++ b/example.py
@@ -1,1 +1,2 @@ class FlakeHellCheckersManager(Manager):
- .
+ import sys
+ a
"""
diff = dedent(diff)
exp = {'tests/test_example.py': {1, 2}, 'example.py': {1, 2}}
assert dict(parse_unified_diff(diff)) == exp
stdin_get_value.cache_clear()
stream = BytesIO(diff.encode())
with patch('sys.stdin.buffer', stream):
with chdir(tmp_path):
result = main(['lint', '--format', 'default', '--diff', 'example.py'])
assert result == (1, '')
captured = capsys.readouterr()
assert captured.err == ''
exp = """
example.py:1:1: F401 'sys' imported but unused
example.py:2:1: F821 undefined name 'a'
"""
assert captured.out.strip() == dedent(exp).strip()
def test_exceptions(capsys, tmp_path: Path):
text = """
[tool.flakehell.plugins]
pyflakes = ["+*"]
[tool.flakehell.exceptions."tests/"]
pyflakes = ["-F401"]
"""
(tmp_path / 'pyproject.toml').write_text(dedent(text))
(tmp_path / 'example.py').write_text('import sys\na')
(tmp_path / 'tests').mkdir()
(tmp_path / 'tests' / 'test_example.py').write_text('import sys\na')
with chdir(tmp_path):
result = main(['lint', '--format', 'default'])
assert result == (1, '')
captured = capsys.readouterr()
assert captured.err == ''
exp = """
./example.py:1:1: F401 'sys' imported but unused
./example.py:2:1: F821 undefined name 'a'
./tests/test_example.py:2:1: F821 undefined name 'a'
"""
assert captured.out.strip() == dedent(exp).strip()
# Copyright 2021 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""black"""
import sys
from flakehell._cli import main
if __name__ == "__main__":
print("flakehell: ", sys.argv)
main(sys.argv[1:])
