diff options
| -rw-r--r-- | tests/unit/middleware/test_correlation_id.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tests/unit/middleware/test_correlation_id.py b/tests/unit/middleware/test_correlation_id.py index 070c23e..dc83cc7 100644 --- a/tests/unit/middleware/test_correlation_id.py +++ b/tests/unit/middleware/test_correlation_id.py @@ -28,14 +28,13 @@ class CorrelationIdMiddlewareTest(utils.BaseTestCase): 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 + self.stubs.Set(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") |