use crate::engine::fold::expr_and::fold_expr_and;
use toasty_core::stmt::{Expr, ExprAnd};
fn nested_and(a: Expr, b: Expr, c: Expr) -> ExprAnd {
ExprAnd {
operands: vec![
a,
Expr::And(ExprAnd {
operands: vec![b, c],
}),
],
}
}
#[test]
fn flatten_all_symbolic() {
let mut expr = nested_and(Expr::arg(0), Expr::arg(1), Expr::arg(2));
let result = fold_expr_and(&mut expr);
assert!(result.is_none()); assert_eq!(expr.operands.len(), 3);
assert_eq!(expr.operands[0], Expr::arg(0));
assert_eq!(expr.operands[1], Expr::arg(1));
assert_eq!(expr.operands[2], Expr::arg(2));
}
#[test]
fn flatten_with_true_in_outer() {
let mut expr = nested_and(true.into(), Expr::arg(1), Expr::arg(2));
let result = fold_expr_and(&mut expr);
assert!(result.is_none());
assert_eq!(expr.operands.len(), 2);
assert_eq!(expr.operands[0], Expr::arg(1));
assert_eq!(expr.operands[1], Expr::arg(2));
}
#[test]
fn flatten_with_true_in_nested_first() {
let mut expr = nested_and(Expr::arg(0), true.into(), Expr::arg(2));
let result = fold_expr_and(&mut expr);
assert!(result.is_none());
assert_eq!(expr.operands.len(), 2);
assert_eq!(expr.operands[0], Expr::arg(0));
assert_eq!(expr.operands[1], Expr::arg(2));
}
#[test]
fn flatten_with_true_in_nested_second() {
let mut expr = nested_and(Expr::arg(0), Expr::arg(1), true.into());
let result = fold_expr_and(&mut expr);
assert!(result.is_none());
assert_eq!(expr.operands.len(), 2);
assert_eq!(expr.operands[0], Expr::arg(0));
assert_eq!(expr.operands[1], Expr::arg(1));
}
#[test]
fn flatten_outer_true_nested_one_true() {
let mut expr = nested_and(true.into(), true.into(), Expr::arg(2));
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert_eq!(result.unwrap(), Expr::arg(2));
}
#[test]
fn flatten_outer_symbolic_nested_all_true() {
let mut expr = nested_and(Expr::arg(0), true.into(), true.into());
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert_eq!(result.unwrap(), Expr::arg(0));
}
#[test]
fn flatten_all_true() {
let mut expr = nested_and(true.into(), true.into(), true.into());
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_true());
}
#[test]
fn flatten_with_false_in_outer() {
let mut expr = nested_and(false.into(), Expr::arg(1), Expr::arg(2));
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_false());
}
#[test]
fn flatten_with_false_in_nested() {
let mut expr = nested_and(Expr::arg(0), false.into(), Expr::arg(2));
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_false());
}
#[test]
fn flatten_true_and_false_mixed() {
let mut expr = nested_and(true.into(), false.into(), true.into());
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_false());
}
#[test]
fn single_operand_unwrapped() {
let mut expr = ExprAnd {
operands: vec![Expr::arg(0)],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert_eq!(result.unwrap(), Expr::arg(0));
}
#[test]
fn empty_after_removing_true() {
let mut expr = ExprAnd {
operands: vec![true.into(), true.into()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_true());
}
#[test]
fn null_and_null_becomes_null() {
let mut expr = ExprAnd {
operands: vec![Expr::null(), Expr::null()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_value_null());
}
#[test]
fn null_and_true_becomes_null() {
let mut expr = ExprAnd {
operands: vec![Expr::null(), true.into()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_value_null());
}
#[test]
fn true_and_null_becomes_null() {
let mut expr = ExprAnd {
operands: vec![true.into(), Expr::null()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_value_null());
}
#[test]
fn null_and_false_becomes_false() {
let mut expr = ExprAnd {
operands: vec![Expr::null(), false.into()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_false());
}
#[test]
fn false_and_null_becomes_false() {
let mut expr = ExprAnd {
operands: vec![false.into(), Expr::null()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_false());
}
#[test]
fn null_and_symbolic_not_simplified() {
let mut expr = ExprAnd {
operands: vec![Expr::null(), Expr::arg(0)],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_none());
assert_eq!(expr.operands.len(), 2);
}
#[test]
fn multiple_nulls_become_null() {
let mut expr = ExprAnd {
operands: vec![Expr::null(), Expr::null(), Expr::null()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_value_null());
}
#[test]
fn multiple_nulls_and_true_becomes_null() {
let mut expr = ExprAnd {
operands: vec![Expr::null(), true.into(), Expr::null(), true.into()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_value_null());
}
#[test]
fn error_operand_not_simplified_in_and() {
let mut expr = ExprAnd {
operands: vec![Expr::error("boom"), Expr::arg(0)],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_none());
assert_eq!(expr.operands.len(), 2);
}
#[test]
fn error_and_true_keeps_error() {
let mut expr = ExprAnd {
operands: vec![Expr::error("boom"), true.into()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(matches!(&result.unwrap(), Expr::Error(e) if e.message == "boom"));
}
#[test]
fn error_and_false_becomes_false() {
let mut expr = ExprAnd {
operands: vec![Expr::error("boom"), false.into()],
};
let result = fold_expr_and(&mut expr);
assert!(result.is_some());
assert!(result.unwrap().is_false());
}