Function grouping_sets_explicit_shapes

nodedb/tests/sql_grouping_sets.rs:168–223

()

/// exactly those three row-shape categories.
#[tokio::test(flavor = "multi_thread", worker_threads = 4)]
async fn grouping_sets_explicit_shapes() {
    let server = TestServer::start().await;
    create_orders(&server).await;

    let rows = server
        .query_rows(
            "SELECT region, country, SUM(sales) AS total \
             FROM orders \
             GROUP BY GROUPING SETS ((region, country), (region), ())",
        )
        .await
        .unwrap();

    // Set 1: (region, country) — 5 rows (US×2 merged, CA, DE, FR) but US has 2 rows
    // so should be 4 unique combinations: US→300, CA→150, DE→300, FR→250
    // Set 2: (region) — 2 rows: AMER→450, EMEA→550
    // Set 3: () — 1 row: grand total 1000
    // Total = 7
    assert_eq!(
        rows.len(),
        7,
        "expected 7 rows for explicit GROUPING SETS, got {}: {:?}",
        rows.len(),
        rows
    );

    // Grand-total row (both NULL).
    let grand = rows
        .iter()
        .filter(|r| {
            r.first().map(|s| s.is_empty()).unwrap_or(false)
                && r.get(1).map(|s| s.is_empty()).unwrap_or(false)
        })
        .count();
    assert_eq!(grand, 1, "expected 1 grand-total row");

    // Regional rows (country NULL, region non-empty).
    let regional = rows
        .iter()
        .filter(|r| {
            r.first().map(|s| !s.is_empty()).unwrap_or(false)
                && r.get(1).map(|s| s.is_empty()).unwrap_or(false)
        })
        .count();
    assert_eq!(regional, 2, "expected 2 regional subtotals");

    // Fully-grouped rows (both non-empty).
    let detailed = rows
        .iter()
        .filter(|r| {
            r.first().map(|s| !s.is_empty()).unwrap_or(false)
                && r.get(1).map(|s| !s.is_empty()).unwrap_or(false)
        })
        .count();
    assert_eq!(detailed, 4, "expected 4 (region, country) rows");
}

/// `GROUPING(col)` returns 0 for real values and 1 for NULL-filled positions.