Function test_pg_attribute_queries

tests/catalog_test.rs:127–183

()

#[tokio::test]
async fn test_pg_attribute_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_attr_{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 NOT NULL, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP)", &session.id).await.unwrap();
    
    // Test pg_attribute query
    let query = "SELECT attname, atttypid, attnotnull FROM pg_catalog.pg_attribute";
    let result = CatalogInterceptor::intercept_query(query, db.clone(), Some(session.clone())).await;
    assert!(result.is_some());
    
    let response = result.unwrap().unwrap();
    // With column projection, we now only get the requested columns
    assert_eq!(response.columns, vec!["attname", "atttypid", "attnotnull"]);
    
    // Count columns for test_table
    let mut column_count = 0;
    let mut found_id = false;
    let mut found_name = false;
    
    for row in &response.rows {
        if let Some(Some(name_bytes)) = row.first() { // attname is at index 0 (first selected column)
            let col_name = String::from_utf8_lossy(name_bytes);
            if col_name == "id" {
                found_id = true;
                column_count += 1;
            } else if col_name == "name" {
                found_name = true;
                column_count += 1;
                // Check NOT NULL constraint
                if let Some(Some(notnull_bytes)) = row.get(2) { // attnotnull is at index 2 (third selected column)
                    assert_eq!(notnull_bytes, b"t");
                }
            } else if col_name == "created_at" {
                column_count += 1;
            }
        }
    }
    
    assert!(found_id, "id column should be in pg_attribute");
    assert!(found_name, "name column should be in pg_attribute");
    assert!(column_count >= 3, "Should have at least 3 columns for test_table");
    
    // 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"));
}