def test_setup(self, register_worker_mock): worker = BaseWorker() worker.setup() self.assertTrue(worker.id) self.assertTrue(worker.uuid) self.assertAlmostEqual(worker.created_dt, now(), delta=datetime.timedelta(seconds=1)) self.assertTrue(worker.host_name) self.assertTrue(register_worker_mock.called)
def test_setup(self, register_worker_mock): worker = BaseWorker() worker.setup() self.assertTrue(worker.id) self.assertTrue(worker.uuid) self.assertAlmostEqual(worker.created_dt, now(), delta=datetime.timedelta(seconds=1)) self.assertTrue(worker.host_name) self.assertTrue(register_worker_mock.called)
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 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_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_prepare_task_options(self): data = self.monit_task.get_data() task_name, task_options = BaseWorker._prepare_task_options( data['task'], self.monit_task.get_task_type()) self.assertEqual(self.monit_task.get_name(), task_name) self.assertEqual({'host': self.monit_task.host_address}, task_options)
def test_prepare_task_options(self): data = self.monit_task.get_data() task_name, task_options = BaseWorker._prepare_task_options(data['task'], self.monit_task.get_task_type()) self.assertEqual(self.monit_task.get_name(), task_name) self.assertEqual({'host': self.monit_task.host_address}, task_options)