fix(kernel): reject streaming PUT without whitespace

Signed-off-by: Aakash Saravanan <aakash.saravanan@databricks.com>

Author: aakash-saravanan-db

src/databricks/sql/backend/kernel/client.py

@@ -567,7 +567,7 @@ def execute_command(
                 stripped_sql = _strip_leading_sql_comments(operation)
                 staging_verb = _staging_verb(stripped_sql)
                 if re.match(
-                    r"""PUT\s+(['"])__input_stream__\1(?:\s|$)""",
+                    r"""PUT\s*(['"])__input_stream__\1(?:\s|$)""",
                     stripped_sql,
                     re.IGNORECASE,
                 ):

tests/unit/test_kernel_client.py

@@ -905,6 +905,8 @@ def execute():
     [
         "PUT '__input_stream__' INTO '/Volumes/c/s/v/file'",
         "/* upload */ PUT '__input_stream__' INTO '/Volumes/c/s/v/file'",
+        "PUT'__input_stream__' INTO '/Volumes/c/s/v/file'",
+        '/* upload */ PUT"__input_stream__" INTO \'/Volumes/c/s/v/file\'',
     ],
 )
 def test_kernel_rejects_streaming_put(query):