def test_logger_debug_task_completed(self):
worker = NowcastWorker("worker_name", "description")
worker.logger = Mock(name="logger")
worker._context = Mock(name="context")
worker.worker_func = Mock(name="worker_func", side_effect=SystemExit)
worker._do_work()
worker.logger.debug.assert_called_once_with(
"shutting down", extra={"logger_name": "worker_name"})
def test_system_exit_context_destroy(self):
worker = NowcastWorker("worker_name", "description")
worker.init_cli()
worker.logger = Mock(name="logger")
worker._context = Mock(name="context")
worker.worker_func = Mock(name="worker_func", side_effect=SystemExit)
worker._do_work()
assert worker._context.destroy.call_count == 1
def test_debug_mode(self):
worker = NowcastWorker("worker_name", "description")
worker._parsed_args = Mock(debug=True)
worker.logger = Mock(name="logger")
worker._context = Mock(name="context")
worker._init_zmq_interface()
assert worker.logger.debug.call_count == 1
assert not worker._context.socket.called
def test_logger_critical_unhandled_exception(self):
worker = NowcastWorker("worker_name", "description")
worker.logger = Mock(name="logger")
worker.tell_manager = Mock(name="tell_manager")
worker._context = Mock(name="context")
worker.worker_func = Mock(name="worker_func", side_effect=Exception)
worker._do_work()
worker.logger.critical.assert_called_once_with("unhandled exception:",
exc_info=True)
def test_socket(self):
worker = NowcastWorker("worker_name", "description")
worker._parsed_args = Mock(debug=False)
worker.logger = Mock(name="logger")
worker._context = Mock(name="context")
worker.config = {
"zmq": {
"host": "example.com",
"ports": {
"workers": 4343
}
}
}
worker._init_zmq_interface()
worker._context.socket.assert_called_once_with(zmq.REQ)
assert worker._socket.setsockopt.call_args_list == [
call(zmq.TCP_KEEPALIVE, 1),
call(zmq.TCP_KEEPALIVE_IDLE, 900),
]
worker._socket.connect.assert_called_once_with(
"tcp://example.com:4343")
assert worker.logger.info.call_count == 1
