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()