Function test_db_handler_transactions

tests/connection_pool_test.rs:30–71

()

#[tokio::test]
async fn test_db_handler_transactions() -> Result<(), Box<dyn std::error::Error>> {
    let temp_dir = tempfile::tempdir()?;
    let db_path = temp_dir.path().join("test_transactions.db");
    let db_handler = Arc::new(DbHandler::new(db_path.to_str().unwrap())?);
    
    // Create a session
    let session_id = Uuid::new_v4();
    db_handler.create_session_connection(session_id).await?;
    
    // Create test table
    db_handler.execute_with_session("CREATE TABLE test_tx (id INTEGER PRIMARY KEY, value INTEGER)", &session_id).await?;
    
    // Test successful transaction
    db_handler.begin_with_session(&session_id).await?;
    db_handler.execute_with_session("INSERT INTO test_tx (value) VALUES (100)", &session_id).await?;
    db_handler.execute_with_session("INSERT INTO test_tx (value) VALUES (200)", &session_id).await?;
    db_handler.commit_with_session(&session_id).await?;
    
    let result = db_handler.query_with_session("SELECT COUNT(*) FROM test_tx", &session_id).await?;
    assert_eq!(result.rows.len(), 1);
    // Check that we have 2 rows
    let count_bytes = result.rows[0][0].as_ref().unwrap();
    let count_str = std::str::from_utf8(count_bytes).unwrap();
    let count: i64 = count_str.parse().unwrap();
    assert_eq!(count, 2);
    
    // Test rollback
    db_handler.begin_with_session(&session_id).await?;
    db_handler.execute_with_session("INSERT INTO test_tx (value) VALUES (300)", &session_id).await?;
    db_handler.rollback_with_session(&session_id).await?;
    
    let result = db_handler.query_with_session("SELECT COUNT(*) FROM test_tx", &session_id).await?;
    let count_bytes = result.rows[0][0].as_ref().unwrap();
    let count_str = std::str::from_utf8(count_bytes).unwrap();
    let count: i64 = count_str.parse().unwrap();
    assert_eq!(count, 2); // Should still be 2
    
    // Clean up
    db_handler.remove_session_connection(&session_id);
    
    Ok(())
}

#[tokio::test(flavor = "multi_thread", worker_threads = 4)]
async fn test_db_handler_concurrent_access() -> Result<(), Box<dyn std::error::Error>> {