def test_failed_secondary_health_check(self): """Verifies that a secondary health check is not invoked when a primary check fails""" task = self.create_task(0, 'a') self._health_check_a.health(task).AndReturn(Retriable.alive()) self._health_check_b.health(task).AndReturn(Retriable.dead()) health_check = ChainedHealthCheck(self._health_check_a, self._health_check_b) self.replay() assert health_check.health(task) == Retriable.dead() self.verify()
def test_simple_chained_health_check(self): """Verify successful health check""" task = self.create_task(0, 'a') self._health_check_a.health(task).AndReturn(Retriable.alive()) self._health_check_b.health(task).AndReturn(Retriable.alive()) health_check = ChainedHealthCheck(self._health_check_a, self._health_check_b) self.replay() assert health_check.health(task) == Retriable.alive() self.verify()