Adds tests for proper recursive upgrade behavior. refs pypa#304

Author: ejucovy

tests/packages/FSPkgUsesInitools/fspkg/__init__.py

@@ -0,0 +1 @@
+#

tests/packages/FSPkgUsesInitools/setup.py

@@ -0,0 +1,25 @@
+from setuptools import setup, find_packages
+
+version = '0.1'
+
+setup(name='FSPkgUsesInitools',
+      version=version,
+      description="File system test package",
+      long_description="""\
+File system test package""",
+      classifiers=[], # Get strings from http://pypi.python.org/pypi?%3Aaction=list_classifiers
+      keywords='pip tests',
+      author='pip',
+      author_email='[email protected]',
+      url='http://pip.openplans.org',
+      license='',
+      packages=find_packages(exclude=['ez_setup', 'examples', 'tests']),
+      include_package_data=True,
+      zip_safe=False,
+      install_requires=[
+        "INItools",
+      ],
+      entry_points="""
+      # -*- Entry points: -*-
+      """,
+      )

tests/packages/FSPkgUsesNewishInitools/FSPkgUsesInitools/fspkg/__init__.py

@@ -0,0 +1 @@
+#

tests/packages/FSPkgUsesNewishInitools/FSPkgUsesInitools/setup.py

@@ -0,0 +1,25 @@
+from setuptools import setup, find_packages
+
+version = '0.1'
+
+setup(name='FSPkgUsesInitools',
+      version=version,
+      description="File system test package",
+      long_description="""\
+File system test package""",
+      classifiers=[], # Get strings from http://pypi.python.org/pypi?%3Aaction=list_classifiers
+      keywords='pip tests',
+      author='pip',
+      author_email='[email protected]',
+      url='http://pip.openplans.org',
+      license='',
+      packages=find_packages(exclude=['ez_setup', 'examples', 'tests']),
+      include_package_data=True,
+      zip_safe=False,
+      install_requires=[
+        "INItools",
+      ],
+      entry_points="""
+      # -*- Entry points: -*-
+      """,
+      )

tests/packages/FSPkgUsesNewishInitools/fspkg/__init__.py

@@ -0,0 +1 @@
+#

tests/packages/FSPkgUsesNewishInitools/setup.py

@@ -0,0 +1,25 @@
+from setuptools import setup, find_packages
+
+version = '0.1'
+
+setup(name='FSPkgUsesNewishInitools',
+      version=version,
+      description="File system test package",
+      long_description="""\
+File system test package""",
+      classifiers=[], # Get strings from http://pypi.python.org/pypi?%3Aaction=list_classifiers
+      keywords='pip tests',
+      author='pip',
+      author_email='[email protected]',
+      url='http://pip.openplans.org',
+      license='',
+      packages=find_packages(exclude=['ez_setup', 'examples', 'tests']),
+      include_package_data=True,
+      zip_safe=False,
+      install_requires=[
+        "INItools>=0.3",
+      ],
+      entry_points="""
+      # -*- Entry points: -*-
+      """,
+      )

tests/test_upgrade.py

@@ -1,12 +1,20 @@
 import textwrap
-from os.path import join
+from os.path import abspath, join
+from pkg_resources import parse_version
 from nose.tools import nottest
 from tests.test_pip import (here, reset_env, run_pip, assert_all_changes,
                             write_file, pyversion, _create_test_package,
                             _change_test_package_version)
 from tests.local_repos import local_checkout
 
 
+def _get_pkg_version(env, pkg_name):
+    result = env.run('python', '-c',
+                     "from pkg_resources import get_distribution; "
+                     "print(get_distribution('%s').version)" % pkg_name)
+    return result.stdout.strip()
+
+
 def test_no_upgrade_unless_requested():
     """
     No upgrade if not specifically requested.
@@ -56,6 +64,86 @@ def test_upgrade_with_newest_already_installed():
     assert 'already up-to-date' in result.stdout
 
 
+def test_upgrade_without_unneeded_recursive_upgrades():
+    """
+    When upgrading a single package, that package's own dependencies should not be
+    upgraded unnecessarily if the user doesn't explicitly ask for them to be upgraded.
+    """
+    env = reset_env()
+    run_pip('install', 'INITools==0.2')
+
+    to_install = abspath(join(here, 'packages', 'FSPkgUsesInitools'))
+    run_pip('install', to_install)
+    run_pip('install', '--upgrade', to_install)
+    assert _get_pkg_version(env, 'initools') == '0.2',\
+           ('pip install --upgrade upgraded recursive dependency INITools '
+            'when it should not have')
+
+
+def test_upgrade_with_needed_recursive_upgrades():
+    """
+    When upgrading a single package A, that package's own dependencies should be
+    upgraded if the installed versions no longer satisfy A's requirements, even if
+    the user doesn't explicitly ask for them to be upgraded, 
+    """
+    env = reset_env()
+    to_install = abspath(join(here, 'packages', 'FSPkgUsesNewishInitools'))
+    run_pip('install', to_install)
+    run_pip('install', 'INITools==0.2')
+    run_pip('install', '--upgrade', to_install)
+    current_version = _get_pkg_version(env, 'initools')
+    assert parse_version(current_version) >= parse_version('0.3'),\
+           ('pip install --upgrade failed to upgrade recursive dependency '
+            'INITools when it should have')
+
+
+def test_upgrade_with_unneeded_recursive_upgrades_explicitly_requested():
+    """
+    When upgrading a single package A with --upgrade-recursive, all of A's
+    dependencies should be upgraded as well, even if the installed versions
+    already satisfy A's requirements.
+    """
+    env = reset_env()
+    run_pip('install', 'INITools==0.2')
+
+    to_install = abspath(join(here, 'packages', 'FSPkgUsesInitools'))
+    run_pip('install', to_install)
+    run_pip('install', '--upgrade-recursive', to_install)
+    current_version = _get_pkg_version(env, 'initools')
+    assert parse_version(current_version) > parse_version('0.2'),\
+           ('pip install --upgrade failed to upgrade recursive dependency '
+            'INITools when it was asked to')
+
+
+def test_upgrade_reqs_file_without_unneeded_recursive_upgrades():
+    """
+    When running non-recursive --upgrade against a requirements file, every package
+    explicitly listed in the requirements file should be upgraded; but any recursive
+    dependencies should not be upgraded.
+    """
+    env = reset_env()
+    run_pip('install', 'INITools==0.2')
+    run_pip('install', 'PyLogo==0.1')
+
+    to_install = abspath(join(here, 'packages', 'FSPkgUsesInitools'))
+    run_pip('install', to_install)
+
+    write_file('test-req.txt', textwrap.dedent("""\
+        %(FSPkgUsesInitools)s
+        PyLogo
+        """ % {'FSPkgUsesInitools': to_install}))
+
+    run_pip('install', '--upgrade', '-r', env.scratch_path/ 'test-req.txt')
+
+    pylogo_version = _get_pkg_version(env, 'pylogo')
+    assert parse_version(pylogo_version) > parse_version('0.1'),\
+           ('pip install --upgrade failed to upgrade explicit dependency '
+            'PyLogo when it should have')
+    assert _get_pkg_version(env, 'initools') == '0.2',\
+           ('pip install --upgrade upgraded recursive dependency INITools '
+            'when it should not have')
+
+
 def test_upgrade_force_reinstall_newest():
     """
     Force reinstallation of a package even if it is already at its newest