X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_ormobject.py;h=0790e4c7ad27ad797be29033c845354e3e815254;hb=7c5adcf962fe99d6a42307595f263f47d0112fbc;hp=a5b775535a28683ca2633175ce398d29b51f9ba0;hpb=a6b1715c117b4f306a7d301ecc4492c67c6ef98b;p=dak.git

diff --git a/tests/dbtest_ormobject.py b/tests/dbtest_ormobject.py
index a5b77553..0790e4c7 100755
--- a/tests/dbtest_ormobject.py
+++ b/tests/dbtest_ormobject.py
@@ -3,8 +3,15 @@
 from db_test import DBDakTestCase
 
 from daklib.dbconn import Architecture, Suite
+from daklib.dak_exceptions import DBUpdateError
+
+try:
+    # python >= 2.6
+    import json
+except:
+    # python <= 2.5
+    import simplejson as json
 
-import json
 import re
 import unittest
 
@@ -29,5 +36,10 @@ class ORMObjectTestCase(DBDakTestCase):
         architecture.suites = [sid, squeeze]
         self.assertTrue(re.search('"suites_count": 2', str(architecture)))
 
+    def test_validation(self):
+        suite = Suite()
+        self.session.add(suite)
+        self.assertRaises(DBUpdateError, self.session.flush)
+
 if __name__ == '__main__':
     unittest.main()