Fix documentation of Queryset.update regarding full_result #1995

2019-02-16 21:54:05 +01:00
parent a657f29439
commit cab659dce6
3 changed files with 29 additions and 6 deletions
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@@ -2233,6 +2233,19 @@ class QuerySetTest(unittest.TestCase):
        bar.reload()
        self.assertEqual(len(bar.foos), 0)

+    def test_update_one_check_return_with_full_result(self):
+        class BlogTag(Document):
+            name = StringField(required=True)
+
+        BlogTag.drop_collection()
+
+        BlogTag(name='garbage').save()
+        default_update = BlogTag.objects.update_one(name='new')
+        self.assertEqual(default_update, 1)
+
+        full_result_update = BlogTag.objects.update_one(name='new', full_result=True)
+        self.assertIsInstance(full_result_update, UpdateResult)
+
    def test_update_one_pop_generic_reference(self):

        class BlogTag(Document):