Function run_roundtrip_tests

datafusion/core/tests/sql/unparser.rs:418–452

Runs roundtrip tests for a collection of queries and reports results. Iterates through all queries, running each through [`collect_results`]. Prints colored status (green checkmark for success, red X for failure) and panics at the end if any tests failed, with detailed error messages. # Type Parameters `F` - Factory function that creates fresh session contexts `Fut` - Future type returned by th

(
    suite_name: &str,
    queries: Vec<TestQuery>,
    create_context: F,
)

///
/// Panics if any query fails the roundtrip test, displaying all failures.
async fn run_roundtrip_tests<F, Fut>(
    suite_name: &str,
    queries: Vec<TestQuery>,
    create_context: F,
) where
    F: Fn() -> Fut,
    Fut: Future<Output = Result<SessionContext>>,
{
    let mut errors: Vec<String> = vec![];
    for sql in queries {
        let ctx = match create_context().await {
            Ok(ctx) => ctx,
            Err(e) => {
                println!("\x1b[31m✗\x1b[0m {} query: {}", suite_name, sql.name);
                errors.push(format!("Failed to create context for {}: {}", sql.name, e));
                continue;
            }
        };
        let result = collect_results(&ctx, &sql.sql).await;
        if result.is_failure() {
            println!("\x1b[31m✗\x1b[0m {} query: {}", suite_name, sql.name);
            errors.push(result.format_error(&sql.name));
        } else {
            println!("\x1b[32m✓\x1b[0m {} query: {}", suite_name, sql.name);
        }
    }
    if !errors.is_empty() {
        panic!(
            "{} {} test(s) failed:\n\n{}",
            errors.len(),
            suite_name,
            errors.join("\n\n---\n\n")
        );
    }
}

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