def test_redis_heartbeat_build(self):
heartbeat = RedisHeartBeat(build=1)
self.assertEqual(heartbeat.redis_key, RedisHeartBeat.KEY_BUILD.format(1))
self.assertEqual(heartbeat.is_alive(), False)
self.assertEqual(RedisHeartBeat.build_is_alive(1), False)
heartbeat.ping()
self.assertEqual(heartbeat.is_alive(), True)
self.assertEqual(RedisHeartBeat.build_is_alive(1), True)
heartbeat.clear()
self.assertEqual(heartbeat.is_alive(), False)
self.assertEqual(RedisHeartBeat.build_is_alive(1), False)
RedisHeartBeat.build_ping(1)
self.assertEqual(heartbeat.is_alive(), True)
self.assertEqual(RedisHeartBeat.build_is_alive(1), True)
def test_redis_heartbeat_job(self):
heartbeat = RedisHeartBeat(job=1)
self.assertEqual(heartbeat.redis_key, RedisHeartBeat.KEY_JOB.format(1))
self.assertEqual(heartbeat.is_alive(), False)
self.assertEqual(RedisHeartBeat.job_is_alive(1), False)
heartbeat.ping()
self.assertEqual(heartbeat.is_alive(), True)
self.assertEqual(RedisHeartBeat.job_is_alive(1), True)
heartbeat.clear()
self.assertEqual(heartbeat.is_alive(), False)
self.assertEqual(RedisHeartBeat.job_is_alive(1), False)
RedisHeartBeat.job_ping(1)
self.assertEqual(heartbeat.is_alive(), True)
self.assertEqual(RedisHeartBeat.job_is_alive(1), True)
def test_redis_heartbeat_experiment(self):
heartbeat = RedisHeartBeat(experiment=1)
self.assertEqual(heartbeat.redis_key, RedisHeartBeat.KEY_EXPERIMENT.format(1))
self.assertEqual(heartbeat.is_alive(), False)
self.assertEqual(RedisHeartBeat.experiment_is_alive(1), False)
heartbeat.ping()
self.assertEqual(heartbeat.is_alive(), True)
self.assertEqual(RedisHeartBeat.experiment_is_alive(1), True)
heartbeat.clear()
self.assertEqual(heartbeat.is_alive(), False)
self.assertEqual(RedisHeartBeat.experiment_is_alive(1), False)
RedisHeartBeat.experiment_ping(1)
self.assertEqual(heartbeat.is_alive(), True)
self.assertEqual(RedisHeartBeat.experiment_is_alive(1), True)
