From 536242c621659e180566178e13b455359d41c9cf Mon Sep 17 00:00:00 2001
From: everpcpc <git@everpcpc.com>
Date: Wed, 25 Oct 2023 08:52:42 +0800
Subject: [PATCH] fix: tokenize stage path

---
 cli/src/ast/tokenizer.rs     |   2 +-
 cli/tests/01-put.sh          |   2 +-
 cli/tests/data/books.parquet | Bin 0 -> 1882 bytes
 3 files changed, 2 insertions(+), 2 deletions(-)
 create mode 100644 cli/tests/data/books.parquet

diff --git a/cli/src/ast/tokenizer.rs b/cli/src/ast/tokenizer.rs
index b2bb73efa..62b314a83 100644
--- a/cli/src/ast/tokenizer.rs
+++ b/cli/src/ast/tokenizer.rs
@@ -122,7 +122,7 @@ pub enum TokenKind {
     #[regex(r#"'([^'\\]|\\.|'')*'"#)]
     QuotedString,
 
-    #[regex(r#"@([^\s`;'"])+"#)]
+    #[regex(r#"@([^\s`;'"])+/"#)]
     AtString,
 
     #[regex(r"[xX]'[a-fA-F0-9]*'")]
diff --git a/cli/tests/01-put.sh b/cli/tests/01-put.sh
index f0c5afd44..533d9481b 100755
--- a/cli/tests/01-put.sh
+++ b/cli/tests/01-put.sh
@@ -6,5 +6,5 @@ echo "CREATE STAGE ss_temp" | ${BENDSQL}
 echo "ABCD" >/tmp/a1.txt
 echo "ABCD" >/tmp/a2.txt
 
-echo 'put fs:///tmp/a*.txt @ss_temp/abc' | ${BENDSQL}
+echo 'put fs://cli/tests/data/*.parquet @ss_temp/abc/' | ${BENDSQL}
 echo 'get @ss_temp/abc fs:///tmp/edf' | ${BENDSQL}
diff --git a/cli/tests/data/books.parquet b/cli/tests/data/books.parquet
new file mode 100644
index 0000000000000000000000000000000000000000..ac610fa347d5560e65f82c8c281acf5cefbac723
GIT binary patch
literal 1882
zcmWG=3^EjD5iJp|@BvaRLJSO6PsA7)7($8?^NJIbOEUBG6atF!lT(X}GxO4AfHFa;
zi77y4u|j5^f=gmaVp3vpszPvOaY<@!u_%Kmn<$qK$Rr?xO~#BtLkzn~5;FMA0b0Y%
z!N|;{wuebZlt<E(g@H{{g0&>GBqvprNsL8o4U^awCbc<?Y8>QSAqjCN69drEETT+6
zD}j`os2408lsh<pVqTfK3hqUTmBK)_Z)S2vVrq^;a7lh%YEn^Rc4`q84+~>aCcyzS
z2kx^EtO%d6C6<<C<QIXxc7ajs2BX?P7PT2HAiohX4XY;<L{(s(5U^nZ+GuELX#}E;
z3=B-LxR?bbF2Mp)1$W*$4utbqQW8rb;oib1*2Adwh+S<FJIHa!8nL=$uqLZl+z8)c
zB&$<gYTG!6Y`R*&kMJ61x?0Dlc7k^x)73peg!9nT)g(c+O9DisD<(-FQw9mP;^d6f
z+(c0}22mbW259L3VMF~5;ec&n5aW>4k+h+2d7?(EJg!EC+$q+;D#oJ5AvS|mgn<DV
zGW5+7YP3qGYE(#Ikf7Pd2?`q~1`VZx#G=B|)Dqp~f&zuI)S_ZgooJ|Ms%NMh?B^H|
W?BO3GX#=boA&6l<uqp`v)~Wy=<d?|+

literal 0
HcmV?d00001