fixes grappa-py#68

sgissinger · Jan 21, 2021 · bf6b6e0 · bf6b6e0
1 parent bef2355
commit bf6b6e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/grappa/operators/raises.py b/grappa/operators/raises.py
@@ -58,7 +58,7 @@ def fn():
     def after_success(self, obj, *keys):
         message = getattr(self.value, 'message', None)
 
-        if not message:
+        if not message and self.value:
             message = ' '.join([str(item) for item in self.value.args])
 
         self.ctx.subject = message

diff --git a/tests/operators/raises_test.py b/tests/operators/raises_test.py
@@ -32,6 +32,7 @@ def error_with_params(foo_param):
     with pytest.raises(AssertionError):
         no_error | should.raise_error(AssertionError)
 
+    no_error | should.do_not.raise_error(AssertionError)
 
 def test_raises_with_message_redirection(should):
     def error():
@@ -41,6 +42,7 @@ def env_error():
         raise EnvironmentError(3501, 'bar')
 
     error | should.raise_error(AssertionError) > should.equal('foo')
+    error | should.raise_error(AssertionError) > should.do_not.equal('fooed')
 
     error | should.raise_error(AssertionError) > should.contain('fo')