Function test_user_defined_table_function

datafusion/ffi/tests/ffi_udtf.rs:36–56

()

    /// using the abs() function as our scalar UDF.
    #[tokio::test]
    async fn test_user_defined_table_function() -> Result<()> {
        let module = get_module()?;
        let (ctx, codec) = super::utils::ctx_and_codec();

        let ffi_table_func = (module.create_table_function)(codec);
        let foreign_table_func: Arc<dyn TableFunctionImpl> = ffi_table_func.into();

        ctx.register_udtf("my_range", foreign_table_func);

        let result = ctx
            .sql("SELECT * FROM my_range(5)")
            .await?
            .collect()
            .await?;
        let expected = create_array!(Int64, [0, 1, 2, 3, 4]) as ArrayRef;

        assert!(result.len() == 1);
        assert!(result[0].column(0) == &expected);

        Ok(())
    }
}