mirror of
https://github.com/pypa/pip
synced 2023-12-13 21:30:23 +01:00
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
import os
|
|
import subprocess
|
|
|
|
_BROKEN_STDOUT_RETURN_CODE = 120
|
|
|
|
|
|
def setup_broken_stdout_test(args, deprecated_python):
|
|
proc = subprocess.Popen(
|
|
args,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
# Call close() on stdout to cause a broken pipe.
|
|
proc.stdout.close()
|
|
returncode = proc.wait()
|
|
stderr = proc.stderr.read().decode("utf-8")
|
|
|
|
expected_msg = "ERROR: Pipe to stdout was broken"
|
|
if deprecated_python:
|
|
assert expected_msg in stderr
|
|
else:
|
|
assert stderr.startswith(expected_msg)
|
|
|
|
return stderr, returncode
|
|
|
|
|
|
def test_broken_stdout_pipe(deprecated_python):
|
|
"""
|
|
Test a broken pipe to stdout.
|
|
"""
|
|
stderr, returncode = setup_broken_stdout_test(
|
|
["pip", "list"],
|
|
deprecated_python=deprecated_python,
|
|
)
|
|
|
|
# Check that no traceback occurs.
|
|
assert "raise BrokenStdoutLoggingError()" not in stderr
|
|
assert stderr.count("Traceback") == 0
|
|
|
|
assert returncode == _BROKEN_STDOUT_RETURN_CODE
|
|
|
|
|
|
def test_broken_stdout_pipe__log_option(deprecated_python, tmpdir):
|
|
"""
|
|
Test a broken pipe to stdout when --log is passed.
|
|
"""
|
|
log_path = os.path.join(str(tmpdir), "log.txt")
|
|
stderr, returncode = setup_broken_stdout_test(
|
|
["pip", "--log", log_path, "list"],
|
|
deprecated_python=deprecated_python,
|
|
)
|
|
|
|
# Check that no traceback occurs.
|
|
assert "raise BrokenStdoutLoggingError()" not in stderr
|
|
assert stderr.count("Traceback") == 0
|
|
|
|
assert returncode == _BROKEN_STDOUT_RETURN_CODE
|
|
|
|
|
|
def test_broken_stdout_pipe__verbose(deprecated_python):
|
|
"""
|
|
Test a broken pipe to stdout with verbose logging enabled.
|
|
"""
|
|
stderr, returncode = setup_broken_stdout_test(
|
|
["pip", "-vv", "list"],
|
|
deprecated_python=deprecated_python,
|
|
)
|
|
|
|
# Check that a traceback occurs and that it occurs at most once.
|
|
# We permit up to two because the exception can be chained.
|
|
assert "raise BrokenStdoutLoggingError()" in stderr
|
|
assert 1 <= stderr.count("Traceback") <= 2
|
|
|
|
assert returncode == _BROKEN_STDOUT_RETURN_CODE
|