From 0a2dbbc58b5ae0df92e5465c1a43d46b71d0adb4 Mon Sep 17 00:00:00 2001
From: Frank Battaglia <frank@Franks-MacBook-Pro.local>
Date: Tue, 12 Nov 2013 16:01:56 -0500
Subject: [PATCH] add tests for mongo query operators

---
 tests/document/instance.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/tests/document/instance.py b/tests/document/instance.py
index 4747c0ed..df6b4fcb 100644
--- a/tests/document/instance.py
+++ b/tests/document/instance.py
@@ -870,6 +870,7 @@ class InstanceTest(unittest.TestCase):
         old_id = w1.save_id
 
         flip(w1)
+        w1.save_id = UUID(2)
         w1.save(save_condition={'save_id':old_id})
         w1.reload()
         self.assertFalse(w1.toggle)
@@ -878,8 +879,20 @@ class InstanceTest(unittest.TestCase):
         flip(w2)
         w2.save(save_condition={'save_id':old_id})
         w2.reload()
-        self.assertFalse(w1.toggle)
-        self.assertEqual(w1.count, 2)
+        self.assertFalse(w2.toggle)
+        self.assertEqual(w2.count, 2)
+
+        # save_condition uses mongoengine-style operator syntax
+        flip(w1)
+        w1.save(save_condition={'count__lt':w1.count})
+        w1.reload()
+        self.assertTrue(w1.toggle)
+        self.assertEqual(w1.count, 3)
+        flip(w1)
+        w1.save(save_condition={'count__gte':w1.count})
+        w1.reload()
+        self.assertTrue(w1.toggle)
+        self.assertEqual(w1.count, 3)
         
 
     def test_update(self):