def create_ioloop_with_timeout(self): """Create IOLoop with test timeout and schedule cleanup to close it """ ioloop = select_connection.IOLoop() self.addCleanup(ioloop.close) def _on_test_timeout(): """Called when test times out""" LOGGER.info('%s TIMED OUT (%s)', datetime.datetime.utcnow(), self) self.fail('Test timed out') ioloop.call_later(self.DEFAULT_TEST_TIMEOUT, _on_test_timeout) return ioloop
def setUp(self): select_type_patch = mock.patch.multiple(select_connection, SELECT_TYPE=self.SELECT_POLLER) select_type_patch.start() self.addCleanup(select_type_patch.stop) self.ioloop = select_connection.IOLoop() self.addCleanup(setattr, self, 'ioloop', None) activate_poller_patch = mock.patch.object( self.ioloop._poller, 'activate_poller', wraps=self.ioloop._poller.activate_poller) activate_poller_patch.start() self.addCleanup(activate_poller_patch.stop) deactivate_poller_patch = mock.patch.object( self.ioloop._poller, 'deactivate_poller', wraps=self.ioloop._poller.deactivate_poller) deactivate_poller_patch.start() self.addCleanup(deactivate_poller_patch.stop)
def setUp(self): select_connection.SELECT_TYPE = self.SELECT_POLLER self.ioloop = select_connection.IOLoop()