"""Tests for distutils.text_file."""
from distutils.tests import support
from distutils.text_file import TextFile
import jaraco.path
import path
TEST_DATA = """# test file
line 3 \\
# intervening comment
continues on next line
"""
class TestTextFile(support.TempdirManager):
def test_class(self):
# old tests moved from text_file.__main__
# so they are really called by the buildbots
# result 1: no fancy options
result1 = [
'# test file\n',
'\n',
'line 3 \\\n',
'# intervening comment\n',
' continues on next line\n',
]
# result 2: just strip comments
result2 = ["\n", "line 3 \\\n", " continues on next line\n"]
# result 3: just strip blank lines
result3 = [
"# test file\n",
"line 3 \\\n",
"# intervening comment\n",
" continues on next line\n",
]
# result 4: default, strip comments, blank lines,
# and trailing whitespace
result4 = ["line 3 \\", " continues on next line"]
# result 5: strip comments and blanks, plus join lines (but don't
# "collapse" joined lines
result5 = ["line 3 continues on next line"]
# result 6: strip comments and blanks, plus join lines (and
# "collapse" joined lines
result6 = ["line 3 continues on next line"]
def test_input(count, description, file, expected_result):
result = file.readlines()
assert result == expected_result
tmp_path = path.Path(self.mkdtemp())
filename = tmp_path / 'test.txt'
jaraco.path.build({filename.name: TEST_DATA}, tmp_path)
in_file = TextFile(
filename,
strip_comments=False,
skip_blanks=False,
lstrip_ws=False,
rstrip_ws=False,
)
try:
test_input(1, "no processing", in_file, result1)
finally:
in_file.close()
in_file = TextFile(
filename,
strip_comments=True,
skip_blanks=False,
lstrip_ws=False,
rstrip_ws=False,
)
try:
test_input(2, "strip comments", in_file, result2)
finally:
in_file.close()
in_file = TextFile(
filename,
strip_comments=False,
skip_blanks=True,
lstrip_ws=False,
rstrip_ws=False,
)
try:
test_input(3, "strip blanks", in_file, result3)
finally:
in_file.close()
in_file = TextFile(filename)
try:
test_input(4, "default processing", in_file, result4)
finally:
in_file.close()
in_file = TextFile(
filename,
strip_comments=True,
skip_blanks=True,
join_lines=True,
rstrip_ws=True,
)
try:
test_input(5, "join lines without collapsing", in_file, result5)
finally:
in_file.close()
in_file = TextFile(
filename,
strip_comments=True,
skip_blanks=True,
join_lines=True,
rstrip_ws=True,
collapse_join=True,
)
try:
test_input(6, "join lines with collapsing", in_file, result6)
finally:
in_file.close()
|