Fix error when using @modify_exceptions in an exception initialized…

… without arguments
ploomber · Jul 13, 2023 · f0d346a · f0d346a
1 parent fd1fa92
commit f0d346a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,7 @@
 # CHANGELOG
 
 ## 0.2.14dev
+* [Fix] Fix error when using `@modify_exceptions` in an exception initialized without arguments
 
 ## 0.2.13 (2023-06-27)
 

diff --git a/src/ploomber_core/exceptions.py b/src/ploomber_core/exceptions.py
@@ -82,7 +82,9 @@ class ValidationError(BaseException):
 
 
 def _add_community_link(e):
-    if COMMUNITY not in e.args[0]:
+    if not len(e.args):
+        e.args = (COMMUNITY,)
+    elif COMMUNITY not in e.args[0]:
         message = e.args[0] + COMMUNITY
         e.args = (message,)
 

diff --git a/tests/test_exceptions.py b/tests/test_exceptions.py
@@ -65,6 +65,17 @@ def crash():
     assert exceptions.get_community_link() in str(excinfo.value)
 
 
+def test_modify_exceptions_no_message():
+    @exceptions.modify_exceptions
+    def crash():
+        raise ValueError
+
+    with pytest.raises(ValueError) as excinfo:
+        crash()
+
+    assert exceptions.get_community_link() in str(excinfo.value)
+
+
 def test_modify_exceptions_value_error_method():
     class Something:
         @exceptions.modify_exceptions