ci: drop table after test

.github/workflows/ci.yaml

@@ -58,7 +58,7 @@ jobs:
       - uses: ./.github/actions/setup
         with:
           cache-key: integration
-      - run: make -C tests
-      - run: make -C tests test-flight-sql
+      - run: make -C tests test-core
+      - run: make -C tests test-driver
       - run: make -C tests test-bendsql
       - run: sudo chown -R runner ~/.cargo/registry

Makefile

@@ -13,14 +13,17 @@ build:
 test:
 	cargo test --all --all-features --lib -- --nocapture
 
+integration:
+	make -C tests
+
 integration-down:
 	make -C tests down
 
-integration-tests:
-	make -C tests
+integration-core:
+	make -C tests test-core
 
-integration-tests-flight-sql:
-	make -C tests test-flight-sql
+integration-driver:
+	make -C tests test-driver
 
 integration-bendsql:
 	make -C tests test-bendsql

README.md

@@ -1,4 +1,4 @@
-# Databend Client
+# BendSQL
 
 Databend Native Client in Rust
 
@@ -13,17 +13,17 @@ Databend Native Client in Rust
 
 ## Installation for BendSQL
 
-* With Cargo:
+* Cargo:
 ```bash
 cargo install bendsql
 ```
 
-* With Homebrew:
+* Homebrew:
 ```bash
 brew install databendcloud/homebrew-tap/bendsql
 ```
 
-* With Binary: check for latest release [here](https://github.com/datafuselabs/databend-client/releases)
+* Binary: check for latest release [here](https://github.com/datafuselabs/databend-client/releases)
 
 
 ## Development
@@ -45,6 +45,5 @@ make test
 *Note: Docker and Docker Compose needed*
 
 ```bash
-make integration-tests
-make integration-tests-flight-sql
+make integration
 ```

cli/src/config.rs

@@ -83,7 +83,7 @@ impl Default for Settings {
         Settings {
             display_pretty_sql: true,
             progress_color: "cyan".to_string(),
-            prompt: "bendsql> ".to_string(),
+            prompt: "{user}@{host}> ".to_string(),
             output_format: OutputFormat::Table,
             show_progress: false,
         }

cli/src/session.rs

@@ -62,6 +62,7 @@ impl Session {
         {
             prompt = prompt.replace("{host}", &info.host);
             prompt = prompt.replace("{user}", &info.user);
+            prompt = prompt.replace("{port}", &info.port.to_string());
         }
 
         Ok(Self {
@@ -96,7 +97,7 @@ impl Session {
             let prompt = if self.query.is_none() {
                 &self.prompt
             } else {
-                "    -> "
+                ""
             };
             match rl.readline(prompt) {
                 Ok(line) => {

cli/tests/http/01-load_stdin.sh

@@ -11,3 +11,7 @@ SQL
 ${BENDSQL} --query='INSERT INTO test_books VALUES;' --format=csv --data=@- <cli/tests/data/books.csv
 
 ${BENDSQL} --query='SELECT * FROM test_books LIMIT 10;' --output=tsv
+
+cat <<SQL | ${BENDSQL}
+DROP TABLE test_books;
+SQL

cli/tests/http/02-load_file.sh

@@ -11,3 +11,7 @@ SQL
 ${BENDSQL} --query='INSERT INTO test_books VALUES;' --format=csv --data=@cli/tests/data/books.csv
 
 ${BENDSQL} --query='SELECT * FROM test_books LIMIT 10;' --output=tsv
+
+cat <<SQL | ${BENDSQL}
+DROP TABLE test_books;
+SQL

cli/tests/http/03-load_file_gzip.sh

@@ -15,3 +15,7 @@ ${BENDSQL} \
 
 echo "SELECT COUNT(*) FROM test_ontime;" | ${BENDSQL} --output=tsv
 echo 'SELECT * FROM test_ontime LIMIT 1;' | ${BENDSQL} --output=csv
+
+cat <<SQL | ${BENDSQL}
+DROP TABLE test_ontime;
+SQL

core/tests/core/stage.rs

@@ -29,9 +29,13 @@ async fn insert_with_stage(presigned: bool) {
     let file = File::open("tests/core/data/sample.csv").await.unwrap();
     let metadata = file.metadata().await.unwrap();
 
-    let path = chrono::Utc::now().format("%Y%m%d%H%M%S%.9f").to_string();
+    let path = chrono::Utc::now().format("%Y%m%d%H%M%S%9f").to_string();
     let stage_location = format!("@~/{}/sample.csv", path);
-    let table = format!("sample_{}", path);
+    let table = if presigned {
+        format!("sample_insert_presigned_{}", path)
+    } else {
+        format!("sample_insert_stream_{}", path)
+    };
 
     client
         .upload_to_stage(&stage_location, file, metadata.len())
@@ -72,6 +76,9 @@ async fn insert_with_stage(presigned: bool) {
         ["6", "Beijing", "99"],
     ];
     assert_eq!(resp.data, expect);
+
+    let sql = format!("DROP TABLE `{}` ALL;", table);
+    client.query_wait(&sql).await.unwrap();
 }
 
 #[tokio::test]

driver/tests/driver/select_iter.rs

@@ -100,6 +100,10 @@ async fn select_iter() {
         assert_eq!(v, expected[row_count]);
         row_count += 1;
     }
+    assert_eq!(row_count, 3);
+
+    let sql_drop = format!("DROP TABLE `{}`", table);
+    conn.exec(&sql_drop).await.unwrap();
 }
 
 #[tokio::test]

driver/tests/driver/stream_load.rs

@@ -39,8 +39,12 @@ async fn stream_load(presigned: bool, file_type: &str) {
         .unwrap();
     let metadata = file.metadata().await.unwrap();
 
-    let path = Utc::now().format("%Y%m%d%H%M%S%.9f").to_string();
-    let table = format!("books_{}", path);
+    let path = Utc::now().format("%Y%m%d%H%M%S%9f").to_string();
+    let table = if presigned {
+        format!("books_stream_load_{}_presigned_{}", file_type, path)
+    } else {
+        format!("books_stream_load_{}_{}", file_type, path)
+    };
 
     let sql = format!(
         "CREATE TABLE `{}` (
@@ -99,6 +103,9 @@ async fn stream_load(presigned: bool, file_type: &str) {
         ),
     ];
     assert_eq!(result, expected);
+
+    let sql = format!("DROP TABLE `{}` ALL;", table);
+    client.exec(&sql).await.unwrap();
 }
 
 #[tokio::test]

tests/Makefile

@@ -1,6 +1,4 @@
-PHONY: test
-
-default: test
+default: run
 
 prepare:
 	mkdir -p data/databend
@@ -10,18 +8,19 @@ up: prepare
 
 start: up
 
-test: up
-	cargo test --test '*'
-
-run: test
+test-core: up
+	cargo test --test core
 
-test-flight-sql: up
+test-driver: up
+	cargo test --test driver
 	TEST_DATABEND_DSN=databend+flight://root:@localhost:8900/default?sslmode=disable cargo test --features flight-sql --test driver
 
 test-bendsql: up
 	cd .. && ./cli/test.sh http
 	cd .. && ./cli/test.sh flight
 
+run: test-core test-driver test-bendsql
+
 down:
 	docker compose down