Function test_pg_class_queries

tests/catalog_test.rs:78–124

()

#[tokio::test]
async fn test_pg_class_queries() {
    // Create a test database handler with temporary file
    let timestamp = SystemTime::now().duration_since(UNIX_EPOCH).unwrap().as_nanos();
    let db_path = format!("/tmp/catalog_test_pg_class_{timestamp}.db");
    let db = Arc::new(DbHandler::new(&db_path).unwrap());
    
    // Create a session
    let session = Arc::new(SessionState::new("test_user".to_string(), "test_db".to_string()));
    
    // Create a connection for the session
    db.create_session_connection(session.id).await.unwrap();
    
    // Create a test table in the session
    db.execute_with_session("CREATE TABLE test_table (id INTEGER PRIMARY KEY, name TEXT)", &session.id).await.unwrap();
    
    // Test pg_class query
    let query = "SELECT relname, relkind FROM pg_catalog.pg_class";
    let result = CatalogInterceptor::intercept_query(query, db.clone(), Some(session.clone())).await;
    assert!(result.is_some());
    
    let response = result.unwrap().unwrap();
    // Now we properly implement column projection
    assert_eq!(response.columns, vec!["relname", "relkind"]);
    assert_eq!(response.columns.len(), 2);
    
    // Find our test table
    let mut found_table = false;
    for row in &response.rows {
        assert_eq!(row.len(), 2, "Should only have 2 columns");
        if let Some(Some(name_bytes)) = row.first() { // relname is at index 0 now
            let name = String::from_utf8_lossy(name_bytes);
            if name == "test_table" {
                found_table = true;
                // Check relkind is 'r' for regular table
                if let Some(Some(kind_bytes)) = row.get(1) { // relkind is at index 1 now
                    assert_eq!(kind_bytes, b"r");
                }
            }
        }
    }
    assert!(found_table, "test_table should be in pg_class");
    
    // Cleanup
    let _ = std::fs::remove_file(&db_path);
    let _ = std::fs::remove_file(format!("{db_path}-wal"));
    let _ = std::fs::remove_file(format!("{db_path}-shm"));
}

#[tokio::test]
async fn test_pg_attribute_queries() {