Removed some instances where I as prematurely normalizing field names…

…, causing a failure to resolve a name when db_field is set
dayorbyte · Dec 15, 2010 · 6ce34a7 · 6ce34a7
1 parent 9d27a35
commit 6ce34a7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/mongoalchemy/query_expression.py b/mongoalchemy/query_expression.py
@@ -108,15 +108,15 @@ def in_(self, *values):
             in ``values``.  Produces a MongoDB ``$in`` expression.
         '''
         return QueryExpression({
-            str(self) : { '$in' : [self.get_type().wrap_value(value) for value in values] }
+            self : { '$in' : [self.get_type().wrap_value(value) for value in values] }
         })
 
     def nin(self, *values):
         ''' A query to check if this query field is not one of the values 
             in ``values``.  Produces a MongoDB ``$nin`` expression.
         '''
         return QueryExpression({
-            str(self) : { '$nin' : [self.get_type().wrap_value(value) for value in values] }
+            self : { '$nin' : [self.get_type().wrap_value(value) for value in values] }
         })
 
     def __str__(self):

diff --git a/test/test_query.py b/test/test_query.py
@@ -298,4 +298,13 @@ def qr_test_clone():
         it2.next()
     except StopIteration:
         pass
-
+
+def test_resolve_fields():
+    class Resolver(Document):
+        i = IntField(db_field='j')
+        k = IntField()
+    s = get_session()
+    s.clear_collection(Resolver)
+
+    q = s.query(Resolver).filter(Resolver.i.in_(6))
+