[soc2010/query-refactor] Implemented __in.

alex · alex · commit 3efa6195411f · 2010-06-22T18:09:25.000Z
git-svn-id: http://code.djangoproject.com/svn/django/branches/soc2010/query-refactor@13384 bcc190cf-cafb-0310-a4f2-bffc1f526a37
diff --git a/django/contrib/mongodb/compiler.py b/django/contrib/mongodb/compiler.py
@@ -10,6 +10,7 @@ class SQLCompiler(object):
         "lt": lambda params, value_annotation, negated: {"$lt": params[0]},
         "isnull": lambda params, value_annotation, negated: {"$ne": None} if value_annotation == negated else None,
         "gt": lambda params, value_annotation, negated: {"$gt": params[0]},
+        "in": lambda params, value_annotation, negated: {"$in": params},
     }
     
     def __init__(self, query, connection, using):
diff --git a/tests/regressiontests/mongodb/tests.py b/tests/regressiontests/mongodb/tests.py
@@ -258,3 +258,29 @@ def test_gt(self):
             ],
             lambda g: g.name,
         )
+    
+    def test_in(self):
+        q = Group.objects.create(name="Queen", year_formed=1971)
+        e = Group.objects.create(name="The E Street Band", year_formed=1972)
+        
+        self.assertQuerysetEqual(
+            Group.objects.filter(year_formed__in=[1972]), [
+                "The E Street Band",
+            ],
+            lambda g: g.name,
+        )
+        
+        self.assertQuerysetEqual(
+            Group.objects.filter(year_formed__in=[1972, 1971]), [
+                "Queen",
+                "The E Street Band",
+            ],
+            lambda g: g.name
+        )
+        
+        self.assertQuerysetEqual(
+            Group.objects.exclude(year_formed__in=[1972]), [
+                "Queen",
+            ],
+            lambda g: g.name,
+        )

Original file line number	Diff line number	Diff line change
`@@ -10,6 +10,7 @@ class SQLCompiler(object):`
`10`	`10`	`"lt": lambda params, value_annotation, negated: {"$lt": params[0]},`
`11`	`11`	`"isnull": lambda params, value_annotation, negated: {"$ne": None} if value_annotation == negated else None,`
`12`	`12`	`"gt": lambda params, value_annotation, negated: {"$gt": params[0]},`
	`13`	`+ "in": lambda params, value_annotation, negated: {"$in": params},`
`13`	`14`	`}`
`14`	`15`
`15`	`16`	`def __init__(self, query, connection, using):`