Add more unit test

Author: WaVEV

tests/model_fields_/models.py

@@ -192,12 +192,20 @@ def __str__(self):
         return self.title
 
 
+class RestorationRecord(EmbeddedModel):
+    date = models.DateField()
+    description = models.TextField()
+    restored_by = models.CharField(max_length=255)
+
+
 class ArtifactDetail(EmbeddedModel):
     """Details about a specific artifact."""
 
     name = models.CharField(max_length=255)
     description = models.CharField(max_length=255)
     metadata = models.JSONField()
+    restorations = EmbeddedModelArrayField(RestorationRecord, null=True)
+    last_restoration = EmbeddedModelField(RestorationRecord, null=True)
 
 
 class ExhibitSection(EmbeddedModel):
@@ -207,11 +215,17 @@ class ExhibitSection(EmbeddedModel):
     artifacts = EmbeddedModelArrayField(ArtifactDetail, null=True)
 
 
+class ExhibitMeta(EmbeddedModel):
+    curator_name = models.CharField(max_length=255)
+    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)
+    meta = EmbeddedModelField(ExhibitMeta, null=True)
 
     def __str__(self):
         return self.exhibit_name

tests/model_fields_/test_embedded_model.py

@@ -1,5 +1,5 @@
 import operator
-from datetime import timedelta
+from datetime import date, timedelta
 
 from django.core.exceptions import FieldDoesNotExist, ValidationError
 from django.db import models
@@ -24,12 +24,14 @@
     Author,
     Book,
     Data,
+    ExhibitMeta,
     ExhibitSection,
     Holder,
     Library,
     Movie,
     MuseumExhibit,
     NestedData,
+    RestorationRecord,
     Review,
 )
 from .utils import truncate_ms
@@ -199,6 +201,36 @@ def setUpTestData(cls):
             exhibit_name="New Discoveries",
             sections=[ExhibitSection(section_number=1, artifacts=[])],
         )
+        cls.lost_empires = MuseumExhibit.objects.create(
+            exhibit_name="Lost Empires",
+            meta=ExhibitMeta(
+                curator_name="Dr. Amina Hale",
+                artifacts=[
+                    ArtifactDetail(
+                        name="Bronze Statue",
+                        description="Statue from the Hellenistic period.",
+                        metadata={"origin": "Pergamon", "material": "bronze"},
+                        restorations=[
+                            RestorationRecord(
+                                date=date(1998, 4, 15),
+                                description="Removed oxidized layer.",
+                                restored_by="Restoration Lab A",
+                            ),
+                            RestorationRecord(
+                                date=date(2010, 7, 22),
+                                description="Reinforced the base structure.",
+                                restored_by="Dr. Liu Cheng",
+                            ),
+                        ],
+                        last_restoration=RestorationRecord(
+                            date=date(2010, 7, 22),
+                            description="Reinforced the base structure.",
+                            restored_by="Dr. Liu Cheng",
+                        ),
+                    )
+                ],
+            ),
+        )
 
     def test_filter_with_field(self):
         self.assertCountEqual(
@@ -217,6 +249,14 @@ def test_filter_with_embeddedfield_path(self):
             [self.egypt, self.wonders, self.new_descoveries],
         )
 
+    def test_filter_with_embeddedfield_array_path(self):
+        self.assertCountEqual(
+            MuseumExhibit.objects.filter(
+                meta__artifacts__restorations__0__restored_by="Restoration Lab A"
+            ),
+            [self.lost_empires],
+        )
+
 
 class QueryingTests(TestCase):
     @classmethod