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