summaryrefslogtreecommitdiff
path: root/tests/test_utils/tests.py
diff options
context:
space:
mode:
authorJon Dufresne <jon.dufresne@gmail.com>2020-02-03 19:07:00 -0800
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2020-02-06 15:35:23 +0100
commit3259983f569151232d8e3b0c3d0de3a858c2b265 (patch)
tree4f1f19368fbb9d73ea5245d8fabe5e3097f51926 /tests/test_utils/tests.py
parentf48f671223a20b161ca819cf7d6298e43b8ba5fe (diff)
Fixed #31233 -- Closed database connections and cursors after use.
Diffstat (limited to 'tests/test_utils/tests.py')
-rw-r--r--tests/test_utils/tests.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/test_utils/tests.py b/tests/test_utils/tests.py
index 32007ebeb2..c61358f4c8 100644
--- a/tests/test_utils/tests.py
+++ b/tests/test_utils/tests.py
@@ -226,7 +226,8 @@ class AssertNumQueriesUponConnectionTests(TransactionTestCase):
if is_opening_connection:
# Avoid infinite recursion. Creating a cursor calls
# ensure_connection() which is currently mocked by this method.
- connection.cursor().execute('SELECT 1' + connection.features.bare_select_suffix)
+ with connection.cursor() as cursor:
+ cursor.execute('SELECT 1' + connection.features.bare_select_suffix)
ensure_connection = 'django.db.backends.base.base.BaseDatabaseWrapper.ensure_connection'
with mock.patch(ensure_connection, side_effect=make_configuration_query):