Add specifics on the "ImportError" to the message in case of an attri…

…bute error.
keondo · Mar 16, 2012 · 14ecb8e · 14ecb8e
1 parent be8f768
commit 14ecb8e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/rq/job.py b/rq/job.py
@@ -15,6 +15,9 @@ def unpickle(pickled_string):
     """
     try:
         obj = loads(pickled_string)
+    except AttributeError as e:
+        raise UnpickleError('Could not unpickle: %s' % e.message,
+                pickled_string)
     except (StandardError, UnpicklingError):
         raise UnpickleError('Could not unpickle.', pickled_string)
     return obj

diff --git a/tests/test_worker.py b/tests/test_worker.py
@@ -52,7 +52,6 @@ def test_work_is_unreadable(self):
         w = Worker([q])
         w.work(burst=True)   # should silently pass
         self.assertEquals(q.count, 0)
-
         self.assertEquals(failed_q.count, 1)
 
     def test_work_fails(self):