diff --git a/tests_integration/test_sql.py b/tests_integration/test_sql.py index 5a35dc7d7a..09dfca23a5 100644 --- a/tests_integration/test_sql.py +++ b/tests_integration/test_sql.py @@ -100,3 +100,14 @@ def _log( assert query_0 == "13" assert time_1 - time_0 >= 10.0 assert "waited 10 seconds" in query_1 + + +@pytest.mark.integration +@pytest.mark.usefixtures("snowflake_session, test_root_path") +def test_trailing_comments_queries(runner): + trailin_comment_query = "select 1;\n\n-- trailing comment\n" + result = runner.invoke_with_connection_json(["sql", "-q", trailin_comment_query]) + assert result.exit_code == 0 + assert result.json == [ + ["1"], + ]