From ba99190f535ba16ed3c9d6fe5acfc4836c64b48d Mon Sep 17 00:00:00 2001
From: Erdenezul Batmunkh <erdenezul@shine.mn>
Date: Sun, 10 Sep 2017 13:09:20 +0900
Subject: [PATCH] add tests to increase coverage

---
 tests/fields/fields.py     | 26 ++++++++++++++++++++++++++
 tests/queryset/queryset.py |  4 ++++
 2 files changed, 30 insertions(+)

diff --git a/tests/fields/fields.py b/tests/fields/fields.py
index dbf148e1..d25f2e25 100644
--- a/tests/fields/fields.py
+++ b/tests/fields/fields.py
@@ -312,6 +312,27 @@ class FieldTest(MongoDBTestCase):
 
         self.assertEqual(1, TestDocument.objects(long_fld__ne=None).count())
 
+    def test_callable_validation(self):
+        """Ensure that callable validation works"""
+        def check_even(value):
+            return value % 2 == 0
+
+        class Order(Document):
+            number = IntField(validation=check_even)
+
+        Order.drop_collection()
+
+        order = Order(number=3)
+        self.assertRaises(ValidationError, order.validate)
+
+        class User(Document):
+            name = StringField(validation=1)
+
+        User.drop_collection()
+
+        user = User(name='test')
+        self.assertRaises(ValidationError, user.validate)
+
     def test_object_id_validation(self):
         """Ensure that invalid values cannot be assigned to an
         ObjectIdField.
@@ -527,6 +548,11 @@ class FieldTest(MongoDBTestCase):
             class User(Document):
                 name = StringField(db_field='name\0')
 
+        # db field should be a string
+        with self.assertRaises(TypeError):
+            class User(Document):
+                name = StringField(db_field=1)
+
     def test_decimal_comparison(self):
         class Person(Document):
             money = DecimalField()
diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py
index 5fc73a54..9c61513a 100644
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@@ -1861,6 +1861,10 @@ class QuerySetTest(unittest.TestCase):
         post = BlogPost(name="Test Post", hits=5, tags=['test'])
         post.save()
 
+        BlogPost.objects.update(hits=11)
+        post.reload()
+        self.assertEqual(post.hits, 11)
+
         BlogPost.objects.update(set__hits=10)
         post.reload()
         self.assertEqual(post.hits, 10)