def test_lint_command_no_color(tmp_path: Path, capsys):
(tmp_path / 'example.py').write_text(TEXT)
count = lint_command(['--nocolor', str(tmp_path)])
assert count == 1
captured = capsys.readouterr()
exp = '6:11 DEAL012 post contract error (-1) return -1 ^'
assert captured.out.split()[1:] == exp.split()
def test_lint_command_json(tmp_path: Path, capsys):
(tmp_path / 'example.py').write_text(TEXT)
count = lint_command(['--json', str(tmp_path)])
assert count == 1
captured = capsys.readouterr()
assert '" return -1"' in captured.out
assert '"-1"' in captured.out
assert '^' not in captured.out
def test_lint_command_two_errors(tmp_path: Path, capsys):
(tmp_path / 'example.py').write_text('from deal import pre\n' + TEXT)
count = lint_command([str(tmp_path)])
assert count == 2
captured = capsys.readouterr()
assert 'return -1' in captured.out
assert '(-1)' in captured.out
assert '^' in captured.out
def test_lint_command(tmp_path: Path, capsys):
(tmp_path / 'example.py').write_text(TEXT)
count = lint_command([str(tmp_path)])
assert count == 1
captured = capsys.readouterr()
assert 'return -1' in captured.out
assert '(-1)' in captured.out
assert '^' in captured.out
def test_lint_command_colors(tmp_path: Path, capsys):
(tmp_path / 'example.py').write_text(TEXT)
count = lint_command([str(tmp_path)])
assert count == 1
captured = capsys.readouterr()
assert '\x1b[34mreturn\x1b[39;49;00m -\x1b[34m1\x1b[39;49;00m' in captured.out
assert '\x1b[95m(-1)\x1b[0m' in captured.out
assert '\x1b[95m^\x1b[0m' in captured.out
def test_lint_command_two_files(tmp_path: Path, capsys):
(tmp_path / 'example1.py').write_text(TEXT)
(tmp_path / 'example2.py').write_text(TEXT)
count = lint_command(['--nocolor', str(tmp_path)])
assert count == 2
captured = capsys.readouterr()
assert 'example1.py' in captured.out
assert 'example2.py' in captured.out
assert 'return -1' in captured.out
assert '(-1)' in captured.out
assert '^' in captured.out