dust/tests/variables.rs

use dust_lang::{
    abstract_tree::{Block, Expression, Identifier, Statement, Type},
    error::{Error, TypeCheckError, 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(TypeCheckError {
                actual: Type::Boolean,
                expected: Type::String
            }),
            span: (0..18).into()
        }])
    );
}

#[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")
            ))])
        )))
    );
}