From 62ca2dea04489bdb23df1c7815439287c6d44630 Mon Sep 17 00:00:00 2001 From: Bryan Marty Date: Thu, 29 Oct 2015 23:24:46 -0700 Subject: Fixed #8065 -- Made id_list an optional argument for QuerySet.in_bulk(). --- tests/lookup/tests.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) (limited to 'tests') diff --git a/tests/lookup/tests.py b/tests/lookup/tests.py index 7e9f0921f8..203f25406a 100644 --- a/tests/lookup/tests.py +++ b/tests/lookup/tests.py @@ -116,6 +116,18 @@ class LookupTests(TestCase): arts = Article.objects.in_bulk([self.a1.id, self.a2.id]) self.assertEqual(arts[self.a1.id], self.a1) self.assertEqual(arts[self.a2.id], self.a2) + self.assertEqual( + Article.objects.in_bulk(), + { + self.a1.id: self.a1, + self.a2.id: self.a2, + self.a3.id: self.a3, + self.a4.id: self.a4, + self.a5.id: self.a5, + self.a6.id: self.a6, + self.a7.id: self.a7, + } + ) self.assertEqual(Article.objects.in_bulk([self.a3.id]), {self.a3.id: self.a3}) self.assertEqual(Article.objects.in_bulk({self.a3.id}), {self.a3.id: self.a3}) self.assertEqual(Article.objects.in_bulk(frozenset([self.a3.id])), {self.a3.id: self.a3}) @@ -124,7 +136,6 @@ class LookupTests(TestCase): self.assertEqual(Article.objects.in_bulk([]), {}) self.assertEqual(Article.objects.in_bulk(iter([self.a1.id])), {self.a1.id: self.a1}) self.assertEqual(Article.objects.in_bulk(iter([])), {}) - self.assertRaises(TypeError, Article.objects.in_bulk) self.assertRaises(TypeError, Article.objects.in_bulk, headline__startswith='Blah') def test_values(self): -- cgit v1.3