test fixes

ploomber · Mar 6, 2022 · 61a1a55 · 61a1a55
1 parent 765d159
commit 61a1a55
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/src/ploomber/exceptions.py b/src/ploomber/exceptions.py
@@ -52,7 +52,7 @@ class DAGRenderError(Exception):
     @cli_endpoint decorator)
     """
     def __init__(self, message):
-        message = message + 'Need help? https://ploomber.io/community'
+        message = message + '\nNeed help? https://ploomber.io/community'
         super().__init__(message)
 
 
@@ -67,7 +67,7 @@ class DAGBuildError(Exception):
     @cli_endpoint decorator)
     """
     def __init__(self, message):
-        message = message + 'Need help? https://ploomber.io/community'
+        message = message + '\nNeed help? https://ploomber.io/community'
         super().__init__(message)
 
 

diff --git a/tests/cli/test_parsers.py b/tests/cli/test_parsers.py
@@ -235,7 +235,7 @@ def test_custom_parser_error_if_unable_to_automatically_locate_entry_point(
     captured = capsys.readouterr()
 
     assert excinfo.value.code == 2
-    assert 'Unable to find a pipeline entry point' in captured.err
+    assert 'Unable to find a pipeline' in captured.err
 
 
 def test_error_if_missing_entry_point_value(monkeypatch, capsys):

diff --git a/tests/dag/test_dag.py b/tests/dag/test_dag.py
@@ -910,7 +910,8 @@ def test_on_render_crashes(tmp_directory):
     with pytest.raises(DAGRenderError) as excinfo:
         dag.build()
 
-    msg = 'Exception when running on_render for DAG "No name": crash!'
+    msg = ('Exception when running on_render for DAG "No name": crash!'
+           '\nNeed help? https://ploomber.io/community')
     assert str(excinfo.value) == msg
     assert 'crash!' in str(excinfo.getrepr())
     assert dag._exec_status == DAGStatus.ErroredRender
@@ -936,7 +937,8 @@ def test_on_finish_crashes(tmp_directory):
     with pytest.raises(DAGBuildError) as excinfo:
         dag.build()
 
-    msg = 'Exception when running on_finish for DAG "No name": crash!'
+    msg = ('Exception when running on_finish for DAG "No name": crash!'
+           '\nNeed help? https://ploomber.io/community')
     assert str(excinfo.value) == msg
     assert 'crash!' in str(excinfo.getrepr())
     assert dag._exec_status == DAGStatus.Errored
@@ -976,7 +978,8 @@ def test_on_failure_crashes(caplog):
             dag.build()
 
     assert hook_crashing.count == 1
-    msg = 'Exception when running on_failure for DAG "dag": crash!'
+    msg = ('Exception when running on_failure for DAG "dag": crash!'
+           '\nNeed help? https://ploomber.io/community')
     assert str(excinfo.value) == msg
     assert 'crash!' in str(excinfo.getrepr())
     assert dag._exec_status == DAGStatus.Errored