Improved isolation of TestGeneralAggregate.test_default_argument().

This commit is contained in:
Simon Charette 2022-12-02 13:57:40 +01:00 committed by Mariusz Felisiak
parent 3e3b7f691b
commit eccda63a49
1 changed files with 2 additions and 2 deletions

View File

@ -1,4 +1,4 @@
from django.db import connection
from django.db import connection, transaction
from django.db.models import (
CharField,
F,
@ -147,7 +147,7 @@ class TestGeneralAggregate(PostgreSQLTestCase):
)
self.assertEqual(values, {"aggregation": expected_result})
# Empty result when query must be executed.
with self.assertNumQueries(1):
with transaction.atomic(), self.assertNumQueries(1):
values = AggregateTestModel.objects.aggregate(
aggregation=aggregation,
)