def create_worker(self):
worker = BaseWorker()
worker.ZMQ_SERVER_ADDRESS = 'test'
worker.ZMQ_WORKER_REGISTRATOR_PORT = 321
worker.scheduler_port = 123
worker._init_worker()
return worker
def create_worker(self):
worker = BaseWorker()
worker.ZMQ_SERVER_ADDRESS = 'test'
worker.ZMQ_WORKER_REGISTRATOR_PORT = 321
worker.scheduler_port = 123
worker._init_worker()
return worker
def test_register_worker(self, context_mock):
answer_data = {'scheduler_port': 123}
register_socket_mock = context_mock.return_value.socket.return_value
register_socket_mock.recv_string.return_value = json.dumps(answer_data)
worker = BaseWorker()
worker._init_worker()
worker._register_worker()
self.assertTrue(register_socket_mock.send_string.called)
self.assertEqual(worker.scheduler_port, answer_data['scheduler_port'])
def test_register_worker(self, context_mock):
answer_data = {'scheduler_port': 123}
register_socket_mock = context_mock.return_value.socket.return_value
register_socket_mock.recv_string.return_value = json.dumps(answer_data)
worker = BaseWorker()
worker._init_worker()
worker._register_worker()
self.assertTrue(register_socket_mock.send_string.called)
self.assertEqual(worker.scheduler_port, answer_data['scheduler_port'])
def test_heart_beat(self, sleep_mock, emit_mock, is_stopped_mock):
is_stopped_mock.side_effect = (False, True)
worker = BaseWorker()
worker._init_worker()
worker._heart_beat()
self.assertEqual(WORKER_EVENT, emit_mock.call_args[0][0])
worker_data = json.loads(emit_mock.call_args[0][1],
object_hook=json_util.object_hook)
self.assertAlmostEqual(now(),
worker_data['heart_beat_dt'],
delta=datetime.timedelta(seconds=1))
def test_heart_beat(self, sleep_mock, emit_mock, is_stopped_mock):
is_stopped_mock.side_effect = (False, True)
worker = BaseWorker()
worker._init_worker()
worker._heart_beat()
self.assertEqual(WORKER_EVENT, emit_mock.call_args[0][0])
worker_data = json.loads(emit_mock.call_args[0][1], object_hook=json_util.object_hook)
self.assertAlmostEqual(
now(),
worker_data['heart_beat_dt'],
delta=datetime.timedelta(seconds=1)
)
