29e25e6e29
Falcon 2.0.0 introduces some breaking changes. The relevant ones here are: - falcon.testing.TestCase.api property was removed - falcon.testing.TestBase class was removed - falcon.HTTPRequestEntityTooLarge exception was renamed to falcon.HTTPPayloadTooLarge Additionally, the default behaviour for handling trailing slashes on URIs also changed: https://falcon.readthedocs.io/en/latest/user/faq.html#how-does-falcon- handle-a-trailing-slash-in-the-request-path This commit adds support for using the new release. Story: 2005695 Task: 33474 Change-Id: I2379522efef81b919098a0143b66cc259a184e70
76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
# Copyright 2016 FUJITSU LIMITED
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import falcon
|
|
import mock
|
|
|
|
from monasca_log_api.app.controller import healthchecks
|
|
from monasca_log_api.healthcheck import kafka_check as healthcheck
|
|
from monasca_log_api.tests import base
|
|
|
|
ENDPOINT = '/healthcheck'
|
|
|
|
|
|
class TestApiHealthChecks(base.BaseApiTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestApiHealthChecks, self).setUp()
|
|
self.resource = healthchecks.HealthChecks()
|
|
self.app.add_route(
|
|
ENDPOINT,
|
|
self.resource
|
|
)
|
|
|
|
def test_should_return_200_for_head(self):
|
|
res = self.simulate_request(
|
|
path=ENDPOINT,
|
|
method='HEAD')
|
|
self.assertEqual(falcon.HTTP_NO_CONTENT, res.status)
|
|
|
|
@mock.patch('monasca_log_api.healthcheck.kafka_check.KafkaHealthCheck')
|
|
def test_should_report_healthy_if_kafka_healthy(self, kafka_check):
|
|
kafka_check.healthcheck.return_value = healthcheck.CheckResult(True,
|
|
'OK')
|
|
self.resource._kafka_check = kafka_check
|
|
|
|
res = self.simulate_request(path=ENDPOINT,
|
|
headers={
|
|
'Content-Type': 'application/json'
|
|
},
|
|
method='GET')
|
|
self.assertEqual(falcon.HTTP_OK, res.status)
|
|
|
|
res = res.json
|
|
self.assertIn('kafka', res)
|
|
self.assertEqual('OK', res.get('kafka'))
|
|
|
|
@mock.patch('monasca_log_api.healthcheck.kafka_check.KafkaHealthCheck')
|
|
def test_should_report_unhealthy_if_kafka_healthy(self, kafka_check):
|
|
url = 'localhost:8200'
|
|
err_str = 'Could not connect to kafka at %s' % url
|
|
kafka_check.healthcheck.return_value = healthcheck.CheckResult(False,
|
|
err_str)
|
|
self.resource._kafka_check = kafka_check
|
|
|
|
res = self.simulate_request(path=ENDPOINT,
|
|
headers={
|
|
'Content-Type': 'application/json'
|
|
},
|
|
method='GET')
|
|
self.assertEqual(falcon.HTTP_SERVICE_UNAVAILABLE, res.status)
|
|
|
|
res = res.json
|
|
self.assertIn('kafka', res)
|
|
self.assertEqual(err_str, res.get('kafka'))
|