class TestReactorMixin:
"""
Mix this in to get TestReactor as self.reactor which is correctly cleaned up
at the end
"""
def setUpTestReactor(self, use_asyncio=False):
self.patch(threadpool, 'ThreadPool', NonThreadPool)
self.reactor = TestReactor()
_setReactor(self.reactor)
def deferToThread(f, *args, **kwargs):
return threads.deferToThreadPool(self.reactor,
self.reactor.getThreadPool(), f,
*args, **kwargs)
self.patch(threads, 'deferToThread', deferToThread)
# During shutdown sequence we must first stop the reactor and only then
# set unset the reactor used for eventually() because any callbacks
# that are run during reactor.stop() may use eventually() themselves.
self.addCleanup(_setReactor, None)
self.addCleanup(self.reactor.stop)
if use_asyncio:
self.asyncio_loop = TwistedLoop(self.reactor)
asyncio.set_event_loop(self.asyncio_loop)
self.asyncio_loop.start()
def stop():
self.asyncio_loop.stop()
self.asyncio_loop.close()
asyncio.set_event_loop(None)
self.addCleanup(stop)
