@@ -54,19 +54,25 @@ def test_recording_chunked_cursor(self):
54
54
# ensure query was logged
55
55
self .assertEqual (len (self .panel ._queries ), 1 )
56
56
57
- @patch ("debug_toolbar.panels.sql.tracking.state" , wraps = sql_tracking .state )
58
- def test_cursor_wrapper_singleton (self , mock_state ):
57
+ @patch (
58
+ "debug_toolbar.panels.sql.tracking.NormalCursorWrapper" ,
59
+ wraps = sql_tracking .NormalCursorWrapper ,
60
+ )
61
+ def test_cursor_wrapper_singleton (self , mock_wrapper ):
59
62
list (User .objects .all ())
60
63
61
64
# ensure that cursor wrapping is applied only once
62
- self .assertEqual (mock_state . Wrapper .call_count , 1 )
65
+ self .assertEqual (mock_wrapper .call_count , 1 )
63
66
64
- @patch ("debug_toolbar.panels.sql.tracking.state" , wraps = sql_tracking .state )
65
- def test_chunked_cursor_wrapper_singleton (self , mock_state ):
67
+ @patch (
68
+ "debug_toolbar.panels.sql.tracking.NormalCursorWrapper" ,
69
+ wraps = sql_tracking .NormalCursorWrapper ,
70
+ )
71
+ def test_chunked_cursor_wrapper_singleton (self , mock_wrapper ):
66
72
list (User .objects .all ().iterator ())
67
73
68
74
# ensure that cursor wrapping is applied only once
69
- self .assertEqual (mock_state . Wrapper .call_count , 1 )
75
+ self .assertEqual (mock_wrapper .call_count , 1 )
70
76
71
77
def test_generate_server_timing (self ):
72
78
self .assertEqual (len (self .panel ._queries ), 0 )
0 commit comments