Function test_alias

datafusion/core/tests/dataframe/mod.rs:6154–6194

()

#[tokio::test]
async fn test_alias() -> Result<()> {
    let df = create_test_table("test")
        .await?
        .select(vec![col("a"), col("test.b"), lit(1).alias("one")])?
        .alias("table_alias")?;
    // All output column qualifiers are changed to "table_alias"
    df.schema().columns().iter().for_each(|c| {
        assert_eq!(c.relation, Some("table_alias".into()));
    });

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

    // Select over the aliased DataFrame
    let df = df.select(vec![
        col("table_alias.a"),
        col("b") + col("table_alias.one"),
    ])?;
    assert_snapshot!(
        batches_to_sort_string(&df.collect().await.unwrap()),
        @r"
    +-----------+---------------------------------+
    | a         | table_alias.b + table_alias.one |
    +-----------+---------------------------------+
    | 123AbcDef | 101                             |
    | CBAdef    | 11                              |
    | abc123    | 11                              |
    | abcDEF    | 2                               |
    +-----------+---------------------------------+
    "
    );
    Ok(())
}

#[tokio::test]
async fn test_alias_with_metadata() -> Result<()> {