Function test_alias_nested

datafusion/core/tests/dataframe/mod.rs:6266–6311

()

#[tokio::test]
async fn test_alias_nested() -> Result<()> {
    let df = create_test_table("test")
        .await?
        .select(vec![col("a"), col("test.b"), lit(1).alias("one")])?
        .alias("alias1")?
        .alias("alias2")?;

    let plan = df
        .clone()
        .into_optimized_plan()?
        .display_indent_schema()
        .to_string();
    assert_snapshot!(plan, @r"
    SubqueryAlias: alias2 [a:Utf8, b:Int32, one:Int32]
      SubqueryAlias: alias1 [a:Utf8, b:Int32, one:Int32]
        Projection: test.a, test.b, Int32(1) AS one [a:Utf8, b:Int32, one:Int32]
          TableScan: test projection=[a, b] [a:Utf8, b:Int32]
    ");

    // Select over the aliased DataFrame
    let select1 = df
        .clone()
        .select(vec![col("alias2.a"), col("b") + col("alias2.one")])?;

    assert_snapshot!(
        batches_to_sort_string(&select1.collect().await.unwrap()),
        @r"
    +-----------+-----------------------+
    | a         | alias2.b + alias2.one |
    +-----------+-----------------------+
    | 123AbcDef | 101                   |
    | CBAdef    | 11                    |
    | abc123    | 11                    |
    | abcDEF    | 2                     |
    +-----------+-----------------------+
    "
    );

    // Only the outermost alias is visible
    let select2 = df.select(vec![col("alias1.a")]);
    assert_snapshot!(
        select2.unwrap_err().strip_backtrace(),
        @"Schema error: No field named alias1.a. Valid fields are alias2.a, alias2.b, alias2.one."
    );
    Ok(())
}

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