Files
interpreter-cpp/test/parser/return.cpp
2025-07-07 17:44:11 +02:00

40 lines
909 B
C++

#include "ast/ast.hpp"
#include "lexer/lexer.hpp"
#include "parser/parser.hpp"
#include "utils.hpp"
#include <doctest.h>
#include <sstream>
TEST_CASE("Parse return statement") {
std::stringstream input("\
return 5;\
return 10;\
return 103213;\
");
lexer::lexer l{input};
parser::parser p{l};
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::let_stmt* let_stmt;
REQUIRE_NOTHROW(let_stmt = dynamic_cast<ast::let_stmt*>(stmt));
REQUIRE_MESSAGE(
let_stmt != nullptr,
"Couldn't cast statement to a return statement"
);
}
delete program;
}