diff options
Diffstat (limited to 'tests/test_web.py')
| -rw-r--r-- | tests/test_web.py | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/tests/test_web.py b/tests/test_web.py index 578559c..94ab5c4 100644 --- a/tests/test_web.py +++ b/tests/test_web.py @@ -270,6 +270,33 @@ class WebMiddlewareTestCase(test.TestCase): self.assertEqual("yeah!", middleware(request)) self.assertEqual(mock_profiler_init.call_count, 0) + @mock.patch("osprofiler.web.profiler.init") + def test_wsgi_middleware_enable_via_python(self, mock_profiler_init): + request = mock.MagicMock() + request.get_response.return_value = "yeah!" + request.url = "someurl" + request.host_url = "someurl" + request.path = "path" + request.query_string = "query" + request.method = "method" + request.scheme = "scheme" + hmac_key = 'super_secret_key2' + + pack = utils.signed_pack({"base_id": "1", "parent_id": "2"}, hmac_key) + request.headers = { + "a": "1", + "b": "2", + "X-Trace-Info": pack[0], + "X-Trace-HMAC": pack[1] + } + + web.enable('super_secret_key1,super_secret_key2') + middleware = web.WsgiMiddleware("app", enabled=True) + self.assertEqual("yeah!", middleware(request)) + mock_profiler_init.assert_called_once_with(hmac_key=hmac_key, + base_id="1", + parent_id="2") + def test_disable(self): web.disable() self.assertTrue(web._DISABLED) |