Exemplo n.º 1
0
    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))
Exemplo n.º 2
0
    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)
        )