flake8/tests/unit/test_filenameonly_formatter.py
2022-08-05 19:51:08 -04:00

41 lines
1.3 KiB
Python

"""Tests for the FilenameOnly formatter object."""
from __future__ import annotations
import argparse
from flake8.formatting import default
from flake8.violation import Violation
def options(**kwargs):
"""Create an argparse.Namespace instance."""
kwargs.setdefault("color", "auto")
kwargs.setdefault("output_file", None)
kwargs.setdefault("tee", False)
return argparse.Namespace(**kwargs)
def test_caches_filenames_already_printed():
"""Verify we cache filenames when we format them."""
formatter = default.FilenameOnly(options())
assert formatter.filenames_already_printed == set()
formatter.format(Violation("code", "file.py", 1, 1, "text", "l"))
assert formatter.filenames_already_printed == {"file.py"}
def test_only_returns_a_string_once_from_format():
"""Verify format ignores the second error with the same filename."""
formatter = default.FilenameOnly(options())
error = Violation("code", "file.py", 1, 1, "text", "1")
assert formatter.format(error) == "file.py"
assert formatter.format(error) is None
def test_show_source_returns_nothing():
"""Verify show_source returns nothing."""
formatter = default.FilenameOnly(options())
error = Violation("code", "file.py", 1, 1, "text", "1")
assert formatter.show_source(error) is None