Fix unchecked select.poll reference in setUp and tearDown for

platforms that don't have it.
This commit is contained in:
Gregory P. Smith 2012-07-16 13:38:45 -07:00
parent 5bcd005a0b
commit ac14aa5187
1 changed files with 7 additions and 5 deletions

View 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