dust/tests/variables.rs

use dust_lang::{
    abstract_tree::{Block, Expression, Identifier, Statement, Type},
    error::{Error, TypeConflict, ValidationError},
    *,
};

#[test]
fn set_and_get_variable() {
    assert_eq!(
        interpret("foobar = true; foobar"),
        Ok(Some(Value::boolean(true)))
    );
}

#[test]
fn set_variable_with_type() {
    assert_eq!(
        interpret("foobar: bool = true; foobar"),
        Ok(Some(Value::boolean(true)))
    );
}

#[test]
fn set_variable_with_type_error() {
    assert_eq!(
        interpret("foobar: str = true"),
        Err(vec![Error::Validation {
            error: ValidationError::TypeCheck(TypeConflict {
                actual: Type::Boolean,
                expected: Type::String
            }),
            span: (0, 18)
        }])
    );
}

#[test]
fn function_variable() {
    assert_eq!(
        interpret("foobar = (x: int): int { x }; foobar"),
        Ok(Some(Value::function(
            vec![(Identifier::new("x"), Type::Integer)],
            Type::Integer,
            Block::new(vec![Statement::expression(
                Expression::Identifier(Identifier::new("x")),
                (0..0).into()
            )])
        )))
    );
}