Python ChainedHealthCheck Examples

Example #1

File: test_health_check.py Project: betepahos/incubator-aurora

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

Example #2

File: test_health_check.py Project: betepahos/incubator-aurora

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