use dala::{eval, eval_with_data, DalaError, DalaValue};
use std::collections::HashMap;

#[test]
fn test_sum() {
    let result = eval("SUM(1, 2, 3, 4, -5)");
    assert_eq!(result.len(), 1);

    let DalaValue::Num(value) = result[0].as_ref().unwrap() else { panic!("Not a number") };
    assert_eq!(value.to_owned(), 5.0);
}

#[test]
fn test_sum_with_single_arg() {
    let result = eval("SUM(1)");
    assert_eq!(result.len(), 1);

    let DalaValue::Num(value) = result[0].as_ref().unwrap() else { panic!("Not a number") };
    assert_eq!(value.to_owned(), 1.0);
}

#[test]
fn test_sum_runtime_error_with_str_arg() {
    let result = eval("SUM(1, UPPER(\"2\"))");
    assert_eq!(result.len(), 1);

    let DalaError::RuntimeError(err) = result[0].as_ref().unwrap_err() else { panic!("Not a runtime error") };
    assert_eq!(err.message, "Cannot sum strings, found: 2");
}

#[test]
fn test_with_data() {
    let data = &HashMap::from([
        ("$othervar", &DalaValue::Num(3.0)),
        ("$1", &DalaValue::Num(1.5)),
        ("$2", &DalaValue::Num(2.0)),
    ]);
    let result = eval_with_data("SUM($1, $2, $othervar, 4, -5)", data);
    assert_eq!(result.len(), 1);

    let DalaValue::Num(value) = result[0].as_ref().unwrap() else { panic!("Not a runtime error") };
    assert_eq!(value.to_owned(), 5.5);
}

#[test]
fn test_with_invalid_data() {
    let foo = DalaValue::Str("foo".to_string());
    let one = DalaValue::Num(1.5);
    let data = HashMap::from([("$foo", &foo), ("$1", &one)]);

    let result = eval_with_data("SUM($1, $foo, 2)", &data);
    assert_eq!(result.len(), 1);

    let DalaError::RuntimeError(err) = result[0].as_ref().unwrap_err() else { panic!("Not a runtime error") };
    assert_eq!(err.message, "Cannot sum strings, found: foo");
}