leo/grammar/tests/function.rs

// 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, [self_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, [self_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, [self_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, [])])])
                    ])])
                ]),
            ])
        ]
    }
}