Merge "Mock log only after app creation"

1 files changed, 13 insertions, 12 deletions

diff --git a/keystonemiddleware/tests/unit/audit/test_logging_notifier.py b/keystonemiddleware/tests/unit/audit/test_logging_notifier.py
index f0174ad..0801861 100644
--- a/keystonemiddleware/tests/unit/audit/test_logging_notifier.py
+++ b/keystonemiddleware/tests/unit/audit/test_logging_notifier.py
@@ -25,15 +25,16 @@ class TestLoggingNotifier(base.BaseAuditMiddlewareTest):
 
         super(TestLoggingNotifier, self).setUp()
 
-    @mock.patch('keystonemiddleware.audit._LOG.info')
-    def test_api_request_no_messaging(self, log):
-        self.create_simple_app().get('/foo/bar',
-                                     extra_environ=self.get_environ_header())
-
-        # Check first notification with only 'request'
-        call_args = log.call_args_list[0][0]
-        self.assertEqual('audit.http.request', call_args[1]['event_type'])
-
-        # Check second notification with request + response
-        call_args = log.call_args_list[1][0]
-        self.assertEqual('audit.http.response', call_args[1]['event_type'])
+    def test_api_request_no_messaging(self):
+        app = self.create_simple_app()
+
+        with mock.patch('keystonemiddleware.audit._LOG.info') as log:
+            app.get('/foo/bar', extra_environ=self.get_environ_header())
+
+            # Check first notification with only 'request'
+            call_args = log.call_args_list[0][0]
+            self.assertEqual('audit.http.request', call_args[1]['event_type'])
+
+            # Check second notification with request + response
+            call_args = log.call_args_list[1][0]
+            self.assertEqual('audit.http.response', call_args[1]['event_type'])