@@ -36,7 +36,7 @@ def __exit__(self, exc_type, exc_value, traceback):
36
36
delattr (settings , k )
37
37
else :
38
38
setattr (settings , k , v )
39
-
39
+
40
40
class BaseTestCase (TestCase ):
41
41
def setUp (self ):
42
42
request = Dingus ('request' )
@@ -52,18 +52,18 @@ def setUp(self):
52
52
53
53
class DebugToolbarTestCase (BaseTestCase ):
54
54
urls = 'tests.urls'
55
-
55
+
56
56
def test_middleware (self ):
57
57
with Settings (INTERNAL_IPS = ['127.0.0.1' ], DEBUG = True ):
58
58
resp = self .client .get ('/execute_sql/' )
59
59
self .assertEquals (resp .status_code , 200 )
60
60
61
61
def test_show_toolbar_DEBUG (self ):
62
62
request = self .request
63
-
63
+
64
64
request .META = {'REMOTE_ADDR' : '127.0.0.1' }
65
65
middleware = DebugToolbarMiddleware ()
66
-
66
+
67
67
with Settings (INTERNAL_IPS = ['127.0.0.1' ], DEBUG = True ):
68
68
self .assertTrue (middleware ._show_toolbar (request ))
69
69
@@ -72,10 +72,10 @@ def test_show_toolbar_DEBUG(self):
72
72
73
73
def test_show_toolbar_TEST (self ):
74
74
request = self .request
75
-
75
+
76
76
request .META = {'REMOTE_ADDR' : '127.0.0.1' }
77
77
middleware = DebugToolbarMiddleware ()
78
-
78
+
79
79
with Settings (INTERNAL_IPS = ['127.0.0.1' ], TEST = True , DEBUG = True ):
80
80
self .assertFalse (middleware ._show_toolbar (request ))
81
81
@@ -84,10 +84,10 @@ def test_show_toolbar_TEST(self):
84
84
85
85
def test_show_toolbar_INTERNAL_IPS (self ):
86
86
request = self .request
87
-
87
+
88
88
request .META = {'REMOTE_ADDR' : '127.0.0.1' }
89
89
middleware = DebugToolbarMiddleware ()
90
-
90
+
91
91
with Settings (INTERNAL_IPS = ['127.0.0.1' ], DEBUG = True ):
92
92
self .assertTrue (middleware ._show_toolbar (request ))
93
93
@@ -96,64 +96,75 @@ def test_show_toolbar_INTERNAL_IPS(self):
96
96
97
97
def test_request_urlconf_string (self ):
98
98
request = self .request
99
-
99
+
100
100
request .urlconf = 'tests.urls'
101
101
request .META = {'REMOTE_ADDR' : '127.0.0.1' }
102
102
middleware = DebugToolbarMiddleware ()
103
-
103
+
104
104
with Settings (INTERNAL_IPS = ['127.0.0.1' ], DEBUG = True ):
105
105
middleware .process_request (request )
106
-
106
+
107
107
self .assertFalse (isinstance (request .urlconf , basestring ))
108
-
108
+
109
109
self .assertTrue (hasattr (request .urlconf .urlpatterns [0 ], '_callback_str' ))
110
110
self .assertEquals (request .urlconf .urlpatterns [0 ]._callback_str , 'debug_toolbar.views.debug_media' )
111
111
self .assertTrue (hasattr (request .urlconf .urlpatterns [1 ], '_callback_str' ))
112
112
self .assertEquals (request .urlconf .urlpatterns [- 1 ]._callback_str , 'tests.views.execute_sql' )
113
113
114
114
def test_request_urlconf_string_per_request (self ):
115
115
request = self .request
116
-
116
+
117
117
request .urlconf = 'debug_toolbar.urls'
118
118
request .META = {'REMOTE_ADDR' : '127.0.0.1' }
119
119
middleware = DebugToolbarMiddleware ()
120
-
120
+
121
121
with Settings (INTERNAL_IPS = ['127.0.0.1' ], DEBUG = True ):
122
122
middleware .process_request (request )
123
123
request .urlconf = 'tests.urls'
124
124
middleware .process_request (request )
125
125
126
126
self .assertFalse (isinstance (request .urlconf , basestring ))
127
-
127
+
128
128
self .assertTrue (hasattr (request .urlconf .urlpatterns [0 ], '_callback_str' ))
129
129
self .assertEquals (request .urlconf .urlpatterns [0 ]._callback_str , 'debug_toolbar.views.debug_media' )
130
130
self .assertTrue (hasattr (request .urlconf .urlpatterns [1 ], '_callback_str' ))
131
131
self .assertEquals (request .urlconf .urlpatterns [- 1 ]._callback_str , 'tests.views.execute_sql' )
132
132
133
133
def test_request_urlconf_module (self ):
134
134
request = self .request
135
-
135
+
136
136
request .urlconf = __import__ ('tests.urls' ).urls
137
137
request .META = {'REMOTE_ADDR' : '127.0.0.1' }
138
138
middleware = DebugToolbarMiddleware ()
139
-
139
+
140
140
with Settings (INTERNAL_IPS = ['127.0.0.1' ], DEBUG = True ):
141
141
middleware .process_request (request )
142
-
142
+
143
143
self .assertFalse (isinstance (request .urlconf , basestring ))
144
-
144
+
145
145
self .assertTrue (hasattr (request .urlconf .urlpatterns [0 ], '_callback_str' ))
146
146
self .assertEquals (request .urlconf .urlpatterns [0 ]._callback_str , 'debug_toolbar.views.debug_media' )
147
147
self .assertTrue (hasattr (request .urlconf .urlpatterns [1 ], '_callback_str' ))
148
148
self .assertEquals (request .urlconf .urlpatterns [- 1 ]._callback_str , 'tests.views.execute_sql' )
149
149
150
+ def test_tuple_urlconf (self ):
151
+ request = self .request
152
+ urls = __import__ ('tests.urls' ).urls
153
+ urls .urlpatterns = tuple (urls .urlpatterns )
154
+ request .urlconf = urls
155
+ request .META = {'REMOTE_ADDR' : '127.0.0.1' }
156
+ middleware = DebugToolbarMiddleware ()
157
+ with Settings (INTERNAL_IPS = ['127.0.0.1' ], DEBUG = True ):
158
+ middleware .process_request (request )
159
+ self .assertFalse (isinstance (request .urlconf , basestring ))
160
+
150
161
def test_with_process_view (self ):
151
162
request = self .request
152
163
response = self .response
153
-
164
+
154
165
def _test_view (request ):
155
166
return HttpResponse ('' )
156
-
167
+
157
168
with Settings (DEBUG = True ):
158
169
panel = self .toolbar .get_panel (RequestVarsDebugPanel )
159
170
panel .process_request (request )
@@ -193,9 +204,9 @@ class SQLPanelTestCase(BaseTestCase):
193
204
def test_recording (self ):
194
205
panel = self .toolbar .get_panel (SQLDebugPanel )
195
206
self .assertEquals (len (panel ._queries ), 0 )
196
-
207
+
197
208
list (User .objects .all ())
198
-
209
+
199
210
# ensure query was logged
200
211
self .assertEquals (len (panel ._queries ), 1 )
201
212
query = panel ._queries [0 ]
@@ -210,10 +221,10 @@ def test_recording(self):
210
221
def test_disable_stacktraces (self ):
211
222
panel = self .toolbar .get_panel (SQLDebugPanel )
212
223
self .assertEquals (len (panel ._queries ), 0 )
213
-
224
+
214
225
with Settings (DEBUG_TOOLBAR_CONFIG = { 'ENABLE_STACKTRACES' : False }):
215
226
list (User .objects .all ())
216
-
227
+
217
228
# ensure query was logged
218
229
self .assertEquals (len (panel ._queries ), 1 )
219
230
query = panel ._queries [0 ]
@@ -252,19 +263,19 @@ def class_method(cls, *args, **kwargs):
252
263
def class_func (self , * args , ** kwargs ):
253
264
"""Used by dispatch tests"""
254
265
return 'blah'
255
-
266
+
256
267
def test_pre_hook (self ):
257
268
foo = {}
258
-
269
+
259
270
@pre_dispatch (module_func )
260
271
def test (** kwargs ):
261
272
foo .update (kwargs )
262
-
273
+
263
274
self .assertTrue (hasattr (module_func , '__wrapped__' ))
264
275
self .assertEquals (len (callbacks ['before' ]), 1 )
265
-
276
+
266
277
module_func ('hi' , foo = 'bar' )
267
-
278
+
268
279
self .assertTrue ('sender' in foo , foo )
269
280
# best we can do
270
281
self .assertEquals (foo ['sender' ].__name__ , 'module_func' )
@@ -278,13 +289,13 @@ def test(**kwargs):
278
289
self .assertTrue (len (foo ['kwargs' ]), 1 )
279
290
self .assertTrue ('foo' in foo ['kwargs' ])
280
291
self .assertEquals (foo ['kwargs' ]['foo' ], 'bar' )
281
-
292
+
282
293
callbacks ['before' ] = {}
283
-
294
+
284
295
@pre_dispatch (TrackingTestCase .class_func )
285
296
def test (** kwargs ):
286
297
foo .update (kwargs )
287
-
298
+
288
299
self .assertTrue (hasattr (TrackingTestCase .class_func , '__wrapped__' ))
289
300
self .assertEquals (len (callbacks ['before' ]), 1 )
290
301
@@ -305,16 +316,16 @@ def test(**kwargs):
305
316
self .assertEquals (foo ['kwargs' ]['foo' ], 'bar' )
306
317
307
318
# callbacks['before'] = {}
308
- #
319
+ #
309
320
# @pre_dispatch(TrackingTestCase.class_method)
310
321
# def test(**kwargs):
311
322
# foo.update(kwargs)
312
- #
323
+ #
313
324
# self.assertTrue(hasattr(TrackingTestCase.class_method, '__wrapped__'))
314
325
# self.assertEquals(len(callbacks['before']), 1)
315
- #
326
+ #
316
327
# TrackingTestCase.class_method()
317
- #
328
+ #
318
329
# self.assertTrue('sender' in foo, foo)
319
330
# # best we can do
320
331
# self.assertEquals(foo['sender'].__name__, 'class_method')
@@ -324,16 +335,16 @@ def test(**kwargs):
324
335
325
336
def test_post_hook (self ):
326
337
foo = {}
327
-
338
+
328
339
@post_dispatch (module_func )
329
340
def test (** kwargs ):
330
341
foo .update (kwargs )
331
-
342
+
332
343
self .assertTrue (hasattr (module_func , '__wrapped__' ))
333
344
self .assertEquals (len (callbacks ['after' ]), 1 )
334
-
345
+
335
346
module_func ('hi' , foo = 'bar' )
336
-
347
+
337
348
self .assertTrue ('sender' in foo , foo )
338
349
# best we can do
339
350
self .assertEquals (foo ['sender' ].__name__ , 'module_func' )
@@ -348,13 +359,13 @@ def test(**kwargs):
348
359
self .assertTrue (len (foo ['kwargs' ]), 1 )
349
360
self .assertTrue ('foo' in foo ['kwargs' ])
350
361
self .assertEquals (foo ['kwargs' ]['foo' ], 'bar' )
351
-
362
+
352
363
callbacks ['after' ] = {}
353
-
364
+
354
365
@post_dispatch (TrackingTestCase .class_func )
355
366
def test (** kwargs ):
356
367
foo .update (kwargs )
357
-
368
+
358
369
self .assertTrue (hasattr (TrackingTestCase .class_func , '__wrapped__' ))
359
370
self .assertEquals (len (callbacks ['after' ]), 1 )
360
371
0 commit comments