old_contents: import mock
  
  from auslib.test.web.test_client import ClientTestBase
  
  
  class TestDockerflowEndpoints(ClientTestBase):
  
      def testVersion(self):
          ret = self.client.get("/__version__")
          self.assertEquals(ret.data, """
  {
    "source":"https://github.com/mozilla/balrog",
    "version":"1.0",
    "commit":"abcdef123456"
  }
  """)
  
      def testHeartbeat(self):
          with mock.patch("auslib.global_state.dbo.rules.countRules") as cr:
              ret = self.client.get("/__heartbeat__")
              self.assertEqual(ret.status_code, 200)
              self.assertEqual(cr.call_count, 1)
  
      def testLbHeartbeat(self):
          ret = self.client.get("/__lbheartbeat__")
          self.assertEqual(ret.status_code, 200)
  
new_contents: import mock
  
  from auslib.test.web.test_client import ClientTestBase
  
  
  class TestDockerflowEndpoints(ClientTestBase):
  
      def testVersion(self):
          ret = self.client.get("/__version__")
          self.assertEquals(ret.data, """
  {
    "source":"https://github.com/mozilla/balrog",
    "version":"1.0",
    "commit":"abcdef123456"
  }
  """)
  
      def testHeartbeat(self):
          with mock.patch("auslib.global_state.dbo.rules.countRules") as cr:
              ret = self.client.get("/__heartbeat__")
              self.assertEqual(ret.status_code, 200)
              self.assertEqual(cr.call_count, 1)
  
      def testHeartbeatWithException(self):
          with mock.patch("auslib.global_state.dbo.rules.countRules") as cr:
              cr.side_effect = Exception("kabom!")
              # Because there's no web server between us and the endpoint, we recieve
              # the Exception directly instead of a 500 error
              self.assertRaises(Exception, self.client.get, "/__heartbeat__")
              self.assertEqual(cr.call_count, 1)
  
      def testLbHeartbeat(self):
          ret = self.client.get("/__lbheartbeat__")
          self.assertEqual(ret.status_code, 200)
  
subject: Add test to make sure public facing app raises exception when it hits an error.