#include "ast/ast.hpp"
#include "parser/parser.hpp"
#include "utils.hpp"

#include <doctest.h>

TEST_SUITE("Parser: return") {
    TEST_CASE_FIXTURE(ParserFixture, "Parse return statement") {
        setup("\
return 5;\
return 10;\
return 103213;\
");

        ast::program* program = p->parse_program();
        check_parser_errors(p->errors);

        REQUIRE_MESSAGE(
            program != nullptr,
            "parse_program() returned a null pointer"
        );
        REQUIRE(program->statements.size() == 3);

        for (const auto stmt : program->statements) {
            REQUIRE(stmt->token_literal() == "return");
            ast::return_stmt* let_stmt;
            REQUIRE_NOTHROW(let_stmt = dynamic_cast<ast::return_stmt*>(stmt));
            REQUIRE_MESSAGE(
                let_stmt != nullptr,
                "Couldn't cast statement to a return statement"
            );
        }

        delete program;
    }
}