mirror of
https://github.com/ProvableHQ/leo.git
synced 2024-11-28 01:01:53 +03:00
136 lines
4.1 KiB
Rust
136 lines
4.1 KiB
Rust
// Copyright (C) 2019-2021 Aleo Systems Inc.
|
|
// This file is part of the Leo library.
|
|
|
|
// The Leo library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// The Leo library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with the Leo library. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
use leo_grammar::ast::{LanguageParser, Rule};
|
|
|
|
use pest::*;
|
|
|
|
#[test]
|
|
fn call_wo_args() {
|
|
parses_to! {
|
|
parser: LanguageParser,
|
|
input: "x()",
|
|
rule: Rule::expression_postfix,
|
|
tokens: [
|
|
expression_postfix(0, 3, [
|
|
keyword_or_identifier(0, 1, [identifier(0, 1, [])]),
|
|
access(1, 3, [access_call(1, 3, [])])
|
|
])
|
|
]
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn call_with_arg() {
|
|
parses_to! {
|
|
parser: LanguageParser,
|
|
input: "x(true)",
|
|
rule: Rule::expression_postfix,
|
|
tokens: [
|
|
expression_postfix(0, 7, [
|
|
keyword_or_identifier(0, 1, [identifier(0, 1, [])]),
|
|
access(1, 7, [access_call(1, 7, [
|
|
expression(2, 6, [expression_term(2, 6, [value(2, 6, [value_boolean(2, 6, [])])])])
|
|
])])
|
|
])
|
|
]
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn call_with_2_args() {
|
|
parses_to! {
|
|
parser: LanguageParser,
|
|
input: "x(true, false)",
|
|
rule: Rule::expression_postfix,
|
|
tokens: [
|
|
expression_postfix(0, 14, [
|
|
keyword_or_identifier(0, 1, [identifier(0, 1, [])]),
|
|
access(1, 14, [access_call(1, 14, [
|
|
expression(2, 6, [expression_term(2, 6, [value(2, 6, [value_boolean(2, 6, [])])])]),
|
|
expression(8, 13, [expression_term(8, 13, [value(8, 13, [value_boolean(8, 13, [])])])])
|
|
])])
|
|
])
|
|
]
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn empty_def() {
|
|
parses_to! {
|
|
parser: LanguageParser,
|
|
input: "function x() {}",
|
|
rule: Rule::function,
|
|
tokens: [
|
|
function(0, 15, [identifier(9, 10, []), block(13, 15, [])])
|
|
]
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn returning_unit_type() {
|
|
parses_to! {
|
|
parser: LanguageParser,
|
|
input: "function x() -> () {}",
|
|
rule: Rule::function,
|
|
tokens: [
|
|
function(0, 21, [identifier(9, 10, []), type_(16, 18, [type_tuple(16, 18, [])]), block(19, 21, [])])
|
|
]
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn returning_unit_value() {
|
|
parses_to! {
|
|
parser: LanguageParser,
|
|
input: "function x() { return () }",
|
|
rule: Rule::function,
|
|
tokens: [
|
|
function(0, 26, [identifier(9, 10, []), block(13, 26, [
|
|
statement(15, 25, [
|
|
statement_return(15, 25, [expression(22, 25, [expression_term(22, 24, [expression_tuple(22, 24, [])])])])
|
|
])
|
|
])])
|
|
]
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn id_def() {
|
|
parses_to! {
|
|
parser: LanguageParser,
|
|
input: "function id(x: u8) -> u8 { return x }",
|
|
rule: Rule::function,
|
|
tokens: [
|
|
function(0, 37, [
|
|
identifier(9, 11, []),
|
|
input(12, 17, [
|
|
function_input(12, 17, [
|
|
identifier(12, 13, []),
|
|
type_(15, 17, [type_data(15, 17, [type_integer(15, 17, [type_integer_unsigned(15, 17, [type_u8(15, 17, [])])])])])
|
|
])
|
|
]),
|
|
type_(22, 24, [type_data(22, 24, [type_integer(22, 24, [type_integer_unsigned(22, 24, [type_u8(22, 24, [])])])])]),
|
|
block(25, 37, [
|
|
statement(27, 36, [statement_return(27, 36, [
|
|
expression(34, 36, [expression_term(34, 35, [identifier(34, 35, [])])])
|
|
])])
|
|
]),
|
|
])
|
|
]
|
|
}
|
|
}
|