Fixed GenericReferenceField query issue

Author: hmarr

mongoengine/fields.py

@@ -509,7 +509,7 @@ def to_mongo(self, document):
         return {'_cls': document.__class__.__name__, '_ref': ref}
 
     def prepare_query_value(self, op, value):
-        return self.to_mongo(value)['_ref']
+        return self.to_mongo(value)
 
 
 class BinaryField(BaseField):

mongoengine/queryset.py

@@ -624,9 +624,6 @@ def _transform_query(cls, _doc_cls=None, **query):
                     # 'in', 'nin' and 'all' require a list of values
                     value = [field.prepare_query_value(op, v) for v in value]
 
-                if field.__class__.__name__ == 'GenericReferenceField':
-                    parts.append('_ref')
-
             # if op and op not in match_operators:
             if op:
                 if op in geo_operators:

tests/fields.py

@@ -545,7 +545,6 @@ class User(Document):
         user.save()
 
         user = User.objects(bookmarks__all=[post_1, link_1]).first()
-        print User.objects(bookmarks__all=[post_1, link_1]).explain()
 
         self.assertEqual(user.bookmarks[0], post_1)
         self.assertEqual(user.bookmarks[1], link_1)