Skip to content

Commit b47a577

Browse files
committed
Update middleware benchmark to use settings.MIDDLEWARE
1 parent 2c84cb3 commit b47a577

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

djangobench/benchmarks/default_middleware/benchmark.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,17 +50,25 @@ def setup():
5050
global req_factory, handler_default_middleware, handler_no_middleware
5151
req_factory = RequestFactory()
5252

53-
settings.MIDDLEWARE_CLASSES = global_settings.MIDDLEWARE_CLASSES
53+
use_MIDDLEWARE_setting = True
54+
try:
55+
settings.MIDDLEWARE = global_settings.MIDDLEWARE
56+
except AttributeError:
57+
settings.MIDDLEWARE_CLASSES = global_settings.MIDDLEWARE_CLASSES
58+
use_MIDDLEWARE_setting = False
5459
handler_default_middleware = WSGIHandler()
5560
handler_default_middleware.load_middleware()
5661

57-
settings.MIDDLEWARE_CLASSES = []
62+
if use_MIDDLEWARE_setting:
63+
settings.MIDDLEWARE = []
64+
else:
65+
settings.MIDDLEWARE_CLASSES = []
5866
handler_no_middleware = WSGIHandler()
5967
handler_no_middleware.load_middleware()
6068

6169

6270
def benchmark_request(middleware_classes):
63-
settings.MIDDLEWARE_CLASSES = middleware_classes
71+
settings.MIDDLEWARE = settings.MIDDLEWARE_CLASSES = middleware_classes
6472
req_factory = RequestFactory()
6573
handler = WSGIHandler()
6674
handler.load_middleware()

0 commit comments

Comments
 (0)