feat: improved unit test coverage

hivesolutions · Mar 27, 2021 · 1a6b8a4 · 1a6b8a4
1 parent 74d6499
commit 1a6b8a4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/quorum/test/model.py b/src/quorum/test/model.py
@@ -149,19 +149,22 @@ def test_advance(self):
         result = mock.Person.count()
         self.assertEqual(result, 1)
 
-        person.advance("age")
+        result = person.advance("age")
+        self.assertEqual(result, 2)
         self.assertEqual(person.age, 2)
 
         person = person.reload()
         self.assertEqual(person.age, 2)
 
-        person.advance("age", delta = 2)
+        result = person.advance("age", delta = 2)
+        self.assertEqual(result, 4)
         self.assertEqual(person.age, 4)
 
         person = person.reload()
         self.assertEqual(person.age, 4)
 
-        person.advance("age", delta = -2)
+        result = person.advance("age", delta = -2)
+        self.assertEqual(result, 2)
         self.assertEqual(person.age, 2)
 
         person = person.reload()