Fix some tests with async

Author: BohuTANG

src/query/service/src/interpreters/access/management_mode_access.rs

@@ -33,6 +33,7 @@ impl ManagementModeAccess {
     }
 }
 
+#[async_trait::async_trait]
 impl AccessChecker for ManagementModeAccess {
     // Check what we can do if in management mode.
     fn check(&self, plan: &PlanNode) -> Result<()> {
@@ -50,7 +51,7 @@ impl AccessChecker for ManagementModeAccess {
     }
 
     // Check what we can do if in management mode.
-    fn check_new(&self, plan: &Plan) -> Result<()> {
+    async fn check_new(&self, plan: &Plan) -> Result<()> {
         // Allows for management-mode.
         if self.ctx.get_config().query.management_mode {
             let ok = match plan {

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

@@ -42,7 +42,7 @@ async fn test_setting_interpreter_error() -> Result<()> {
 
     let query = "SET max_block_size=1";
     let (plan, _, _) = planner.plan_sql(query).await?;
-    let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+    let executor = InterpreterFactoryV2::get(ctx.clone(), &plan).await?;
     if let Err(e) = executor.execute(ctx.clone()).await {
         let expect = "Code: 1020, displayText = Unknown variable: \"xx\".";
         assert_eq!(expect, e.to_string());

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

@@ -156,11 +156,11 @@ async fn interpreter_show_create_table_with_comments_test() -> Result<()> {
         for stmt in case.create_stmt {
             // create table in the new planner to support column comment.
             let (plan, _, _) = planner.plan_sql(stmt).await?;
-            let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+            let executor = InterpreterFactoryV2::get(ctx.clone(), &plan).await?;
             let _ = executor.execute(ctx.clone()).await?;
         }
         let (plan, _, _) = planner.plan_sql(case.show_stmt).await?;
-        let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+        let executor = InterpreterFactoryV2::get(ctx.clone(), &plan).await?;
         assert_eq!(executor.name(), "ShowCreateTableInterpreter");
         let result = executor
             .execute(ctx.clone())

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

@@ -26,7 +26,7 @@ async fn test_use_interpreter() -> Result<()> {
 
     let query = "USE default";
     let (plan, _, _) = planner.plan_sql(query).await?;
-    let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
+    let executor = InterpreterFactoryV2::get(ctx.clone(), &plan).await?;
     assert_eq!(executor.name(), "UseDatabaseInterpreter");
 
     let mut stream = executor.execute(ctx.clone()).await?;