diff --git a/src/quorum/test/model.py b/src/quorum/test/model.py index de319b9..3772b96 100644 --- 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()