SNOW-1019480-sql-filter-nonsql-statements: strip comments from stream… (

#700)

* SNOW-1019480-sql-filter-nonsql-statements: strip comments from streamed queries

src/snowflake/cli/plugins/sql/manager.py

@@ -28,5 +28,10 @@ def execute(
         elif file:
             query = file.read_text()
 
-        single_statement = len(list(split_statements(StringIO(query)))) == 1
-        return single_statement, self._execute_string(query)
+        statements = tuple(
+            statement
+            for statement, _ in split_statements(StringIO(query), remove_comments=True)
+        )
+        single_statement = len(statements) == 1
+
+        return single_statement, self._execute_string("\n".join(statements))

tests_integration/test_sql.py

@@ -100,3 +100,15 @@ def _log(
     assert query_0 == "13"
     assert time_1 - time_0 >= 10.0
     assert "waited 10 seconds" in query_1
+
+
+@pytest.mark.integration
+def test_trailing_comments_queries(runner, snowflake_session, test_root_path):
+    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": 1},
+        ],
+    ]