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'])
