use lemma::DateTimeValue;
use lemma::{ValueKind, *};
use rust_decimal::Decimal;
use std::collections::HashMap;
#[test]
fn test_percentage_arithmetic() {
let code = r#"
spec pricing
data discount: 25%
rule net_multiplier: 1 - discount
"#;
let mut engine = Engine::new();
engine.load(code, lemma::SourceType::Volatile).unwrap();
let now = DateTimeValue::now();
let response = engine
.run(None, "pricing", Some(&now), HashMap::new(), true)
.unwrap();
let result = response
.results
.get("net_multiplier")
.unwrap()
.trace
.as_ref()
.expect("explanation")
.result
.value()
.expect("value");
match result {
LiteralValue {
value: ValueKind::Number(n),
..
} => assert_eq!(
lemma::ValueKind::Number(*n).as_decimal_magnitude().unwrap(),
Decimal::new(75, 2)
),
_ => panic!("Expected Number, got {:?}", result),
}
}
#[test]
fn test_duration_operations() {
let code = r#"
spec scheduling
uses lemma units
data meeting_length: 30 minutes
rule double_meeting: meeting_length * 2
"#;
let mut engine = Engine::new();
engine.load(code, lemma::SourceType::Volatile).unwrap();
let now = DateTimeValue::now();
let response = engine
.run(None, "scheduling", Some(&now), HashMap::new(), true)
.unwrap();
let rule = response.results.get("double_meeting").unwrap();
let quantity = rule.quantity.as_ref().expect("quantity map");
assert_eq!(
quantity.get("minutes").map(String::as_str),
Some("60"),
"30 minutes * 2 = 60 minutes"
);
}
#[test]
fn test_date_arithmetic_with_duration() {
let code = r#"
spec dates
uses lemma units
data start: 2024-01-15
rule end: start + 7 days
"#;
let mut engine = Engine::new();
engine.load(code, lemma::SourceType::Volatile).unwrap();
let now = DateTimeValue::now();
let response = engine
.run(None, "dates", Some(&now), HashMap::new(), true)
.unwrap();
let result = response
.results
.get("end")
.unwrap()
.trace
.as_ref()
.expect("explanation")
.result
.value()
.expect("value");
match result {
LiteralValue {
value: ValueKind::Date(dt),
..
} => {
assert_eq!(dt.year, 2024);
assert_eq!(dt.month, 1);
assert_eq!(dt.day, 22);
}
_ => panic!("Expected Date, got {:?}", result),
}
}
#[test]
fn test_boolean_operations() {
let code = r#"
spec logic
data is_active: true
data is_premium: false
rule can_access: is_active and not is_premium
"#;
let mut engine = Engine::new();
engine.load(code, lemma::SourceType::Volatile).unwrap();
let now = DateTimeValue::now();
let response = engine
.run(None, "logic", Some(&now), HashMap::new(), true)
.unwrap();
let result = response
.results
.get("can_access")
.unwrap()
.trace
.as_ref()
.expect("explanation")
.result
.value()
.expect("value");
match result {
LiteralValue {
value: ValueKind::Boolean(b),
..
} => {
assert!(b);
}
_ => panic!("Expected Boolean, got {:?}", result),
}
}