summaryrefslogtreecommitdiff
path: root/tests/bulk_create
diff options
context:
space:
mode:
authorSage Abdullah <me@laymonage.com>2025-05-08 09:10:00 +0200
committerSarah Boyce <42296566+sarahboyce@users.noreply.github.com>2025-05-08 11:37:11 +0100
commit38660a612cd924199df83b93807604cadf6d5125 (patch)
tree105b175df274349c9f2e9253b5b613ed9548fa41 /tests/bulk_create
parent37f2a77c729ccb71059c8e66c49b07499d2edf60 (diff)
Refs #36143 -- Tested bulk_batch_size limit for bulk_update and bulk_create.
Diffstat (limited to 'tests/bulk_create')
-rw-r--r--tests/bulk_create/tests.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/bulk_create/tests.py b/tests/bulk_create/tests.py
index e1b7e42d14..d590a292de 100644
--- a/tests/bulk_create/tests.py
+++ b/tests/bulk_create/tests.py
@@ -294,6 +294,14 @@ class BulkCreateTests(TestCase):
Country.objects.bulk_create(objs, batch_size=max_batch_size + 1)
@skipUnlessDBFeature("has_bulk_insert")
+ def test_max_batch_size(self):
+ objs = [Country(name=f"Country {i}") for i in range(1000)]
+ fields = ["name", "iso_two_letter", "description"]
+ max_batch_size = connection.ops.bulk_batch_size(fields, objs)
+ with self.assertNumQueries(ceil(len(objs) / max_batch_size)):
+ Country.objects.bulk_create(objs)
+
+ @skipUnlessDBFeature("has_bulk_insert")
def test_bulk_insert_expressions(self):
Restaurant.objects.bulk_create(
[