Correct EmbeddedModelField.db_type() to "object"

"embeddedDocuments" isn't a BSON type.

Author: timgraham

django_mongodb_backend/fields/embedded_model.py

@@ -23,7 +23,7 @@ def __init__(self, embedded_model, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def db_type(self, connection):
-        return "embeddedDocuments"
+        return "object"
 
     def check(self, **kwargs):
         from ..models import EmbeddedModel  # noqa: PLC0415

django_mongodb_backend/fields/polymorphic_embedded_model.py

@@ -25,7 +25,7 @@ def __init__(self, embedded_models, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def db_type(self, connection):
-        return "embeddedDocuments"
+        return "object"
 
     def check(self, **kwargs):
         from ..models import EmbeddedModel  # noqa: PLC0415

django_mongodb_backend/indexes.py

@@ -141,7 +141,7 @@ def search_index_data_types(self, db_type):
             return "boolean"
         if db_type == "object":
             return "document"
-        if db_type == "array":
+        if db_type in {"array", "object"}:
             return "embeddedDocuments"
         return db_type
 

tests/model_fields_/test_embedded_model.py

@@ -30,6 +30,9 @@
 
 
 class MethodTests(SimpleTestCase):
+    def test_db_type(self):
+        self.assertEqual(EmbeddedModelField("Data").db_type(None), "object")
+
     def test_deconstruct(self):
         field = EmbeddedModelField("Data", null=True)
         field.name = "field_name"

tests/model_fields_/test_polymorphic_embedded_model.py

@@ -18,6 +18,9 @@ def test_not_editable(self):
         field = PolymorphicEmbeddedModelField(["Data"], null=True)
         self.assertIs(field.editable, False)
 
+    def test_db_type(self):
+        self.assertEqual(PolymorphicEmbeddedModelField(["Data"]).db_type(None), "object")
+
     def test_deconstruct(self):
         field = PolymorphicEmbeddedModelField(["Data"], null=True)
         field.name = "field_name"