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)