Function array_agg_aggregate

nodedb/tests/executor_tests/test_array_ops.rs:193–235

()

#[test]
fn array_agg_aggregate() {
    let (mut core, mut tx, mut rx, _dir) = make_core();
    seed(&mut core, &mut tx, &mut rx);

    // GROUP BY brand, ARRAY_AGG(color).
    let payload = send_ok(
        &mut core,
        &mut tx,
        &mut rx,
        PhysicalPlan::Query(QueryOp::Aggregate {
            collection: "products".into(),
            group_by: vec!["brand".into()],
            aggregates: vec![AggregateSpec {
                function: "array_agg".into(),
                alias: "array_agg_color".into(),
                user_alias: None,
                field: "color".into(),
                expr: None,
            }],
            filters: Vec::new(),
            having: Vec::new(),
            limit: 100,
            sub_group_by: Vec::new(),
            sub_aggregates: Vec::new(),
            grouping_sets: Vec::new(),
            sort_keys: Vec::new(),
        }),
    );

    let v = payload_value(&payload);
    let rows: Vec<serde_json::Value> = if let serde_json::Value::Array(arr) = v {
        arr
    } else {
        serde_json::from_str(&payload_json(&payload)).unwrap_or_default()
    };

    // Nike has 2 products: colors ["red", "blue"].
    let nike = rows.iter().find(|r| r["brand"] == "Nike").unwrap();
    let colors = nike["array_agg_color"].as_array().unwrap();
    assert_eq!(colors.len(), 2);
    assert!(colors.contains(&serde_json::json!("red")));
    assert!(colors.contains(&serde_json::json!("blue")));
}

#[test]
fn array_length_function() {