From eccda63a49b92ab14d523d69da2a3b3246cf91d1 Mon Sep 17 00:00:00 2001 From: Simon Charette Date: Fri, 2 Dec 2022 13:57:40 +0100 Subject: [PATCH] Improved isolation of TestGeneralAggregate.test_default_argument(). --- tests/postgres_tests/test_aggregates.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/postgres_tests/test_aggregates.py b/tests/postgres_tests/test_aggregates.py index 183cff10c9..b5474d361e 100644 --- a/tests/postgres_tests/test_aggregates.py +++ b/tests/postgres_tests/test_aggregates.py @@ -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, )