Function test_call_with_description_in_yield

graphlite/src/ast/parser.rs:5157–5187

()

#[test]
fn test_call_with_description_in_yield() {
    // Bug #1: 'description' is a keyword but should be allowed as column name in YIELD
    let query = r#"CALL system.list_functions()
YIELD name, category, description;"#;

    let result = parse_query(query);
    assert!(
        result.is_ok(),
        "Failed to parse CALL with 'description' in YIELD"
    );

    let doc = result.unwrap();
    if let Statement::Call(ref call_stmt) = doc.statement {
        assert_eq!(call_stmt.procedure_name, "system.list_functions");

        let yield_clause = call_stmt
            .yield_clause
            .as_ref()
            .expect("YIELD clause should be present");

        assert_eq!(yield_clause.items.len(), 3, "Should have 3 YIELD items");
        assert_eq!(yield_clause.items[0].column_name, "name");
        assert_eq!(yield_clause.items[1].column_name, "category");
        assert_eq!(yield_clause.items[2].column_name, "description");
    } else {
        panic!(
            "Expected CallStatement, got {:?}",
            std::mem::discriminant(&doc.statement)
        );
    }
}

#[test]
fn test_call_with_where_and_description() {