mirror of
https://github.com/sprockets/sprockets.mixins.correlation.git
synced 2024-11-15 03:00:29 +00:00
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
import uuid
|
|
import unittest
|
|
|
|
from tornado import testing, web
|
|
|
|
from sprockets.mixins import correlation
|
|
|
|
|
|
class CorrelatedRequestHandler(correlation.HandlerMixin, web.RequestHandler):
|
|
|
|
def get(self, status_code):
|
|
status_code = int(status_code)
|
|
if status_code >= 300:
|
|
raise web.HTTPError(status_code)
|
|
self.write('status {0}'.format(status_code))
|
|
|
|
|
|
class CorrelationMixinTests(testing.AsyncHTTPTestCase):
|
|
|
|
def get_app(self):
|
|
return web.Application([
|
|
(r'/status/(?P<status_code>\d+)', CorrelatedRequestHandler),
|
|
])
|
|
|
|
def test_that_correlation_id_is_returned_when_successful(self):
|
|
self.http_client.fetch(self.get_url('/status/200'), self.stop)
|
|
response = self.wait()
|
|
self.assertIsNotNone(response.headers.get('Correlation-ID'))
|
|
|
|
def test_that_correlation_id_is_returned_in_error(self):
|
|
self.http_client.fetch(self.get_url('/status/500'), self.stop)
|
|
response = self.wait()
|
|
self.assertIsNotNone(response.headers.get('Correlation-ID'))
|
|
|
|
def test_that_correlation_id_is_copied_from_request(self):
|
|
correlation_id = uuid.uuid4().hex
|
|
self.http_client.fetch(self.get_url('/status/200'), self.stop,
|
|
headers={'Correlation-Id': correlation_id})
|
|
response = self.wait()
|
|
self.assertEqual(response.headers['correlation-id'], correlation_id)
|