Merge pull request #481 from hectord/first_idea

Remove non-ASCII characters from a critical exception's message
gabrielfalcao · Aug 27, 2015 · 764456d · 764456d
2 parents 40c940d + de914ad
commit 764456d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lettuce/__init__.py b/lettuce/__init__.py
@@ -189,6 +189,9 @@ def run(self):
         except exceptions.LettuceSyntaxError, e:
             sys.stderr.write(e.msg)
             failed = True
+        except exceptions.NoDefinitionFound, e:
+            sys.stderr.write(e.msg)
+            failed = True
         except:
             if not self.failfast:
                 e = sys.exc_info()[1]

diff --git a/lettuce/exceptions.py b/lettuce/exceptions.py
@@ -28,8 +28,11 @@ class NoDefinitionFound(Exception):
     """
     def __init__(self, step):
         self.step = step
-        super(NoDefinitionFound, self).__init__(
-            'The step r"%s" is not defined' % self.step.sentence)
+
+        error = filter(lambda x : 0 <= ord(x) <= 127,
+                       'The step r"%s" is not defined' % self.step.sentence)
+        super(NoDefinitionFound, self).__init__(error)
+
 
 
 class ReasonToFail(object):