Fix unit test

Author: WaVEV

django_mongodb_backend/fields/embedded_model.py

@@ -186,9 +186,11 @@ def as_mql(self, compiler, connection):
             key_transforms.insert(0, previous.key_name)
             previous = previous.lhs
         mql = previous.as_mql(compiler, connection)
+        # transform = ".".join(key_transforms)
         for key in key_transforms:
             mql = {"$getField": {"input": mql, "field": key}}
         return mql
+        # return f"{mql}.{transform}"
 
     @property
     def output_field(self):

django_mongodb_backend/fields/embedded_model_array.py

@@ -70,16 +70,20 @@ def as_mql(self, compiler, connection):
         lhs_mql = process_lhs(self, compiler, connection)
         value = process_rhs(self, compiler, connection)
         if isinstance(self.lhs, Col | KeyTransform):
+            if isinstance(self.lhs, Col):
+                inner_lhs_mql = "$$item"
+            else:
+                lhs_mql, inner_lhs_mql = lhs_mql
             if isinstance(value, models.Model):
                 value, emf_data = self.model_to_dict(value)
                 # Get conditions for any nested EmbeddedModelFields.
-                conditions = self.get_conditions({lhs_mql[1]: (value, emf_data)})
+                conditions = self.get_conditions({inner_lhs_mql: (value, emf_data)})
                 return {
                     "$anyElementTrue": {
                         "$ifNull": [
                             {
                                 "$map": {
-                                    "input": lhs_mql[0],
+                                    "input": lhs_mql,
                                     "as": "item",
                                     "in": {"$and": conditions},
                                 }
@@ -93,9 +97,9 @@ def as_mql(self, compiler, connection):
                     "$ifNull": [
                         {
                             "$map": {
-                                "input": lhs_mql[0],
+                                "input": lhs_mql,
                                 "as": "item",
-                                "in": {"$eq": [lhs_mql[1], value]},
+                                "in": {"$eq": [inner_lhs_mql, value]},
                             }
                         },
                         [],
@@ -149,10 +153,7 @@ def get_transform(self, name):
         )
 
     def as_mql(self, compiler, connection):
-        if isinstance(self._lhs, Transform):
-            inner_lhs_mql = self._lhs.as_mql(compiler, connection)
-        else:
-            inner_lhs_mql = None
+        inner_lhs_mql = self._lhs.as_mql(compiler, connection)
         lhs_mql = process_lhs(self, compiler, connection)
         return lhs_mql, inner_lhs_mql