def test_worker_generate_futures_method(self, monitor, expected):
scheduler = MagicMock()
self.worker.scheduler_pool = [scheduler]
with override_settings(SCHEDULER_MONITOR=monitor):
futures = [f for f in self.worker.generate_futures()]
self.assertEqual(len(futures), expected)
def test_connector_close_method(self, monitor):
self.scheduler.pull = MagicMock()
self.scheduler.poller = MagicMock()
self.scheduler.monitor_socket = MagicMock()
self.scheduler.monitor_poller = MagicMock()
with override_settings(SCHEDULER_MONITOR=monitor):
self.scheduler.close()
self.assertTrue(self.scheduler.pull.close.called)
self.assertTrue(self.scheduler.poller.unregister.called)
self.assertEqual(self.scheduler.monitor_socket.close.called,
monitor)
self.assertEqual(self.scheduler.monitor_poller.unregister.called,
monitor)
def test_worker_monitor_method(self, monitor, event, warn, info, logger):
scheduler = MagicMock()
scheduler.receive_event.side_effect = [
self.get_future(event), self.mark_soft_exit()
]
self.worker.scheduler_pool = [scheduler]
future = asyncio.async(self.worker.monitor(scheduler))
with override_settings(SCHEDULER_MONITOR=monitor):
self.loop.run_until_complete(future)
self.assertTrue(future.done())
self.assertEqual(logger.warning.called, warn)
self.assertEqual(logger.info.called, info)
