init sqllogictest

Author: GE-fighting

Cargo.toml

@@ -47,3 +47,8 @@ ctor = "0.2.0"
 env_logger = "0.10"
 paste = "^1.0"
 rstest = "0.17"
+
+[workspace]
+members = [
+  "tests/sqllogictest",
+]

tests/slt/create_table.slt

@@ -0,0 +1,38 @@
+onlyif sqlrs_v2
+statement ok
+create table t1(v1 varchar, v2 varchar, v3 varchar);
+insert into t1 values('a', 'b', 'c');
+
+
+onlyif sqlrs_v2
+statement error
+create table t1(v1 int);
+
+
+onlyif sqlrs_v2
+statement ok
+create table t2(v1 boolean, v2 tinyint, v3 smallint, v4 int, v5 bigint, v6 float, v7 double, v8 varchar);
+insert into t2 values(true, 1, 2, 3, 4, 5.1, 6.2, '7');
+
+
+onlyif sqlrs_v2
+statement ok
+create table t3(v1 boolean, v2 tinyint unsigned, v3 smallint unsigned, v4 int unsigned, v5 bigint unsigned, v6 float, v7 double, v8 varchar);
+insert into t3 values(true, 1, 2, 3, 4, 5.1, 6.2, '7');
+
+
+onlyif sqlrs_v2
+statement ok
+create table t4(v1 int);
+select v1 from t4;
+
+
+onlyif sqlrs_v2
+statement ok
+create table read_csv_table as select * from read_csv('tests/csv/t2.csv');
+
+onlyif sqlrs_v2
+query I
+select a from read_csv_table limit 1;
+----
+10

tests/slt/csv/csv.slt

@@ -0,0 +1,20 @@
+onlyif sqlrs_v2
+statement ok
+create table state(id varchar, state_code varchar, state_name varchar);
+
+onlyif sqlrs_v2
+statement ok
+copy state from 'tests/slt/csv/state1.csv' ( DELIMITER '|' );
+
+onlyif sqlrs_v2
+statement ok
+copy state from 'tests/slt/csv/state2.csv' ( DELIMITER '|', HEADER false);
+
+onlyif sqlrs_v2
+query I
+SELECT id FROM state
+----
+1
+2
+3
+4

tests/slt/csv/state1.csv

@@ -0,0 +1,5 @@
+id|state_code|state_name
+1|CA|California State
+2|CO|Colorado State
+3|NJ|New Jersey
+

tests/slt/csv/state2.csv

@@ -0,0 +1,2 @@
+4|CA|California State
+

tests/sqllogictest/Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "sqllogictest-test"
+version = "0.4.0"
+edition = "2021"
+
+[dependencies]
+"kip-sql" = { path = "../.." }
+sqllogictest = "0.6"
+glob = "0.3"
+async-trait = "0.1"
+libtest-mimic = "0.6"
+
+[[test]]
+name = "sqllogictest"
+harness = false

tests/sqllogictest/src/lib.rs

@@ -0,0 +1,39 @@
+
+use std::sync::Arc;
+
+use sqllogictest::{AsyncDB, Runner};
+use kip_sql::db::{Database,DatabaseError};
+
+fn init_tables(){
+    
+    let db = Arc::new( Database::new_on_mem());
+    let mut tester = Runner::new(DatabaseWrapper {db});
+    tester.run_file(sqlfile).unwrap()
+    // tokio_test::block_on(async move {
+    //     let _ = database.run("create table t1 (a int, b boolean)").await?;
+    //     let _ = database.run("insert into t1 values (1, true), (2, false)").await?;
+    //     let vec_batch = database.run("select * from t1").await?;
+
+    //     let table = database.storage
+    //         .get_catalog()
+    //         .get_table(0).unwrap().clone();
+    //     println!("{:#?}", concat_batches(&table.schema(), &vec_batch));
+
+    //     Ok(())
+    // })
+
+}
+
+struct DatabaseWrapper {
+    db: Arc<Database>,
+}
+
+#[async_trait::async_trait]
+impl AsyncDB for DatabaseWrapper {
+    type Error = DatabaseError;
+    async fn run(&mut self, sql: &str) -> Result<String, Self::Error> {
+        let chunks = self.db.run(sql).await?;
+        let output = chunks.iter().map(record_batch_to_string).try_collect();
+        Ok(output?)
+    }
+}