fix support for `dependency_links`

2 files changed, 17 insertions, 2 deletions

diff --git a/ptr.py b/ptr.py
index ecba60c..56a5c00 100644
--- a/ptr.py
+++ b/ptr.py
@@ -173,7 +173,10 @@ class PyTest(orig.test):
 		dist = CustomizedDist()
 		for attr in 'allow_hosts index_url'.split():
 			setattr(dist, attr, getattr(self, attr))
-		for attr in 'install_requires tests_require extras_require'.split():
+		for attr in (
+			'dependency_links install_requires '
+			'tests_require extras_require '
+		).split():
 			setattr(dist, attr, getattr(self.distribution, attr))
 		installed_dists = self.install_dists(dist)
 		if self.dry_run:
diff --git a/tests/test_ptr.py b/tests/test_ptr.py
index aeb0f8d..521c59a 100644
--- a/tests/test_ptr.py
+++ b/tests/test_ptr.py
@@ -3,6 +3,7 @@ from __future__ import unicode_literals
 import contextlib
 import io
 import os
+import shutil
 import sys
 import tarfile
 import textwrap
@@ -101,6 +102,13 @@ def test_egg_fetcher(venv, setuptools_req, test_args):
                 </body></html>
                 '''
             ).format(dist_sdist=dist_sdist))
+    # Move barbazquux1 out of the index.
+    shutil.move(index_dir / 'barbazquux1', venv.workspace)
+    barbazquux1_link = (
+        'file://' + str(venv.workspace.abspath())
+        + '/barbazquux1/barbazquux1-0.1.tar.gz'
+        + '#egg=barbazquux1-0.1'
+    )
     # Prepare fake project.
     project_dir = (venv.workspace / 'project-0.1').mkdir()
     with open(project_dir / 'setup.py', 'w') as fp:
@@ -110,6 +118,9 @@ def test_egg_fetcher(venv, setuptools_req, test_args):
             setup(
                 name='project',
                 version='0.1',
+                dependency_links = [
+                    {barbazquux1_link!r},
+                ],
                 setup_requires=[
                     'pytest-runner',
                 ],
@@ -126,7 +137,8 @@ def test_egg_fetcher(venv, setuptools_req, test_args):
                     'extra': 'barbazquux5',
                 }}
             )
-            ''').format(sys_platform=sys.platform))
+            ''').format(sys_platform=sys.platform,
+                        barbazquux1_link=barbazquux1_link))
     with open(project_dir / 'setup.cfg', 'w') as fp:
         fp.write(DALS(
             '''