X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdbtest_fingerprint.py;h=8d57ea0f5c7e56e9890be4b3cab5add7b317ff1b;hb=7c5adcf962fe99d6a42307595f263f47d0112fbc;hp=122ea62161374982f7d3e74a83eecee42b74e32f;hpb=0ee7dd7d85c4d43978b9ddaa8de1b951250a60c1;p=dak.git

diff --git a/tests/dbtest_fingerprint.py b/tests/dbtest_fingerprint.py
index 122ea621..8d57ea0f 100755
--- a/tests/dbtest_fingerprint.py
+++ b/tests/dbtest_fingerprint.py
@@ -3,6 +3,7 @@
 from db_test import DBDakTestCase
 
 from daklib.dbconn import Fingerprint, Uid
+from daklib.dak_exceptions import DBUpdateError
 
 from sqlalchemy.exc import IntegrityError
 import unittest
@@ -55,11 +56,11 @@ class FingerprintTestCase(DBDakTestCase):
         self.session.flush()
 
     def test_exceptions(self):
-        self.assertRaises(IntegrityError, self.fingerprint_no_fingerprint)
+        self.assertRaises(DBUpdateError, self.fingerprint_no_fingerprint)
         self.session.rollback()
         self.assertRaises(IntegrityError, self.fingerprint_duplicate_fingerprint)
         self.session.rollback()
-        self.assertRaises(IntegrityError, self.uid_no_uid)
+        self.assertRaises(DBUpdateError, self.uid_no_uid)
         self.session.rollback()
         self.assertRaises(IntegrityError, self.uid_duplicate_uid)
         self.session.rollback()