mirror of
https://github.com/pypa/pip
synced 2023-12-13 21:30:23 +01:00
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
if sys.version_info < (3, 6):
|
|
_BROKEN_STDOUT_RETURN_CODE = 1
|
|
else:
|
|
# The new exit status was added in Python 3.6 as a result of:
|
|
# https://bugs.python.org/issue5319
|
|
_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', '-v', '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
|