Function test_transactions

tests/query_test.rs:129–188

()

#[tokio::test]
async fn test_transactions() {
    // Use a temporary file instead of in-memory database
    let test_id = Uuid::new_v4().to_string().replace("-", "");
    let db_path = format!("/tmp/pgsqlite_test_{test_id}.db");
    let db_path_clone = db_path.clone();
    
    // Start test server
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let port = listener.local_addr().unwrap().port();
    
    let server_handle = tokio::spawn(async move {
        let db_handler = std::sync::Arc::new(
            pgsqlite::session::DbHandler::new(&db_path_clone).unwrap()
        );
        
        let (stream, addr) = listener.accept().await.unwrap();
        pgsqlite::handle_test_connection_with_pool(stream, addr, db_handler).await.unwrap();
    });
    
    tokio::time::sleep(Duration::from_millis(100)).await;
    
    // Connect with tokio-postgres
    let (client, connection) = timeout(
        Duration::from_secs(5),
        tokio_postgres::connect(
            &format!("host=localhost port={port} dbname=test user=testuser"),
            NoTls,
        )
    ).await.unwrap().unwrap();
    
    tokio::spawn(connection);
    
    // Create table
    client.execute("CREATE TABLE counter (value INTEGER)", &[]).await.unwrap();
    client.execute("INSERT INTO counter VALUES (0)", &[]).await.unwrap();
    
    // Test transaction commit
    client.execute("BEGIN", &[]).await.unwrap();
    client.execute("UPDATE counter SET value = 1", &[]).await.unwrap();
    client.execute("COMMIT", &[]).await.unwrap();
    
    let rows = client.query("SELECT value FROM counter", &[]).await.unwrap();
    assert_eq!(rows[0].get::<_, i32>(0), 1);
    
    // Test transaction rollback
    client.execute("BEGIN", &[]).await.unwrap();
    client.execute("UPDATE counter SET value = 2", &[]).await.unwrap();
    client.execute("ROLLBACK", &[]).await.unwrap();
    
    let rows = client.query("SELECT value FROM counter", &[]).await.unwrap();
    assert_eq!(rows[0].get::<_, i32>(0), 1); // Should still be 1
    
    server_handle.abort();
    
    // Clean up
    let _ = std::fs::remove_file(&db_path);
    let _ = std::fs::remove_file(format!("{db_path}-journal"));
    let _ = std::fs::remove_file(format!("{db_path}-wal"));