add tests for mongo query operators

This commit is contained in:
Frank Battaglia 2013-11-12 16:01:56 -05:00 committed by Frank Battaglia
parent 673a966541
commit 0a2dbbc58b

View File

@ -870,6 +870,7 @@ class InstanceTest(unittest.TestCase):
old_id = w1.save_id old_id = w1.save_id
flip(w1) flip(w1)
w1.save_id = UUID(2)
w1.save(save_condition={'save_id':old_id}) w1.save(save_condition={'save_id':old_id})
w1.reload() w1.reload()
self.assertFalse(w1.toggle) self.assertFalse(w1.toggle)
@ -878,8 +879,20 @@ class InstanceTest(unittest.TestCase):
flip(w2) flip(w2)
w2.save(save_condition={'save_id':old_id}) w2.save(save_condition={'save_id':old_id})
w2.reload() w2.reload()
self.assertFalse(w1.toggle) self.assertFalse(w2.toggle)
self.assertEqual(w1.count, 2) 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): def test_update(self):