37 lines
936 B
C++
37 lines
936 B
C++
#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;
|
|
}
|
|
}
|