From: Gregory P. Smith Date: Mon, 16 Jul 2012 20:38:45 +0000 (-0700) Subject: Fix unchecked select.poll reference in setUp and tearDown for X-Git-Tag: v3.3.0b2~182^2 X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=ac14aa51876203be01ac4d62dec53c96dd1883c9;p=python Fix unchecked select.poll reference in setUp and tearDown for platforms that don't have it. --- diff --git a/Lib/test/test_telnetlib.py b/Lib/test/test_telnetlib.py index 5ccde23b05..f8f854c110 100644 --- a/Lib/test/test_telnetlib.py +++ b/Lib/test/test_telnetlib.py @@ -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