Skip to content

Commit bb8a388

Browse files
committed
Fix broken tests
1 parent a5a7a81 commit bb8a388

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

tests/panels/test_sql.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -54,19 +54,25 @@ def test_recording_chunked_cursor(self):
5454
# ensure query was logged
5555
self.assertEqual(len(self.panel._queries), 1)
5656

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):
5962
list(User.objects.all())
6063

6164
# 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)
6366

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):
6672
list(User.objects.all().iterator())
6773

6874
# 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)
7076

7177
def test_generate_server_timing(self):
7278
self.assertEqual(len(self.panel._queries), 0)

0 commit comments

Comments
 (0)