Add related field test

Author: WaVEV

tests/model_fields_/models.py

@@ -194,3 +194,11 @@ class MuseumExhibit(models.Model):
 
     def __str__(self):
         return self.exhibit_name
+
+
+class Tour(models.Model):
+    guide = models.CharField(max_length=100)
+    exhibit = models.ForeignKey(MuseumExhibit, on_delete=models.CASCADE)
+
+    def __str__(self):
+        return f"Tour by {self.guide}"

tests/model_fields_/test_embedded_model_array.py

@@ -15,6 +15,7 @@
     MuseumExhibit,
     RestorationRecord,
     Review,
+    Tour,
 )
 
 
@@ -150,6 +151,9 @@ def setUpTestData(cls):
                 ],
             ),
         )
+        cls.egypt_tour = Tour.objects.create(guide="Amira", exhibit=cls.egypt)
+        cls.wonders_tour = Tour.objects.create(guide="Carlos", exhibit=cls.wonders)
+        cls.lost_tour = Tour.objects.create(guide="Yelena", exhibit=cls.lost_empires)
 
     def test_filter_with_field(self):
         self.assertCountEqual(
@@ -296,6 +300,15 @@ def test_slice(self):
             MuseumExhibit.objects.filter(sections__0_1__section_number=2), [self.new_descoveries]
         )
 
+    def test_foreign_field_exact(self):
+        qs = Tour.objects.filter(exhibit__sections__section_number=1)
+        self.assertCountEqual(qs, [self.egypt_tour, self.wonders_tour])
+
+    def test_foreign_field_with_slice(self):
+        # Only wonders exhibit has exactly two sections, and this slice matches first two
+        qs = Tour.objects.filter(exhibit__sections__0_2__section_number__all=[1, 2])
+        self.assertEqual(list(qs), [self.wonders_tour])
+
 
 @isolate_apps("model_fields_")
 class CheckTests(SimpleTestCase):