Fix mock

pypa · Apr 8, 2019 · 2882042 · 2882042
1 parent 4dfda6b
commit 2882042
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/tests/unit/test_unit_outdated.py b/tests/unit/test_unit_outdated.py
@@ -1,4 +1,3 @@
-import collections
 import datetime
 import os
 import sys
@@ -13,6 +12,14 @@
 from pip._internal.utils import outdated
 
 
+class MockFoundCandidates(object):
+    def __init__(self, best):
+        self._best = best
+
+    def get_best(self):
+        return self._best
+
+
 class MockPackageFinder(object):
 
     BASE_URL = 'https://pypi.org/simple/pip-{0}.tar.gz'
@@ -29,10 +36,8 @@ class MockPackageFinder(object):
     def __init__(self, *args, **kwargs):
         pass
 
-    def find_candidates(self, project_name, specifier):
-        return collections.namedtuple("FoundCandidates", "best")(
-            best=self.INSTALLATION_CANDIDATES[0],
-        )
+    def find_candidates(self, project_name):
+        return MockFoundCandidates(self.INSTALLATION_CANDIDATES[0])
 
 
 class MockDistribution(object):