add nested test

Author: timgraham

tests/model_fields_/models.py

@@ -239,6 +239,7 @@ class Dog(EmbeddedModel):
     barks = models.BooleanField(default=True)
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
+    toys = PolymorphicEmbeddedModelField(["Bone"], blank=True, null=True)
 
     def __str__(self):
         return self.name
@@ -248,6 +249,15 @@ class Cat(EmbeddedModel):
     name = models.CharField(max_length=100)
     purs = models.BooleanField(default=True)
     weight = models.DecimalField(max_digits=4, decimal_places=2, blank=True, null=True)
+    toys = PolymorphicEmbeddedModelField(["Mouse"], blank=True, null=True)
 
     def __str__(self):
         return self.name
+
+
+class Bone(EmbeddedModel):
+    brand = models.CharField(max_length=100)
+
+
+class Mouse(EmbeddedModel):
+    manufacturer = models.CharField(max_length=100)

tests/model_fields_/test_polymorphic_embedded_model.py

@@ -9,7 +9,7 @@
 from django_mongodb_backend.fields import PolymorphicEmbeddedModelField
 from django_mongodb_backend.models import EmbeddedModel
 
-from .models import Cat, Dog, Library, Person
+from .models import Bone, Cat, Dog, Library, Mouse, Person
 from .utils import truncate_ms
 
 
@@ -103,6 +103,7 @@ def setUpTestData(cls):
                 pet=Cat(
                     name=f"Cat {x}",
                     weight=f"{x}.5",
+                    toys=Mouse(manufacturer=f"Maker {x}"),
                 ),
             )
             for x in range(6)
@@ -113,6 +114,7 @@ def setUpTestData(cls):
                 pet=Dog(
                     name=f"Dog {x}",
                     barks=x % 2 == 0,
+                    toys=Bone(brand=f"Brand {x}"),
                 ),
             )
             for x in range(6)
@@ -148,6 +150,19 @@ def test_boolean(self):
             [x for i, x in enumerate(self.dog_owners) if i % 2 == 0],
         )
 
+    def test_nested(self):
+        # Cat and Dog both have field toys = PolymorphicEmbeddedModelField(...)
+        # but with different models. It's possible to query the fields of the
+        # Dog's toys because it's the first model in Person.pet.
+        self.assertCountEqual(
+            Person.objects.filter(pet__toys__brand="Brand 1"),
+            [self.dog_owners[1]],
+        )
+        # The fields of Cat can't be queried.
+        msg = "The models of field 'toys' have no field named 'manufacturer'."
+        with self.assertRaisesMessage(FieldDoesNotExist, msg):
+            (Person.objects.filter(pet__toys__manufacturer="Maker 1"),)
+
 
 class InvalidLookupTests(SimpleTestCase):
     def test_invalid_field(self):