diff options
author | Jenkins <jenkins@review.openstack.org> | 2013-05-20 15:53:58 +0000 |
---|---|---|
committer | Gerrit Code Review <review@openstack.org> | 2013-05-20 15:53:58 +0000 |
commit | 97bb81ddbcc47343c78e0a6efe724878fcb35ecb (patch) | |
tree | aa07e40965829d3073e677f42e05c1894cd47bab /tests | |
parent | 2d579dd3f311d3c5cb9aa1658d81488bb526b09c (diff) | |
parent | dfb4e68e005e68a2e272f92c62fc8455712f293a (diff) | |
download | oslo-97bb81ddbcc47343c78e0a6efe724878fcb35ecb.tar.gz oslo-97bb81ddbcc47343c78e0a6efe724878fcb35ecb.tar.xz oslo-97bb81ddbcc47343c78e0a6efe724878fcb35ecb.zip |
Merge "Add Middleware for correlation_id"
Diffstat (limited to 'tests')
-rw-r--r-- | tests/unit/middleware/test_correlation_id.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/tests/unit/middleware/test_correlation_id.py b/tests/unit/middleware/test_correlation_id.py new file mode 100644 index 0000000..070c23e --- /dev/null +++ b/tests/unit/middleware/test_correlation_id.py @@ -0,0 +1,51 @@ +# vim: tabstop=4 shiftwidth=4 softtabstop=4 + +# Copyright (c) 2013 Rackspace Hosting +# All Rights Reserved. +# +# 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 mock + +from openstack.common.middleware import correlation_id +from openstack.common import uuidutils +from tests import utils + + +class CorrelationIdMiddlewareTest(utils.BaseTestCase): + + def test_process_request(self): + app = mock.Mock() + req = mock.Mock() + req.headers = {} + original_method = uuidutils.generate_uuid + mock_generate_uuid = mock.Mock() + mock_generate_uuid.return_value = "fake_uuid" + uuidutils.generate_uuid = mock_generate_uuid + + middleware = correlation_id.CorrelationIdMiddleware(app) + middleware(req) + uuidutils.generate_uuid = original_method + + self.assertEquals(req.headers.get("X_CORRELATION_ID"), "fake_uuid") + + def test_process_request_should_not_regenerate_correlation_id(self): + app = mock.Mock() + req = mock.Mock() + req.headers = {"X_CORRELATION_ID": "correlation_id"} + + middleware = correlation_id.CorrelationIdMiddleware(app) + middleware(req) + + self.assertEquals(req.headers.get("X_CORRELATION_ID"), + "correlation_id") |