Added test for upsert & update_one #336

2013-06-03 13:05:52 +00:00 · 2013-06-03 13:05:52 +00:00 · b2f78fadd9
commit b2f78fadd9
parent 2fe1c20475
1 changed files with 9 additions and 0 deletions
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@ -545,6 +545,15 @@ class QuerySetTest(unittest.TestCase):
        self.assertEqual("Bob", bob.name)
        self.assertEqual(30, bob.age)

+    def test_upsert_one(self):
+        self.Person.drop_collection()
+
+        self.Person.objects(name="Bob", age=30).update_one(upsert=True)
+
+        bob = self.Person.objects.first()
+        self.assertEqual("Bob", bob.name)
+        self.assertEqual(30, bob.age)
+
    def test_set_on_insert(self):
        self.Person.drop_collection()