Skip to content

Commit

Permalink
add full tree test
Browse files Browse the repository at this point in the history
  • Loading branch information
@SkymanOne
SkymanOne committed Jan 30, 2024
1 parent fcfa90c commit b3b271f
Showing 1 changed file with 236 additions and 5 deletions.
241 changes: 236 additions & 5 deletions crates/parser/src/tests.rs
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,8 @@
use crate::ast::{Declaration, FuncReturnType, Statement, TypeVariant};
use crate::ast::{
self, AccessAttribute, BinaryExpression, Declaration, Expression, FuncReturnType, FunctionCall,
FunctionDeclaration, FunctionVisibility, Identifier, IfElse, Param, Source, StBlock,
StateDeclaration, Statement, StatementBlock, StructInit, TypeVariant, UnaryExpression,
};
use crate::folidity;
use crate::lexer::{Lexer, Token};

Expand Down Expand Up @@ -105,10 +109,7 @@ st [
@(any)
fn int get_factorial(value: int)
st value < 100
{
return calculate(value);
}
st value < 100 = return calculate(value);
"#;

#[test]
Expand All @@ -130,3 +131,233 @@ fn test_factorial() {
assert_eq!(state.st_block, None);
}
}

#[test]
fn test_factorial_tree() {
let mut lexer_errors = Vec::new();
let tokens = Lexer::new(FACTORIAL_SRC, &mut lexer_errors);
let mut parser_errors = Vec::new();
let tree = folidity::FolidityTreeParser::new()
.parse(&mut parser_errors, tokens)
.unwrap();
let parsed = Source {
declarations: vec![
Declaration::StateDeclaration(Box::new(StateDeclaration {
loc: 1..18,
name: Identifier {
loc: 7..17,
name: "EmptyState".to_string(),
},
body: None,
from: None,
st_block: None,
})),
Declaration::FunDeclaration(Box::new(FunctionDeclaration {
loc: 19..351,
is_init: false,
access_attributes: vec![],
vis: FunctionVisibility::Priv,
return_ty: FuncReturnType::ParamType(Param {
loc: 23..31,
ty: ast::Type {
loc: 28..31,
ty: TypeVariant::Int,
},
name: Identifier {
loc: 23..26,
name: "out".to_string(),
},
is_mut: true,
}),
name: Identifier {
loc: 33..42,
name: "calculate".to_string(),
},
params: vec![Param {
loc: 43..53,
ty: ast::Type {
loc: 50..53,
ty: TypeVariant::Int,
},
name: Identifier {
loc: 43..48,
name: "value".to_string(),
},
is_mut: true,
}],
state_bound: None,
st_block: Some(StBlock {
loc: 64..90,
exprs: vec![
Expression::Greater(BinaryExpression {
loc: 64..73,
left: Box::new(Expression::Variable(Identifier {
loc: 64..69,
name: "value".to_string(),
})),
right: Box::new(Expression::Number(UnaryExpression {
loc: 72..73,
element: "0".to_string(),
})),
}),
Expression::Less(BinaryExpression {
loc: 79..90,
left: Box::new(Expression::Variable(Identifier {
loc: 79..82,
name: "out".to_string(),
})),
right: Box::new(Expression::Number(UnaryExpression {
loc: 85..90,
element: "10000".to_string(),
})),
}),
],
}),
body: Statement::Block(StatementBlock {
loc: 93..351,
statements: vec![Statement::IfElse(IfElse {
loc: 99..349,
condition: Expression::Equal(BinaryExpression {
loc: 102..112,
left: Box::new(Expression::Variable(Identifier {
loc: 102..107,
name: "value".to_string(),
})),
right: Box::new(Expression::Number(UnaryExpression {
loc: 111..112,
element: "1".to_string(),
})),
}),
body: Box::new(StatementBlock {
loc: 113..169,
statements: vec![
Statement::StateTransition(StructInit {
loc: 128..140,
name: Identifier {
loc: 128..138,
name: "EmptyState".to_string(),
},
args: vec![],
auto_object: None,
}),
Statement::Return(Expression::Variable(Identifier {
loc: 157..162,
name: "value".to_string(),
})),
],
}),
else_part: Some(Box::new(Statement::Block(StatementBlock {
loc: 175..349,
statements: vec![Statement::Return(Expression::FunctionCall(
FunctionCall {
loc: 192..342,
name: Identifier {
loc: 192..201,
name: "calculate".to_string(),
},
args: vec![Expression::Pipe(BinaryExpression {
loc: 296..324,
left: Box::new(Expression::Multiply(BinaryExpression {
loc: 296..315,
left: Box::new(Expression::Variable(Identifier {
loc: 296..301,
name: "value".to_string(),
})),
right: Box::new(Expression::Subtract(
BinaryExpression {
loc: 305..314,
left: Box::new(Expression::Variable(
Identifier {
loc: 305..310,
name: "value".to_string(),
},
)),
right: Box::new(Expression::Number(
UnaryExpression {
loc: 313..314,
element: "1".to_string(),
},
)),
},
)),
})),
right: Box::new(Expression::FunctionCall(FunctionCall {
loc: 319..324,
name: Identifier {
loc: 319..321,
name: "or".to_string(),
},
args: vec![Expression::Number(UnaryExpression {
loc: 322..323,
element: "1".to_string(),
})],
})),
})],
},
))],
}))),
})],
}),
})),
Declaration::FunDeclaration(Box::new(FunctionDeclaration {
loc: 353..434,
is_init: false,
access_attributes: vec![AccessAttribute {
loc: 353..359,
members: vec![Expression::Variable(Identifier {
loc: 355..358,
name: "any".to_string(),
})],
}],
vis: FunctionVisibility::Pub,
return_ty: FuncReturnType::Type(ast::Type {
loc: 363..366,
ty: TypeVariant::Int,
}),
name: Identifier {
loc: 367..380,
name: "get_factorial".to_string(),
},
params: vec![Param {
loc: 381..391,
ty: ast::Type {
loc: 388..391,
ty: TypeVariant::Int,
},
name: Identifier {
loc: 381..386,
name: "value".to_string(),
},
is_mut: true,
}],
state_bound: None,
st_block: Some(StBlock {
loc: 393..407,
exprs: vec![Expression::Less(BinaryExpression {
loc: 396..407,
left: Box::new(Expression::Variable(Identifier {
loc: 396..401,
name: "value".to_string(),
})),
right: Box::new(Expression::Number(UnaryExpression {
loc: 404..407,
element: "100".to_string(),
})),
})],
}),
body: Statement::Return(Expression::FunctionCall(FunctionCall {
loc: 417..433,
name: Identifier {
loc: 417..426,
name: "calculate".to_string(),
},
args: vec![Expression::Variable(Identifier {
loc: 427..432,
name: "value".to_string(),
})],
})),
})),
],
};
assert_eq!(tree, parsed)
}

0 comments on commit b3b271f

Please sign in to comment.