Merge pull request #6 from ixcat/pr-703

tests/{schema,test_jobs}.py: add tests/test_jobs.py:test_suppress_dj_errors

Author: dimitri-yatsenko

tests/schema.py

@@ -277,6 +277,38 @@ def make(self, key):
         raise SystemExit('SIGTERM received')
 
 
+@schema
+class DjExceptionNames(dj.Lookup):
+    definition = """
+    dj_exception_name:    char(64)
+    """
+    @property
+    def contents(self):
+        ret = []
+        for e in dir(dj.errors):
+            ea = getattr(dj.errors, e) 
+            if callable(ea):
+                try:
+                    werks = (isinstance(ea, type(Exception))
+                             and isinstance(ea(), Exception))
+                except TypeError as te:
+                    pass
+                if werks:
+                    ret.append((e,))
+        return ret
+
+
+@schema
+class ErrorClassTable(dj.Computed):
+    definition = """
+    -> DjExceptionNames
+    """
+    def make(self, key):
+        ename = key['dj_exception_name']
+        raise getattr(dj.errors, ename)(ename)
+
+
+
 @schema
 class DecimalPrimaryKey(dj.Lookup):
     definition = """

tests/test_jobs.py

@@ -88,6 +88,18 @@ def test_sigterm():
     assert_equals(error_message, 'SystemExit: SIGTERM received')
     schema.schema.jobs.delete()
 
+
+def test_suppress_dj_errors():
+    ''' test_suppress_dj_errors: dj errors suppressable w/o native py blobs'''
+    schema.schema.jobs.delete()
+    with dj.config(enable_python_native_blobs=False):
+        schema.ErrorClassTable().populate(
+            reserve_jobs=True, suppress_errors=True)
+
+        assert len(schema.DjExceptionNames()) == len(schema.DjExceptionNames
+                                                     & schema.schema.jobs)
+
+
 def test_long_error_message():
     # clear out jobs table
     schema.schema.jobs.delete()