diff --git a/src/tests/simulation/tests/integration_tests/recovery/background_ddl.rs b/src/tests/simulation/tests/integration_tests/recovery/background_ddl.rs index 010b06389325c..1a3742c1c1769 100644 --- a/src/tests/simulation/tests/integration_tests/recovery/background_ddl.rs +++ b/src/tests/simulation/tests/integration_tests/recovery/background_ddl.rs @@ -32,20 +32,9 @@ const DROP_MV1: &str = "DROP MATERIALIZED VIEW mv1;"; const WAIT: &str = "WAIT;"; async fn kill_cn_and_wait_recover(cluster: &Cluster) { - // Kill it again - for _ in 0..5 { - cluster - .kill_node(&KillOpts { - kill_rate: 1.0, - kill_meta: false, - kill_frontend: false, - kill_compute: true, - kill_compactor: false, - restart_delay_secs: 1, - }) - .await; - sleep(Duration::from_secs(2)).await; - } + cluster + .kill_nodes(["compute-1", "compute-2", "compute-3"], 0) + .await; sleep(Duration::from_secs(10)).await; } @@ -198,10 +187,7 @@ async fn test_foreground_ddl_no_recovery() -> Result<()> { sleep(Duration::from_secs(2)).await; // Kill CN should stop the job - cluster - .kill_nodes(["compute-1", "compute-2", "compute-3"], 0) - .await; - sleep(Duration::from_secs(10)).await; + kill_cn_and_wait_recover(&cluster).await; // Create MV should succeed, since the previous foreground job should be cancelled. session.run(SET_RATE_LIMIT_2).await?; @@ -257,7 +243,7 @@ async fn test_foreground_sink_cancel() -> Result<()> { let mut session2 = cluster.start_session(); tokio::spawn(async move { session2.run(SET_RATE_LIMIT_2).await.unwrap(); - let result = session2.run("CREATE SINK s ON t (v1);").await; + let result = session2.run("CREATE SINK s FROM t WITH (connector='blackhole');").await; assert!(result.is_err()); }); @@ -269,7 +255,7 @@ async fn test_foreground_sink_cancel() -> Result<()> { // Create MV should succeed, since the previous foreground job should be cancelled. session.run(SET_RATE_LIMIT_2).await?; - session.run("CREATE SINK s ON t (v1);").await?; + session.run("CREATE SINK s FROM t WITH (connector='blackhole');").await?; session.run("DROP SINK s;").await?; session.run(DROP_TABLE).await?;