Fixed a function name lookup in distutils2.pypi.wrapper (#14263)

1 files changed, 37 insertions, 0 deletions

diff --git a/distutils2/tests/test_pypi_wrapper.py b/distutils2/tests/test_pypi_wrapper.py
new file mode 100644
index 0000000..4a9f8ca
--- /dev/null
+++ b/distutils2/tests/test_pypi_wrapper.py
@@ -0,0 +1,37 @@
+"""Tests for the distutils2.pypi.wrapper module."""
+
+
+from distutils2.tests import unittest
+from distutils2.pypi.wrapper import switch_index_if_fails
+
+
+class Index1(object):
+    def test(self):
+        raise Exception("boo")
+
+
+class Index2(object):
+    def test(self):
+        return 'OK'
+
+
+class Indexes(object):
+    _indexes = {'one': Index1(), 'two': Index2()}
+
+
+class TestPyPIWrapper(unittest.TestCase):
+
+    def test_wrapper(self):
+        index = Indexes._indexes['one']
+        func = switch_index_if_fails(getattr(index, 'test'), Indexes)
+        self.assertEqual(func(), 'OK')
+
+
+def test_suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(TestPyPIWrapper))
+    return suite
+
+
+if __name__ == '__main__':
+    unittest.main(defaultTest='test_suite')