[dak/master] ValidatorTestCase: test before_update, too.
Signed-off-by: Torsten Werner <twerner@debian.org>
---
tests/dbtest_validation.py | 24 ++++++++++--------------
1 files changed, 10 insertions(+), 14 deletions(-)
diff --git a/tests/dbtest_validation.py b/tests/dbtest_validation.py
index 597097c..32f47eb 100755
--- a/tests/dbtest_validation.py
+++ b/tests/dbtest_validation.py
@@ -12,26 +12,22 @@ class ValidatorTestCase(DBDakTestCase):
The ValidatorTestCase tests the validation mechanism.
"""
- def must_fail(self):
- ''''
- This function must fail with DBUpdateError because arch_string is not
- set. It rolls back the transaction before re-raising the exception.
- '''
- try:
- architecture = Architecture()
- self.session.add(architecture)
- self.session.flush()
- except:
- self.session.rollback()
- raise
-
def test_validation(self):
'tests validate()'
- self.assertRaises(DBUpdateError, self.must_fail)
+
+ # before_insert validation should fail
+ architecture = Architecture()
+ self.session.add(architecture)
+ self.assertRaises(DBUpdateError, self.session.flush)
+ self.session.rollback()
# should not fail
architecture = Architecture('i386')
self.session.add(architecture)
self.session.flush()
+ # before_update validation should fail
+ architecture.arch_string = ''
+ self.assertRaises(DBUpdateError, self.session.flush)
+ self.session.rollback()
if __name__ == '__main__':
unittest.main()
--
1.5.6.5
Reply to: