Added test for upsert & update_one #336

MongoEngine · Jun 3, 2013 · b2f78fa · b2f78fa
1 parent 2fe1c20
commit b2f78fa
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py
@@ -545,6 +545,15 @@ def test_upsert(self):
         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()