Skip to content

Commit e16a8c1

Browse files
committed
backends edits
1 parent 8170dc2 commit e16a8c1

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tests/backends/base/test_base.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,6 +393,8 @@ def test_multi_database_init_connection_state_called_once(self):
393393
connections[db],
394394
"check_database_version_supported",
395395
) as mocked_check_database_version_supported:
396+
if connections[db].connection is None:
397+
connections[db].connection.connect()
396398
connections[db].init_connection_state()
397399
after_first_calls = len(
398400
mocked_check_database_version_supported.mock_calls

tests/backends/tests.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def test_last_executed_query_without_previous_query(self):
7979
def test_debug_sql(self):
8080
list(Reporter.objects.filter(first_name="test"))
8181
sql = connection.queries[-1]["sql"].lower()
82-
self.assertIn("select", sql)
82+
self.assertIn("$match", sql)
8383
self.assertIn(Reporter._meta.db_table, sql)
8484

8585
def test_query_encoding(self):
@@ -262,14 +262,12 @@ def receiver(sender, connection, **kwargs):
262262

263263
connection_created.connect(receiver)
264264
connection.close()
265-
with connection.cursor():
266-
pass
265+
connection.connect()
267266
self.assertIs(data["connection"].connection, connection.connection)
268-
267+
connection.close()
269268
connection_created.disconnect(receiver)
270269
data.clear()
271-
with connection.cursor():
272-
pass
270+
connection.connect()
273271
self.assertEqual(data, {})
274272

275273

0 commit comments

Comments
 (0)