diff options
| author | Gregory P. Smith <greg@krypto.org> | 2012-07-16 20:38:45 (GMT) |
|---|---|---|
| committer | Gregory P. Smith <greg@krypto.org> | 2012-07-16 20:38:45 (GMT) |
| commit | ac14aa51876203be01ac4d62dec53c96dd1883c9 (patch) | |
| tree | d8ba2203a1a17426feb7a9dacdefce07c3b2c7b4 /Lib/test/test_telnetlib.py | |
| parent | 5bcd005a0b3af059ae3a91bb71064edee8cb8bf8 (diff) | |
| download | cpython-ac14aa51876203be01ac4d62dec53c96dd1883c9.zip cpython-ac14aa51876203be01ac4d62dec53c96dd1883c9.tar.gz cpython-ac14aa51876203be01ac4d62dec53c96dd1883c9.tar.bz2 | |
Fix unchecked select.poll reference in setUp and tearDown for
platforms that don't have it.
Diffstat (limited to 'Lib/test/test_telnetlib.py')
| -rw-r--r-- | Lib/test/test_telnetlib.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/Lib/test/test_telnetlib.py b/Lib/test/test_telnetlib.py index 5ccde23..f8f854c 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 |