added tests

ploomber · Dec 18, 2023 · e0b769b · e0b769b
1 parent a1b500c
commit e0b769b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/tests/test_parse.py b/src/tests/test_parse.py
@@ -772,7 +772,7 @@ def test_connections_file_get_default_connection_url(tmp_empty, content, expecte
         ),
         (
             "select 'hello'[:-2]",
-            "hell",
+            "hel",
         ),
     ],
 )
@@ -897,6 +897,11 @@ def test_escape_string_slicing_notation(query, expected_escaped, expected_found)
             "-p --save snippet -N ",
             "insert into authors values('[100]'::json->0)",
         ),
+        (
+            "--save snippet SELECT TRIM(' padded ')",
+            "--save snippet ",
+            "SELECT TRIM(' padded ')",
+        ),
     ],
     ids=[
         "no-query",
@@ -910,6 +915,7 @@ def test_escape_string_slicing_notation(query, expected_escaped, expected_found)
         "update",
         "delete",
         "insert",
+        "select-uppercase",
     ],
 )
 def test_split_args_and_sql(line, expected_args, expected_sql):