Function test_cte_with_arithmetic_operations

tests/subquery_cte_test.rs:193–215

()

#[test]
fn test_cte_with_arithmetic_operations() {
    let conn = setup_test_db();
    
    let sql = "WITH customer_metrics AS (
                   SELECT 
                       customer_id,
                       SUM(total) as total_amount,
                       SUM(tax) as total_tax,
                       SUM(total + tax) as total_with_tax
                   FROM orders 
                   GROUP BY customer_id
               )
               SELECT 
                   c.name,
                   cm.total_with_tax / cm.total_amount as tax_ratio
               FROM customers c 
               JOIN customer_metrics cm ON c.id = cm.customer_id";
    let result = rewrite_query(&conn, sql).unwrap();
    
    println!("CTE with arithmetic rewritten to: {result}");
    assert!(result.contains("decimal_add"));
    assert!(result.contains("decimal_div"));
}

#[test]
fn test_multiple_ctes() {