diff options
Diffstat (limited to 'tests/test_utils/tests.py')
| -rw-r--r-- | tests/test_utils/tests.py | 3 |
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): |