add test case

Author: zhyass

src/query/service/src/procedures/systems/fuse_block.rs

@@ -1,4 +1,4 @@
-//  Copyright 2021 Datafuse Labs.
+//  Copyright 2022 Datafuse Labs.
 //
 //  Licensed under the Apache License, Version 2.0 (the "License");
 //  you may not use this file except in compliance with the License.

src/query/service/tests/it/interpreters/interpreter_call.rs

@@ -110,6 +110,73 @@ async fn test_call_fuse_snapshot_interpreter() -> Result<()> {
     Ok(())
 }
 
+
+#[tokio::test(flavor = "multi_thread", worker_threads = 1)]
+async fn test_call_fuse_block_interpreter() -> Result<()> {
+    let (_guard, ctx) = crate::tests::create_query_context().await?;
+    let mut planner = Planner::new(ctx.clone());
+
+    // NumberArgumentsNotMatch
+    {
+        let query = "call system$fuse_block()";
+        let (plan, _, _) = planner.plan_sql(query).await?;
+        let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+        assert_eq!(executor.name(), "CallInterpreter");
+        let res = executor.execute().await;
+        assert_eq!(res.is_err(), true);
+        let expect = "Code: 1028, displayText = Function `FUSE_BLOCK` expect to have 3 arguments, but got 0.";
+        assert_eq!(expect, res.err().unwrap().to_string());
+    }
+
+    // UnknownTable
+    {
+        let query = "call system$fuse_block(default, test, xxxx)";
+        let (plan, _, _) = planner.plan_sql(query).await?;
+        let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+        assert_eq!(executor.name(), "CallInterpreter");
+        let res = executor.execute().await;
+        assert_eq!(res.is_err(), true);
+        assert_eq!(
+            res.err().unwrap().code(),
+            ErrorCode::UnknownTable("").code()
+        );
+    }
+
+    // BadArguments
+    {
+        let query = "call system$fuse_block(system, tables, xxxx)";
+        let (plan, _, _) = planner.plan_sql(query).await?;
+        let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+        assert_eq!(executor.name(), "CallInterpreter");
+        let res = executor.execute().await;
+        assert_eq!(res.is_err(), true);
+        let expect =
+            "Code: 1015, displayText = expects table of engine FUSE, but got SystemTables.";
+        assert_eq!(expect, res.err().unwrap().to_string());
+    }
+
+    // Create table
+    {
+        let query = "\
+            CREATE TABLE default.a(a bigint)\
+        ";
+
+        let (plan, _, _) = planner.plan_sql(query).await?;
+        let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+        let _ = executor.execute().await?;
+    }
+
+    // FuseHistory
+    {
+        let query = "call system$fuse_block(default, a, xxxx)";
+        let (plan, _, _) = planner.plan_sql(query).await?;
+        let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+        let _ = executor.execute().await?;
+    }
+
+    Ok(())
+}
+
 #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
 async fn test_call_clustering_information_interpreter() -> Result<()> {
     let (_guard, ctx) = crate::tests::create_query_context().await?;

src/query/storages/fuse/src/table_functions/fuse_blocks/fuse_block_table.rs

@@ -21,11 +21,11 @@ use common_exception::Result;
 use common_meta_app::schema::TableIdent;
 use common_meta_app::schema::TableInfo;
 use common_meta_app::schema::TableMeta;
-use common_planners::Expression;
-use common_planners::Extras;
-use common_planners::Partitions;
-use common_planners::ReadDataSourcePlan;
-use common_planners::Statistics;
+use common_legacy_planners::Expression;
+use common_legacy_planners::Extras;
+use common_legacy_planners::Partitions;
+use common_legacy_planners::ReadDataSourcePlan;
+use common_legacy_planners::Statistics;
 
 use crate::pipelines::processors::port::OutputPort;
 use crate::pipelines::processors::processor::ProcessorPtr;