diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/composite_pk/tests.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tests/composite_pk/tests.py b/tests/composite_pk/tests.py index 1a28a585fa..2245a472e4 100644 --- a/tests/composite_pk/tests.py +++ b/tests/composite_pk/tests.py @@ -147,6 +147,22 @@ class CompositePKTests(TestCase): result = Comment.objects.in_bulk([self.comment.pk]) self.assertEqual(result, {self.comment.pk: self.comment}) + @unittest.mock.patch.object( + type(connection.features), "max_query_params", new_callable=lambda: 10 + ) + def test_in_bulk_batching(self, mocked_max_query_params): + Comment.objects.all().delete() + num_requiring_batching = (connection.features.max_query_params // 2) + 1 + comments = [ + Comment(id=i, tenant=self.tenant, user=self.user) + for i in range(1, num_requiring_batching + 1) + ] + Comment.objects.bulk_create(comments) + id_list = list(Comment.objects.values_list("pk", flat=True)) + with self.assertNumQueries(2): + comment_dict = Comment.objects.in_bulk(id_list=id_list) + self.assertQuerySetEqual(comment_dict, id_list) + def test_iterator(self): """ Test the .iterator() method of composite_pk models. |