Skip to content

Commit 52d9f14

Browse files
committed
refactor: add doc string to helper function and tests
1 parent cb77283 commit 52d9f14

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/panels/test_sql.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,13 @@ def sql_call(*, use_iterator=False):
3535

3636

3737
def sql_call_ddt():
38-
"""Test query to fetch DDT related entries from one of the DDT models"""
38+
"""Helper function to query one of the DDT models to test tracking of DDT entries."""
3939
qs = HistoryEntry.objects.all()
4040
return list(qs)
4141

4242

4343
async def async_sql_call_ddt():
44+
"""(async) Helper function to query one of the DDT models to test tracking of DDT entries."""
4445
qs = HistoryEntry.objects.all()
4546
return await sync_to_async(list)(qs)
4647

@@ -118,6 +119,7 @@ async def test_recording_concurrent_async(self):
118119

119120
@override_settings(DEBUG_TOOLBAR_CONFIG={"TRACK_DDT_MODELS": True})
120121
def test_ddt_models_tracked(self):
122+
"""test if DDT models are being tracked when the `TRACK_DDT_MODELS` is set to True"""
121123
self.assertEqual(len(self.panel._queries), 0)
122124

123125
sql_call_ddt()
@@ -129,6 +131,7 @@ def test_ddt_models_tracked(self):
129131

130132
@override_settings(DEBUG_TOOLBAR_CONFIG={"TRACK_DDT_MODELS": True})
131133
async def test_ddt_models_tracked_async(self):
134+
"""(async) test if DDT models are being tracked when the `TRACK_DDT_MODELS` is set to True"""
132135
self.assertEqual(len(self.panel._queries), 0)
133136

134137
await async_sql_call_ddt()
@@ -139,13 +142,15 @@ async def test_ddt_models_tracked_async(self):
139142
self.assertTrue(HistoryEntry._meta.db_table in query["sql"])
140143

141144
def test_ddt_models_not_tracked(self):
145+
"""Tests whether DDt models are not being tracked when the `TRACK_DDT_MODELS` is set to False"""
142146
self.assertEqual(len(self.panel._queries), 0)
143147

144148
sql_call_ddt()
145149

146150
self.assertEqual(len(self.panel._queries), 0)
147151

148152
async def test_ddt_models_not_tracked_async(self):
153+
"""(async) Tests whether DDt models are not being tracked when the `TRACK_DDT_MODELS` is set to False"""
149154
self.assertEqual(len(self.panel._queries), 0)
150155

151156
await async_sql_call_ddt()

0 commit comments

Comments
 (0)