common: fix UnicodeDecodeError in activitycontextimpl()

Closes #338

circle/common/models.py

@@ -49,7 +49,15 @@ class WorkerNotFound(Exception):
     pass
     pass
 
 
 
 
+def get_error_msg(exception):
+    try:
+        return unicode(exception)
+    except UnicodeDecodeError:
+        return unicode(str(exception), encoding='utf-8', errors='replace')
+
+
 def activitycontextimpl(act, on_abort=None, on_commit=None):
 def activitycontextimpl(act, on_abort=None, on_commit=None):
+    result = None
     try:
     try:
         try:
         try:
             yield act
             yield act
@@ -62,7 +70,7 @@ def activitycontextimpl(act, on_abort=None, on_commit=None):
             result = create_readable(
             result = create_readable(
                 ugettext_noop("Failure."),
                 ugettext_noop("Failure."),
                 ugettext_noop("Unhandled exception: %(error)s"),
                 ugettext_noop("Unhandled exception: %(error)s"),
-                error=unicode(e))
+                error=get_error_msg(e))
             raise
             raise
     except:
     except:
         logger.exception("Failed activity %s" % unicode(act))
         logger.exception("Failed activity %s" % unicode(act))

circle/common/tests/test_models.py

@@ -22,6 +22,7 @@ from mock import MagicMock
 
 
 from .models import TestClass
 from .models import TestClass
 from ..models import HumanSortField
 from ..models import HumanSortField
+from ..models import activitycontextimpl
 
 
 
 
 class MethodCacheTestCase(TestCase):
 class MethodCacheTestCase(TestCase):
@@ -80,3 +81,22 @@ class TestHumanSortField(TestCase):
 
 
             test_result = HumanSortField.get_normalized_value(obj, val)
             test_result = HumanSortField.get_normalized_value(obj, val)
             self.assertEquals(test_result, result)
             self.assertEquals(test_result, result)
+
+
+class ActivityContextTestCase(TestCase):
+    class MyException(Exception):
+        pass
+
+    def test_unicode(self):
+        act = MagicMock()
+        gen = activitycontextimpl(act)
+        gen.next()
+        with self.assertRaises(self.MyException):
+            gen.throw(self.MyException(u'test\xe1'))
+
+    def test_str(self):
+        act = MagicMock()
+        gen = activitycontextimpl(act)
+        gen.next()
+        with self.assertRaises(self.MyException):
+            gen.throw(self.MyException('test\xbe'))