Integration tests

ploomber · Jul 4, 2023 · fec122d · fec122d
1 parent 72454ee
commit fec122d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/sql/error_message.py b/src/sql/error_message.py
@@ -57,7 +57,10 @@ def detail(original_error, query=None):
                 parse_msg = parse_sqlglot_error(e, q)
                 return_msg = return_msg + parse_msg if parse_msg else return_msg
 
-        return return_msg + "\n" + ORIGINAL_ERROR + original_error + "\n"
+        cte_msg = "If using CTEs, you may pass the --with argument explicitly."
+        return (
+            return_msg + "\n" + ORIGINAL_ERROR + original_error + "\n" + cte_msg + "\n"
+        )
 
     if "fe_sendauth: no password supplied" in original_error:
         return (

diff --git a/src/tests/integration/test_generic_db_operations.py b/src/tests/integration/test_generic_db_operations.py
@@ -667,4 +667,6 @@ def test_sql_query_snippet_error(ip_with_dynamic_db, request):
     %%sql
 SELECT * FROM second_cte"""
     )
-    assert out.error_in_exec is None
+    assert "If using CTEs, you may pass the --with argument explicitly." in str(
+        out.error_in_exec
+    )