add handling for integrity violations

atramhasis/views/exception_views.py

@@ -6,6 +6,7 @@
 import logging
 
 from pyramid.view import view_config, notfound_view_config
+from sqlalchemy.exc import IntegrityError
 
 from atramhasis.errors import SkosRegistryNotFoundException, ValidationError
 from atramhasis.protected_resources import ProtectedResourceException
@@ -67,6 +68,16 @@ def provider_unavailable(exc, request):
     return {'message': exc.message}
 
 
+@view_config(context=IntegrityError, renderer='json')
+def data_integrity(exc, request):
+    '''
+    View invoked when IntegrityError was raised.
+    '''
+    log.error(exc)
+    request.response.status_int = 409
+    return {'message': 'this operation violates the data integrity and could not be executed '}
+
+
 @view_config(context=Exception, renderer='json')
 def failed(exc, request):
     '''

tests/test_errors.py

@@ -1,7 +1,12 @@
 import unittest
-
+from sqlalchemy.exc import IntegrityError
+try:
+    from unittest.mock import Mock, MagicMock
+except ImportError:
+    from mock import Mock, MagicMock  # pragma: no cover
 from atramhasis.errors import SkosRegistryNotFoundException, ConceptSchemeNotFoundException, \
     ConceptNotFoundException, DbNotFoundException, ValidationError, LanguageNotFoundException
+from atramhasis.views.exception_views import data_integrity
 
 
 class TestErrors(unittest.TestCase):
@@ -33,4 +38,11 @@ def test_db_error(self):
     def test_language_notfound_exception(self):
         error = LanguageNotFoundException("af")
         self.assertIsNotNone(error)
-        self.assertEqual("'No language found with the given id af'", str(error))
+        self.assertEqual("'No language found with the given id af'", str(error))
+
+
+class TestErrorsViews(unittest.TestCase):
+    def test_integrity(self):
+        error = IntegrityError(orig=MagicMock(), statement='', params={})
+        res = data_integrity(error, MagicMock())
+        self.assertEqual({'message': 'this operation violates the data integrity and could not be executed '}, res)