Function test_nvl2

datafusion/core/tests/dataframe/dataframe_functions.rs:248–276

()

#[tokio::test]
async fn test_nvl2() -> Result<()> {
    let lit_null = lit(ScalarValue::Utf8(None));
    // nvl2(CASE WHEN a = 'abcDEF' THEN NULL ELSE a END, 'NON_NUll', 'TURNED_NULL')
    let expr = nvl2(
        when(col("a").eq(lit("abcDEF")), lit_null)
            .otherwise(col("a"))
            .unwrap(),
        lit("NON_NULL"),
        lit("TURNED_NULL"),
    )
    .alias("nvl2_expr");

    let batches = get_batches(expr).await?;

    assert_snapshot!(
        batches_to_string(&batches),
        @r"
    +-------------+
    | nvl2_expr   |
    +-------------+
    | TURNED_NULL |
    | NON_NULL    |
    | NON_NULL    |
    | NON_NULL    |
    +-------------+
    ");

    Ok(())
}

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