Function test_count_with_id_argument

graphlite/tests/function_tests.rs:221–261

()

#[test]
fn test_count_with_id_argument() {
    let fixture = TestFixture::new().expect("Failed to create test fixture");
    // Setup fresh graph for this test to avoid interference
    fixture
        .setup_graph("test_count_with_id_argument")
        .expect("Failed to setup graph");
    // Re-insert fraud data since we have a fresh graph
    fixture
        .insert_fraud_data()
        .expect("Failed to insert fraud data");

    // Test COUNT(account) vs COUNT(account.id)
    let result1 = fixture.assert_query_succeeds(
        "MATCH (account:Account) WHERE account.balance > 2000 RETURN COUNT(account) as count_node",
    );

    let result2 = fixture.assert_query_succeeds(
        "MATCH (account:Account) WHERE account.balance > 2000 RETURN COUNT(account.id) as count_id",
    );

    // Both should return the same count since id is never null
    if let (Some(row1), Some(row2)) = (result1.rows.first(), result2.rows.first()) {
        let count_node = row1
            .values
            .get("count_node")
            .expect("Should have count_node");
        let count_id = row2.values.get("count_id").expect("Should have count_id");

        if let (Value::Number(count1), Value::Number(count2)) = (count_node, count_id) {
            assert_eq!(
                count1, count2,
                "COUNT(account) and COUNT(account.id) should be equal"
            );
            assert!(*count1 >= 0.0, "Count should be non-negative");
        } else {
            panic!("Both COUNT operations should return Numbers");
        }
    } else {
        panic!("Both queries should return at least one row");
    }
}

#[test]
fn test_sum_end_to_end_with_where_clause() {