]> granicus.if.org Git - python/commitdiff
Fix unchecked select.poll reference in setUp and tearDown for
authorGregory P. Smith <greg@krypto.org>
Mon, 16 Jul 2012 20:38:45 +0000 (13:38 -0700)
committerGregory P. Smith <greg@krypto.org>
Mon, 16 Jul 2012 20:38:45 +0000 (13:38 -0700)
platforms that don't have it.

Lib/test/test_telnetlib.py

index 5ccde23b056e35de27ddf40eb445316e3b3a4a96..f8f854c110f39fb6073fb0417e6644555c2fd6ed 100644 (file)
@@ -168,14 +168,16 @@ def test_telnet(reads=(), cls=TelnetAlike, use_poll=None):
 class ExpectAndReadTestCase(TestCase):
     def setUp(self):
         self.old_select = select.select
-        self.old_poll = select.poll
         select.select = mock_select
-        select.poll = MockPoller
-        MockPoller.test_case = self
+        if hasattr(select, 'poll'):
+            self.old_poll = select.poll
+            select.poll = MockPoller
+            MockPoller.test_case = self
 
     def tearDown(self):
-        MockPoller.test_case = None
-        select.poll = self.old_poll
+        if hasattr(select, 'poll'):
+            MockPoller.test_case = None
+            select.poll = self.old_poll
         select.select = self.old_select