102 lines
2.9 KiB
Rust
102 lines
2.9 KiB
Rust
use dust_lang::*;
|
|
|
|
#[test]
|
|
fn define_local() {
|
|
let source = "let x = 42;";
|
|
|
|
assert_eq!(
|
|
compile(source),
|
|
Ok(Chunk::with_data(
|
|
None,
|
|
FunctionType {
|
|
type_parameters: None,
|
|
value_parameters: None,
|
|
return_type: Box::new(Type::None),
|
|
},
|
|
vec![
|
|
(
|
|
Instruction::load_constant(Destination::Register(0), 0, false),
|
|
Type::Integer,
|
|
Span(8, 10)
|
|
),
|
|
(
|
|
Instruction::define_local(0, 0, false),
|
|
Type::None,
|
|
Span(4, 5)
|
|
),
|
|
(Instruction::r#return(false), Type::None, Span(11, 11))
|
|
],
|
|
vec![ConcreteValue::Integer(42), ConcreteValue::string("x")],
|
|
vec![Local::new(1, Type::Integer, false, Scope::default())]
|
|
)),
|
|
);
|
|
|
|
assert_eq!(run(source), Ok(None));
|
|
}
|
|
|
|
#[test]
|
|
fn let_statement_expects_identifier() {
|
|
let source = "let 1 = 2";
|
|
|
|
assert_eq!(
|
|
compile(source),
|
|
Err(DustError::Compile {
|
|
error: CompileError::ExpectedToken {
|
|
expected: TokenKind::Identifier,
|
|
found: Token::Integer("1").to_owned(),
|
|
position: Span(4, 5)
|
|
},
|
|
source
|
|
})
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn set_local() {
|
|
let source = "let mut x = 41; x = 42; x";
|
|
|
|
assert_eq!(
|
|
compile(source),
|
|
Ok(Chunk::with_data(
|
|
None,
|
|
FunctionType {
|
|
type_parameters: None,
|
|
value_parameters: None,
|
|
return_type: Box::new(Type::Integer),
|
|
},
|
|
vec![
|
|
(
|
|
Instruction::load_constant(Destination::Register(0), 0, false),
|
|
Type::Integer,
|
|
Span(12, 14)
|
|
),
|
|
(
|
|
Instruction::define_local(0, 0, true),
|
|
Type::None,
|
|
Span(8, 9)
|
|
),
|
|
(
|
|
Instruction::load_constant(Destination::Register(1), 2, false),
|
|
Type::Integer,
|
|
Span(20, 22)
|
|
),
|
|
(Instruction::set_local(1, 0), Type::None, Span(16, 17)),
|
|
(
|
|
Instruction::get_local(Destination::Register(2), 0),
|
|
Type::Integer,
|
|
Span(24, 25)
|
|
),
|
|
(Instruction::r#return(true), Type::None, Span(25, 25)),
|
|
],
|
|
vec![
|
|
ConcreteValue::Integer(41),
|
|
ConcreteValue::string("x"),
|
|
ConcreteValue::Integer(42)
|
|
],
|
|
vec![Local::new(1, Type::Integer, true, Scope::default())]
|
|
)),
|
|
);
|
|
|
|
assert_eq!(run(source), Ok(Some(ConcreteValue::Integer(42))));
|
|
}
|