Skip to content

Commit 9aa35cc

Browse files
committed
refactor(interpreter): try fix test failure
1 parent c770d0c commit 9aa35cc

File tree

3 files changed

+9
-10
lines changed

3 files changed

+9
-10
lines changed

src/query/service/src/interpreters/interpreter_table_create_v2.rs

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -153,11 +153,10 @@ impl CreateTableInterpreterV2 {
153153
overwrite: false,
154154
source: InsertInputSource::SelectPlan(select_plan),
155155
};
156-
let insert_interpreter_v2 =
157-
InsertInterpreterV2::try_create(self.ctx.clone(), insert_plan, false)?;
158-
insert_interpreter_v2.execute(self.ctx.clone()).await?;
159156

160-
Ok(PipelineBuildResult::create())
157+
InsertInterpreterV2::try_create(self.ctx.clone(), insert_plan, false)?
158+
.execute2()
159+
.await
161160
}
162161

163162
async fn create_table(&self) -> Result<PipelineBuildResult> {

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

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ async fn test_call_fuse_block_interpreter() -> Result<()> {
121121
let (plan, _, _) = planner.plan_sql(query).await?;
122122
let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
123123
assert_eq!(executor.name(), "CallInterpreter");
124-
let res = executor.execute().await;
124+
let res = executor.execute(ctx.clone()).await;
125125
assert_eq!(res.is_err(), true);
126126
let expect = "Code: 1028, displayText = Function `FUSE_BLOCK` expect to have [2, 3] arguments, but got 0.";
127127
assert_eq!(expect, res.err().unwrap().to_string());
@@ -133,7 +133,7 @@ async fn test_call_fuse_block_interpreter() -> Result<()> {
133133
let (plan, _, _) = planner.plan_sql(query).await?;
134134
let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
135135
assert_eq!(executor.name(), "CallInterpreter");
136-
let res = executor.execute().await;
136+
let res = executor.execute(ctx.clone()).await;
137137
assert_eq!(res.is_err(), true);
138138
assert_eq!(
139139
res.err().unwrap().code(),
@@ -147,7 +147,7 @@ async fn test_call_fuse_block_interpreter() -> Result<()> {
147147
let (plan, _, _) = planner.plan_sql(query).await?;
148148
let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
149149
assert_eq!(executor.name(), "CallInterpreter");
150-
let res = executor.execute().await;
150+
let res = executor.execute(ctx.clone()).await;
151151
assert_eq!(res.is_err(), true);
152152
let expect =
153153
"Code: 1015, displayText = expects table of engine FUSE, but got SystemTables.";
@@ -162,15 +162,15 @@ async fn test_call_fuse_block_interpreter() -> Result<()> {
162162

163163
let (plan, _, _) = planner.plan_sql(query).await?;
164164
let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
165-
let _ = executor.execute().await?;
165+
let _ = executor.execute(ctx.clone()).await?;
166166
}
167167

168168
// fuse_block
169169
{
170170
let query = "call system$fuse_block(default, a)";
171171
let (plan, _, _) = planner.plan_sql(query).await?;
172172
let executor = InterpreterFactoryV2::get(ctx.clone(), &plan)?;
173-
let _ = executor.execute().await?;
173+
let _ = executor.execute(ctx.clone()).await?;
174174
}
175175

176176
Ok(())

src/query/service/tests/it/storages/fuse/table_functions/fuse_block_table.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ async fn test_fuse_block_table() -> Result<()> {
3232
// test db & table
3333
let create_table_plan = fixture.default_crate_table_plan();
3434
let interpreter = CreateTableInterpreterV2::try_create(ctx.clone(), create_table_plan)?;
35-
interpreter.execute().await?;
35+
interpreter.execute(ctx.clone()).await?;
3636

3737
{
3838
let expected = vec![

0 commit comments

Comments
 (0)