[soc2010/query-refactor] Implemented __gt.

alex · alex · commit 33523c9f95d4 · 2010-06-22T18:08:47.000Z
git-svn-id: http://code.djangoproject.com/svn/django/branches/soc2010/query-refactor@13381 bcc190cf-cafb-0310-a4f2-bffc1f526a37
diff --git a/django/contrib/mongodb/compiler.py b/django/contrib/mongodb/compiler.py
@@ -9,6 +9,7 @@ class SQLCompiler(object):
         "exact": lambda params, value_annotation, negated: params[0],
         "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]},
     }
     
     def __init__(self, query, connection, using):
diff --git a/tests/regressiontests/mongodb/tests.py b/tests/regressiontests/mongodb/tests.py
@@ -224,3 +224,35 @@ def test_isnull(self):
             ],
             lambda g: g.name
         )
+    
+    def test_gt(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__gt=1970), [
+                "Queen",
+                "The E Street Band",
+            ],
+            lambda g: g.name
+        )
+        
+        self.assertQuerysetEqual(
+            Group.objects.filter(year_formed__gt=1971), [
+                "The E Street Band",
+            ],
+            lambda g: g.name
+        )
+        
+        self.assertQuerysetEqual(
+            Group.objects.filter(year_formed__gt=1972),
+            [],
+            lambda g: g.name
+        )
+        
+        self.assertQuerysetEqual(
+            Group.objects.exclude(year_formed__gt=1971), [
+                "Queen",
+            ],
+            lambda g: g.name,
+        )

Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,7 @@ class SQLCompiler(object):`
`9`	`9`	`"exact": lambda params, value_annotation, negated: params[0],`
`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`	`+ "gt": lambda params, value_annotation, negated: {"$gt": params[0]},`
`12`	`13`	`}`
`13`	`14`
`14`	`15`	`def __init__(self, query, connection, using):`