102 lines
4.2 KiB
C++
102 lines
4.2 KiB
C++
#include "ast/expressions/identifier.hpp"
|
|
#include "ast/expressions/infix.hpp"
|
|
#include "ast/expressions/integer.hpp"
|
|
#include "ast/expressions/prefix.hpp"
|
|
#include "utils.hpp"
|
|
|
|
#include <doctest.h>
|
|
#include <sstream>
|
|
|
|
void test_integer_literal(ast::expression* expr, int value) {
|
|
ast::integer_literal* int_lit = cast<ast::integer_literal>(expr);
|
|
|
|
REQUIRE(int_lit->value == value);
|
|
std::ostringstream oss;
|
|
oss << value;
|
|
REQUIRE(int_lit->token_literal() == oss.str());
|
|
}
|
|
|
|
TEST_SUITE("Parser: expression") {
|
|
TEST_CASE_FIXTURE(
|
|
ParserFixture,
|
|
"Simple expression statement with identifier"
|
|
) {
|
|
setup("foobar;");
|
|
REQUIRE(program->statements.size() == 1);
|
|
ast::expression_stmt* expression_stmt =
|
|
cast<ast::expression_stmt>(program->statements[0]);
|
|
|
|
ast::identifier* ident =
|
|
cast<ast::identifier>(expression_stmt->expression);
|
|
|
|
REQUIRE(ident->value == "foobar");
|
|
REQUIRE(ident->token_literal() == "foobar");
|
|
};
|
|
|
|
TEST_CASE_FIXTURE(
|
|
ParserFixture,
|
|
"Simple expression statement with integer"
|
|
) {
|
|
setup("5;");
|
|
REQUIRE(program->statements.size() == 1);
|
|
|
|
ast::expression_stmt* expression_stmt =
|
|
cast<ast::expression_stmt>(program->statements[0]);
|
|
|
|
test_integer_literal(expression_stmt->expression, 5);
|
|
};
|
|
|
|
TEST_CASE_FIXTURE(
|
|
ParserFixture,
|
|
"Simple expression statement with prefix before integer"
|
|
) {
|
|
#define CASE(name, input, _op, _right) \
|
|
SUBCASE(name) { \
|
|
setup(input); \
|
|
\
|
|
REQUIRE(program->statements.size() == 1); \
|
|
ast::expression_stmt* expression_stmt = \
|
|
cast<ast::expression_stmt>(program->statements[0]); \
|
|
\
|
|
ast::prefix_expr* prefix_expr = \
|
|
cast<ast::prefix_expr>(expression_stmt->expression); \
|
|
\
|
|
REQUIRE(prefix_expr->op == _op); \
|
|
test_integer_literal(prefix_expr->right, _right); \
|
|
}
|
|
|
|
CASE("Prefix: '!'", "!5;", "!", 5);
|
|
CASE("Prefix: '-'", "-15;", "-", 15);
|
|
#undef CASE
|
|
}
|
|
|
|
TEST_CASE_FIXTURE(
|
|
ParserFixture,
|
|
"Simple expression statement with infix between integers"
|
|
) {
|
|
#define CASE(name, input, _left, _op, _right) \
|
|
SUBCASE(name) { \
|
|
setup(input); \
|
|
REQUIRE(program->statements.size() == 1); \
|
|
ast::expression_stmt* expression_stmt = \
|
|
cast<ast::expression_stmt>(program->statements[0]); \
|
|
\
|
|
ast::infix_expr* infix_expr = \
|
|
cast<ast::infix_expr>(expression_stmt->expression); \
|
|
\
|
|
test_integer_literal(infix_expr->left, _left); \
|
|
REQUIRE(infix_expr->op == _op); \
|
|
test_integer_literal(infix_expr->right, _right); \
|
|
}
|
|
CASE("Infix: '+'", "5 + 5;", 5, "+", 5);
|
|
CASE("Infix: '-'", "5- 5;", 5, "-", 5);
|
|
CASE("Infix: '*'", "15 *5;", 15, "*", 5);
|
|
CASE("Infix: '/'", "15 / 5;", 15, "/", 5);
|
|
CASE("Infix: '<'", "15 < 15;", 15, "<", 15);
|
|
CASE("Infix: '>'", "25 > 15;", 25, ">", 15);
|
|
CASE("Infix: '=='", "5 == 5;", 5, "==", 5);
|
|
CASE("Infix: '!='", "15 != 5;", 15, "!=", 5);
|
|
#undef CASE
|
|
}
|
|
}
|