grafeo-engine 0.5.41

//! Tests for label mutation rollback behavior.
//!
//! Verifies that ADD/REMOVE label operations are correctly undone
//! when a transaction is rolled back.

use grafeo_engine::GrafeoDB;

#[test]
fn test_add_label_rollback_removes_label() {
    let db = GrafeoDB::new_in_memory();
    let mut session = db.session();

    session.execute("INSERT (:Person {name: 'Alix'})").unwrap();

    // Verify initial labels
    let result = session
        .execute("MATCH (p:Person {name: 'Alix'}) RETURN labels(p)")
        .unwrap();
    assert_eq!(result.row_count(), 1);

    // Begin transaction, add a label, then rollback
    session.begin_transaction().unwrap();
    session
        .execute("MATCH (p:Person {name: 'Alix'}) SET p:Employee")
        .unwrap();

    // Verify label was added within transaction
    let result = session
        .execute("MATCH (p:Employee {name: 'Alix'}) RETURN p.name")
        .unwrap();
    assert_eq!(
        result.row_count(),
        1,
        "should find node with Employee label in tx"
    );

    session.rollback().unwrap();

    // Label should be gone after rollback
    let result = session
        .execute("MATCH (p:Employee {name: 'Alix'}) RETURN p.name")
        .unwrap();
    assert_eq!(
        result.row_count(),
        0,
        "Employee label should not exist after rollback"
    );

    // Original label should still be there
    let result = session
        .execute("MATCH (p:Person {name: 'Alix'}) RETURN p.name")
        .unwrap();
    assert_eq!(result.row_count(), 1, "Person label should still exist");
}

#[test]
fn test_remove_label_rollback_restores_label() {
    let db = GrafeoDB::new_in_memory();
    let mut session = db.session();

    session
        .execute("INSERT (:Person:Employee {name: 'Gus'})")
        .unwrap();

    // Verify both labels exist
    let result = session
        .execute("MATCH (p:Employee {name: 'Gus'}) RETURN p.name")
        .unwrap();
    assert_eq!(result.row_count(), 1);

    // Begin transaction, remove a label, then rollback
    session.begin_transaction().unwrap();
    session
        .execute("MATCH (p:Person {name: 'Gus'}) REMOVE p:Employee")
        .unwrap();

    // Verify label was removed within transaction
    let result = session
        .execute("MATCH (p:Employee {name: 'Gus'}) RETURN p.name")
        .unwrap();
    assert_eq!(result.row_count(), 0, "Employee label should be gone in tx");

    session.rollback().unwrap();

    // Label should be restored after rollback
    let result = session
        .execute("MATCH (p:Employee {name: 'Gus'}) RETURN p.name")
        .unwrap();
    assert_eq!(
        result.row_count(),
        1,
        "Employee label should be restored after rollback"
    );
}

#[test]
fn test_add_label_committed_stays() {
    let db = GrafeoDB::new_in_memory();
    let mut session = db.session();

    session
        .execute("INSERT (:Person {name: 'Vincent'})")
        .unwrap();

    // Add label in a committed transaction
    session.begin_transaction().unwrap();
    session
        .execute("MATCH (p:Person {name: 'Vincent'}) SET p:VIP")
        .unwrap();
    session.commit().unwrap();

    // Label should persist
    let result = session
        .execute("MATCH (p:VIP {name: 'Vincent'}) RETURN p.name")
        .unwrap();
    assert_eq!(
        result.row_count(),
        1,
        "VIP label should persist after commit"
    );
}

// ============================================================================
// Label undo via transaction rollback: covers property_ops.rs LabelAdded/LabelRemoved
// ============================================================================

#[test]
fn test_label_add_undo_on_transaction_rollback() {
    let db = GrafeoDB::new_in_memory();
    let mut session = db.session();

    session
        .execute("INSERT (:Animal {species: 'Cat'})")
        .unwrap();

    session.begin_transaction().unwrap();
    session.execute("MATCH (n:Animal) SET n:Pet").unwrap();

    let during = session.execute("MATCH (n:Pet) RETURN n.species").unwrap();
    assert_eq!(during.rows().len(), 1);

    session.rollback().unwrap();

    let after = session.execute("MATCH (n:Pet) RETURN n.species").unwrap();
    assert!(
        after.rows().is_empty(),
        "Label should be removed after rollback"
    );
}

#[test]
fn test_label_remove_undo_on_transaction_rollback() {
    let db = GrafeoDB::new_in_memory();
    let mut session = db.session();

    session
        .execute("INSERT (:Animal:Pet {species: 'Dog'})")
        .unwrap();

    session.begin_transaction().unwrap();
    session.execute("MATCH (n:Animal) REMOVE n:Pet").unwrap();

    let during = session.execute("MATCH (n:Pet) RETURN n.species").unwrap();
    assert!(during.rows().is_empty());

    session.rollback().unwrap();

    let after = session.execute("MATCH (n:Pet) RETURN n.species").unwrap();
    assert_eq!(
        after.rows().len(),
        1,
        "Label should be restored after rollback"
    );
}