Function collect_results

datafusion/core/tests/sql/unparser.rs:310–402

Executes a roundtrip test for a single SQL query. This is the core test logic that: 1. Parses the original SQL and creates a logical plan 2. Unparses the logical plan back to SQL 3. Executes both the original and unparsed queries 4. Compares the results (sorting if the query has no ORDER BY) This always uses [`DefaultDialect`] for unparsing. # Arguments `ctx` - Session context with tables regi

(ctx: &SessionContext, original: &str)

///
/// A [`TestCaseResult`] indicating success or the specific failure mode.
async fn collect_results(ctx: &SessionContext, original: &str) -> TestCaseResult {
    let unparser = Unparser::new(&DefaultDialect {});

    // Parse and create logical plan from original SQL
    let df = match ctx.sql(original).await {
        Ok(df) => df,
        Err(e) => {
            return TestCaseResult::ExecutionError {
                original: original.to_string(),
                error: e.to_string(),
            };
        }
    };

    // Unparse the logical plan back to SQL
    let unparsed = match unparser.plan_to_sql(df.logical_plan()) {
        Ok(sql) => format!("{sql:#}"),
        Err(e) => {
            return TestCaseResult::UnparseError {
                original: original.to_string(),
                error: e.to_string(),
            };
        }
    };

    // Collect results from original query
    let mut expected = match df.collect().await {
        Ok(batches) => batches,
        Err(e) => {
            return TestCaseResult::ExecutionError {
                original: original.to_string(),
                error: e.to_string(),
            };
        }
    };

    // Parse and execute the unparsed SQL
    let actual_df = match ctx.sql(&unparsed).await {
        Ok(df) => df,
        Err(e) => {
            return TestCaseResult::UnparsedExecutionError {
                original: original.to_string(),
                unparsed,
                error: e.to_string(),
            };
        }
    };

    // Collect results from unparsed query
    let mut actual = match actual_df.collect().await {
        Ok(batches) => batches,
        Err(e) => {
            return TestCaseResult::UnparsedExecutionError {
                original: original.to_string(),
                unparsed,
                error: e.to_string(),
            };
        }