pip/tests/functional/test_warning.py

import platform
import textwrap

import pytest

from tests.lib import skip_if_not_python2, skip_if_python2


@pytest.fixture
def warnings_demo(tmpdir):
    demo = tmpdir.joinpath('warnings_demo.py')
    demo.write_text(textwrap.dedent('''
        from logging import basicConfig
        from pip._internal.utils import deprecation

        deprecation.install_warning_logger()
        basicConfig()

        deprecation.deprecated("deprecated!", replacement=None, gone_in=None)
    '''))
    return demo


def test_deprecation_warnings_are_correct(script, warnings_demo):
    result = script.run('python', warnings_demo, expect_stderr=True)
    expected = 'WARNING:pip._internal.deprecations:DEPRECATION: deprecated!\n'
    assert result.stderr == expected


def test_deprecation_warnings_can_be_silenced(script, warnings_demo):
    script.environ['PYTHONWARNINGS'] = 'ignore'
    result = script.run('python', warnings_demo)
    assert result.stderr == ''


DEPRECATION_TEXT = "drop support for Python 2.7"
CPYTHON_DEPRECATION_TEXT = "January 1st, 2020"


@skip_if_python2
def test_version_warning_is_not_shown_if_python_version_is_not_2(script):
    result = script.pip("debug", allow_stderr_warning=True)
    assert DEPRECATION_TEXT not in result.stderr, str(result)
    assert CPYTHON_DEPRECATION_TEXT not in result.stderr, str(result)


@skip_if_python2
def test_flag_does_nothing_if_python_version_is_not_2(script):
    script.pip("list", "--no-python-version-warning")


@skip_if_not_python2
def test_version_warning_is_shown_if_python_version_is_2(script):
    result = script.pip("debug", allow_stderr_warning=True)
    assert DEPRECATION_TEXT in result.stderr, str(result)
    if platform.python_implementation() == 'CPython':
        assert CPYTHON_DEPRECATION_TEXT in result.stderr, str(result)
    else:
        assert CPYTHON_DEPRECATION_TEXT not in result.stderr, str(result)


@skip_if_not_python2
def test_version_warning_is_not_shown_when_flag_is_passed(script):
    result = script.pip(
        "debug", "--no-python-version-warning", allow_stderr_warning=True
    )
    assert DEPRECATION_TEXT not in result.stderr, str(result)
    assert CPYTHON_DEPRECATION_TEXT not in result.stderr, str(result)
    assert "--no-python-version-warning" not in result.stderr