Add unit test

Author: WaVEV

tests/model_fields_/models.py

@@ -126,14 +126,12 @@ class Address(EmbeddedModel):
     city = models.CharField(max_length=20)
     state = models.CharField(max_length=2)
     zip_code = models.IntegerField(db_index=True)
-    tags = ArrayField(models.CharField(max_length=100), null=True, blank=True)
 
 
 class Author(EmbeddedModel):
     name = models.CharField(max_length=10)
     age = models.IntegerField()
     address = EmbeddedModelField(Address)
-    skills = ArrayField(models.CharField(max_length=100), null=True, blank=True)
 
 
 class Book(models.Model):
@@ -165,3 +163,28 @@ class Movie(models.Model):
 
     def __str__(self):
         return self.title
+
+
+class ArtifactDetail(EmbeddedModel):
+    """Details about a specific artifact."""
+
+    name = models.CharField(max_length=255)
+    description = models.CharField(max_length=255)
+    metadata = models.JSONField()
+
+
+class ExhibitSection(EmbeddedModel):
+    """A section within an exhibit, containing multiple artifacts."""
+
+    section_number = models.IntegerField()
+    artifacts = EmbeddedModelArrayField(ArtifactDetail, null=True)
+
+
+class MuseumExhibit(models.Model):
+    """An exhibit in the museum, composed of multiple sections."""
+
+    exhibit_name = models.CharField(max_length=255)
+    sections = EmbeddedModelArrayField(ExhibitSection, null=True)
+
+    def __str__(self):
+        return self.exhibit_name

tests/model_fields_/test_embedded_model.py

@@ -19,12 +19,15 @@
 
 from .models import (
     Address,
+    ArtifactDetail,
     Author,
     Book,
     Data,
+    ExhibitSection,
     Holder,
     Library,
     Movie,
+    MuseumExhibit,
     NestedData,
     Review,
 )
@@ -142,6 +145,59 @@ def setUpTestData(cls):
             Review(title="Classic", rating=7),
         ]
         cls.bears = Movie.objects.create(title="Bears", reviews=reviews)
+        cls.egypt = MuseumExhibit.objects.create(
+            exhibit_name="Ancient Egypt",
+            sections=[
+                ExhibitSection(
+                    section_number=1,
+                    artifacts=[
+                        ArtifactDetail(
+                            name="Ptolemaic Crown",
+                            description="Royal headpiece worn by Ptolemy kings.",
+                            metadata={
+                                "material": "gold",
+                                "origin": "Egypt",
+                                "era": "Ptolemaic Period",
+                            },
+                        )
+                    ],
+                )
+            ],
+        )
+        cls.wonders = MuseumExhibit.objects.create(
+            exhibit_name="Wonders of the Ancient World",
+            sections=[
+                ExhibitSection(
+                    section_number=1,
+                    artifacts=[
+                        ArtifactDetail(
+                            name="Statue of Zeus",
+                            description="One of the Seven Wonders, created by Phidias.",
+                            metadata={"location": "Olympia", "height_m": 12},
+                        ),
+                        ArtifactDetail(
+                            name="Hanging Gardens",
+                            description="Legendary gardens of Babylon.",
+                            metadata={"debated_existence": True},
+                        ),
+                    ],
+                ),
+                ExhibitSection(
+                    section_number=2,
+                    artifacts=[
+                        ArtifactDetail(
+                            name="Lighthouse of Alexandria",
+                            description="Guided sailors safely to port.",
+                            metadata={"height_m": 100, "built": "3rd century BC"},
+                        )
+                    ],
+                ),
+            ],
+        )
+        cls.new_descoveries = MuseumExhibit.objects.create(
+            exhibit_name="New Discoveries",
+            sections=[ExhibitSection(section_number=1, artifacts=[])],
+        )
 
     def test_filter_with_field(self):
         self.assertCountEqual(
@@ -154,6 +210,12 @@ def test_filter_with_model(self):
             [self.clouds, self.frozen],
         )
 
+    def test_filter_with_embeddedfield_path(self):
+        self.assertCountEqual(
+            MuseumExhibit.objects.filter(sections__0__section_number=1),
+            [self.egypt, self.wonders, self.new_descoveries],
+        )
+
 
 class QueryingTests(TestCase):
     @classmethod
@@ -273,56 +335,6 @@ def test_nested(self):
         self.assertCountEqual(Book.objects.filter(author__address__city="NYC"), [obj])
 
 
-class ArrayFieldTests(TestCase):
-    @classmethod
-    def setUpTestData(cls):
-        cls.book = Book.objects.create(
-            author=Author(
-                name="Shakespeare",
-                age=55,
-                skills=["writing", "editing"],
-                address=Address(city="NYC", state="NY", tags=["home", "shipping"]),
-            ),
-        )
-
-    def test_contains(self):
-        self.assertCountEqual(Book.objects.filter(author__skills__contains=["nonexistent"]), [])
-        self.assertCountEqual(
-            Book.objects.filter(author__skills__contains=["writing"]), [self.book]
-        )
-        # Nested
-        self.assertCountEqual(
-            Book.objects.filter(author__address__tags__contains=["nonexistent"]), []
-        )
-        self.assertCountEqual(
-            Book.objects.filter(author__address__tags__contains=["home"]), [self.book]
-        )
-
-    def test_contained_by(self):
-        self.assertCountEqual(
-            Book.objects.filter(author__skills__contained_by=["writing", "publishing"]), []
-        )
-        self.assertCountEqual(
-            Book.objects.filter(author__skills__contained_by=["writing", "editing", "publishing"]),
-            [self.book],
-        )
-        # Nested
-        self.assertCountEqual(
-            Book.objects.filter(author__address__tags__contained_by=["home", "work"]), []
-        )
-        self.assertCountEqual(
-            Book.objects.filter(author__address__tags__contained_by=["home", "work", "shipping"]),
-            [self.book],
-        )
-
-    def test_len(self):
-        self.assertCountEqual(Book.objects.filter(author__skills__len=1), [])
-        self.assertCountEqual(Book.objects.filter(author__skills__len=2), [self.book])
-        # Nested
-        self.assertCountEqual(Book.objects.filter(author__address__tags__len=1), [])
-        self.assertCountEqual(Book.objects.filter(author__address__tags__len=2), [self.book])
-
-
 class InvalidLookupTests(SimpleTestCase):
     def test_invalid_field(self):
         msg = "Author has no field named 'first_name'"