Function cleanup_executes_cleanup_queries

benchmarks/src/sql_benchmark.rs:2867–2894

()

    #[tokio::test]
    async fn cleanup_executes_cleanup_queries() {
        let mut benchmark = parse_benchmark(
            r#"
run
SELECT 1;

cleanup
CREATE TABLE cleanup_marker_a AS SELECT 7 AS value;

cleanup
CREATE TABLE cleanup_marker_b AS SELECT value + 1 AS value FROM cleanup_marker_a;
"#,
        )
        .await
        .expect("benchmark should parse");
        let ctx = SessionContext::new();

        benchmark.cleanup(&ctx).await.expect("cleanup should run");

        let rows = ctx
            .sql("SELECT value FROM cleanup_marker_b")
            .await
            .expect("query should plan")
            .collect()
            .await
            .expect("query should run");
        assert_eq!(format_record_batches(&rows).unwrap(), vec![vec!["8"]]);
    }

    #[tokio::test]
    async fn cleanup_propagates_query_failures() {