From b2f78fadd92823120ea73b3ed27ebac7585e1fcb Mon Sep 17 00:00:00 2001
From: Ross Lawley <ross.lawley@gmail.com>
Date: Mon, 3 Jun 2013 13:05:52 +0000
Subject: [PATCH] Added test for upsert & update_one #336

---
 tests/queryset/queryset.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py
index 01c53d04..54257410 100644
--- 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()