1717import django
1818from django .db import connection , connections
1919from django .http import HttpRequest
20- from django .test import TestCase , override_settings , modify_settings
20+ from django .test import TestCase , modify_settings , override_settings
2121from django .urls import resolve , reverse
22- from google .cloud .sqlcommenter .django .middleware import SqlCommenter , QueryWrapper
22+ from google .cloud .sqlcommenter .django .middleware import (
23+ QueryWrapper , SqlCommenter ,
24+ )
2325
2426from ..compat import mock
2527from ..opencensus_mock import mock_opencensus_tracer
3032# Adding the middleware twice in modify_settings
3133# doesn't work. The middleware is only used once
3234# if used in modify_settings
35+
36+
3337class TestMiddleware :
3438 def __init__ (self , get_response ):
3539 self .get_response = get_response
@@ -38,10 +42,12 @@ def __call__(self, request):
3842 with connection .execute_wrapper (QueryWrapper (request )):
3943 return self .get_response (request )
4044
45+
4146# Query log only active if DEBUG=True.
4247@override_settings (DEBUG = True )
4348class Tests (TestCase ):
4449 databases = '__all__'
50+
4551 @staticmethod
4652 def get_request (path ):
4753 request = HttpRequest ()
0 commit comments