Fixed #37036 -- Added missing flat=True arg in DeferredAttribute.fetch_many().

2 files changed, 27 insertions, 0 deletions

diff --git a/tests/defer/tests.py b/tests/defer/tests.py
index 1a5f333fa1..ea7703c1e3 100644
--- a/tests/defer/tests.py
+++ b/tests/defer/tests.py
@@ -219,6 +219,20 @@ class DeferTests(AssertionMixin, TestCase):
         with self.assertNumQueries(1):
             p1.value
 
+    def test_defer_fk_fetch_mode_fetch_peers(self):
+        p1, p2 = Primary.objects.fetch_mode(FETCH_PEERS).defer("related")
+        with self.assertNumQueries(2):
+            self.assertEqual(p1.related, self.s1)
+        with self.assertNumQueries(0):
+            self.assertEqual(p2.related, self.s1)
+
+    def test_only_fk_fetch_mode_fetch_peers(self):
+        p1, p2 = Primary.objects.fetch_mode(FETCH_PEERS).only("name")
+        with self.assertNumQueries(2):
+            self.assertEqual(p1.related, self.s1)
+        with self.assertNumQueries(0):
+            self.assertEqual(p2.related, self.s1)
+
     def test_only_fetch_mode_raise(self):
         p1 = Primary.objects.fetch_mode(RAISE).only("name").get(name="p1")
         msg = "Fetching of Primary.value blocked."
diff --git a/tests/many_to_one/tests.py b/tests/many_to_one/tests.py
index e161686906..d6149d521f 100644
--- a/tests/many_to_one/tests.py
+++ b/tests/many_to_one/tests.py
@@ -1002,3 +1002,16 @@ class ManyToOneTests(TestCase):
             a2._state.fetch_mode,
             FETCH_PEERS,
         )
+
+    def test_fetch_mode_fetch_peers_reverse_with_deferred_fk(self):
+        Article.objects.create(
+            headline="Another article",
+            pub_date=datetime.date(2005, 7, 27),
+            reporter=self.r,
+        )
+        r = Reporter.objects.fetch_mode(FETCH_PEERS).get(pk=self.r.pk)
+        a1, a2 = r.article_set.defer("reporter")
+        with self.assertNumQueries(2):
+            self.assertEqual(a1.reporter, self.r)
+        with self.assertNumQueries(0):
+            self.assertEqual(a2.reporter, self.r)