From f6d02d7b3ca3966d5e91c0fa450f83ed5e5a42a3 Mon Sep 17 00:00:00 2001
From: xudong963 <wxd963996380@gmail.com>
Date: Thu, 22 Dec 2022 14:45:18 +0800
Subject: [PATCH] feat: support complete for sqllogictest

---
 Cargo.lock                      | 26 ++++++++++++++++++++++----
 tests/sqllogictests/Cargo.toml  |  2 +-
 tests/sqllogictests/README.md   |  5 +++++
 tests/sqllogictests/src/arg.rs  | 11 ++++++++++-
 tests/sqllogictests/src/main.rs | 14 ++++++++++++--
 5 files changed, 50 insertions(+), 8 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index d96888c78b2b2..37c795ad99fe7 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -3383,6 +3383,17 @@ dependencies = [
  "zeroize",
 ]
 
+[[package]]
+name = "derivative"
+version = "2.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fcc3dd5e9e9c0b295d6e1e4d811fb6f157d5ffd784b8d202fc62eac8035a770b"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
 [[package]]
 name = "derive_more"
 version = "0.99.17"
@@ -3951,6 +3962,12 @@ dependencies = [
  "syn",
 ]
 
+[[package]]
+name = "fs-err"
+version = "2.9.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0845fa252299212f0389d64ba26f34fa32cfe41588355f21ed507c59a0f64541"
+
 [[package]]
 name = "fs2"
 version = "0.4.3"
@@ -8173,12 +8190,13 @@ dependencies = [
 
 [[package]]
 name = "sqllogictest"
-version = "0.9.0"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a89f3be6c90ef2b67d55f999663b367ede63747a2311e480e7710f70c52fef65"
+version = "0.10.0"
+source = "git+https://github.com/xudong963/sqllogictest-rs?branch=runner#0d26c2ebc98eecd4645c40eb4a2bcf54fe0b517e"
 dependencies = [
  "async-trait",
+ "derivative",
  "difference",
+ "fs-err",
  "futures",
  "glob",
  "humantime",
@@ -9006,7 +9024,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "97fee6b57c6a41524a810daee9286c02d7752c4253064d0b05472833a438f675"
 dependencies = [
  "cfg-if",
- "rand 0.7.3",
+ "rand 0.8.5",
  "static_assertions",
 ]
 
diff --git a/tests/sqllogictests/Cargo.toml b/tests/sqllogictests/Cargo.toml
index 7435acf297045..4ea81bec397ee 100644
--- a/tests/sqllogictests/Cargo.toml
+++ b/tests/sqllogictests/Cargo.toml
@@ -17,7 +17,7 @@ regex = "1"
 reqwest = { version = "0.11.13", features = ["json"] }
 serde = "1.0.150"
 serde_json = "1.0.89"
-sqllogictest = "0.9.0"
+sqllogictest = { git = "https://github.com/xudong963/sqllogictest-rs", branch = "runner" }
 thiserror = "1.0.37"
 tokio = { version = "1.21.1", features = ["full"] }
 walkdir = "2.3.2"
diff --git a/tests/sqllogictests/README.md b/tests/sqllogictests/README.md
index 032abf3e73401..68f33086ac0f6 100644
--- a/tests/sqllogictests/README.md
+++ b/tests/sqllogictests/README.md
@@ -26,6 +26,11 @@ Run tests under specific file. This is the most commonly used command because us
 cargo run -p sqllogictests -- --run_file <file_name>
 ```
 ---
+Auto complete test file which is very convenient. What you need to do is just a final check to see if the generated results meet expectations.
+```
+cargo run -p sqllogictests -- --run_file <file_name> --complete
+```
+---
 For more information, run help command:
 ```shell
 cargo run -p sqllogictests -- --help
diff --git a/tests/sqllogictests/src/arg.rs b/tests/sqllogictests/src/arg.rs
index 34c51dbbc758d..11dae0ac07ad3 100644
--- a/tests/sqllogictests/src/arg.rs
+++ b/tests/sqllogictests/src/arg.rs
@@ -43,7 +43,7 @@ pub struct SqlLogicTestArgs {
 
     // Set handler to run tests
     #[arg(
-        short = 'c',
+        short = 'l',
         long = "handlers",
         use_value_delimiter = true,
         value_delimiter = ',',
@@ -59,4 +59,13 @@ pub struct SqlLogicTestArgs {
         default_value = "tests/sqllogictests/suites"
     )]
     pub suites: String,
+
+    // If enable complete mode
+    #[arg(
+        short = 'c',
+        long = "complete",
+        default_missing_value = "true",
+        help = "The arg is used to enable auto complete mode"
+    )]
+    pub complete: bool,
 }
diff --git a/tests/sqllogictests/src/main.rs b/tests/sqllogictests/src/main.rs
index dc2b8c02d7d08..715c9ba0508f7 100644
--- a/tests/sqllogictests/src/main.rs
+++ b/tests/sqllogictests/src/main.rs
@@ -17,6 +17,8 @@ use std::path::PathBuf;
 
 use clap::Parser;
 use client::ClickhouseHttpClient;
+use sqllogictest::default_validator;
+use sqllogictest::update_test_file;
 use sqllogictest::DBOutput;
 use walkdir::DirEntry;
 use walkdir::WalkDir;
@@ -174,8 +176,16 @@ async fn run_suits(suits: ReadDir, databend: Databend) -> Result<()> {
                     continue;
                 }
             }
-            println!("test file: [{}] is running", file_name,);
-            runner.run_file_async(file.unwrap().path()).await?;
+            if args.complete {
+                let col_separator = " ";
+                let validator = default_validator;
+                update_test_file(file.unwrap().path(), &mut runner, col_separator, validator)
+                    .await
+                    .unwrap();
+            } else {
+                println!("test file: [{}] is running", file_name,);
+                runner.run_file_async(file.unwrap().path()).await?;
+            }
         }
     }