summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_telnetlib.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_telnetlib.py')
-rw-r--r--Lib/test/test_telnetlib.py12
1 files changed, 7 insertions, 5 deletions
diff --git a/Lib/test/test_telnetlib.py b/Lib/test/test_telnetlib.py
index b7c0052..c8da421 100644
--- a/Lib/test/test_telnetlib.py
+++ b/Lib/test/test_telnetlib.py
@@ -169,14 +169,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