uclisp/test/test_e2e.c

#include <stdlib.h>
#include <unity.h>
#include <string.h>

#include "testing_helpers.h"

#include "uclisp.h"
#include "common.h"
#include "scope.h"
#include "builtins.h"
#include "special.h"

static struct ucl_object *input;
static struct ucl_object *response;
static struct ucl_scope *scope;

void setUp(void) {
    input = NULL;
    response = NULL;

    scope = ucl_scope_create();
    ucl_scope_put(scope, "let", ucl_special_create(ucl_special_let));
    ucl_scope_put(scope, "defun", ucl_special_create(ucl_special_defun));
    ucl_scope_put(scope, "+", ucl_builtin_create(ucl_builtin_add));
    ucl_scope_put(scope, "error", ucl_builtin_create(ucl_builtin_error));
    ucl_scope_put(scope, "list", ucl_builtin_create(ucl_builtin_list));
    ucl_scope_put(scope, "setq", ucl_special_create(ucl_special_setq));
    ucl_scope_put(scope, "car", ucl_builtin_create(ucl_builtin_car));
    ucl_scope_put(scope, "cdr", ucl_builtin_create(ucl_builtin_cdr));
    ucl_scope_put(scope, "nth", ucl_builtin_create(ucl_builtin_nth));
    ucl_scope_put(scope, "mapcar", ucl_builtin_create(ucl_builtin_mapcar));
    ucl_scope_put(scope, "filter", ucl_builtin_create(ucl_builtin_filter));
    ucl_scope_put(scope, "reduce", ucl_builtin_create(ucl_builtin_reduce));
    ucl_scope_put(scope, "lambda", ucl_special_create(ucl_special_lambda));
    ucl_scope_put(scope, "quote", ucl_special_create(ucl_special_quote));
    ucl_scope_put(scope, "%", ucl_builtin_create(ucl_builtin_mod));
}

void tearDown(void) {
    ucl_scope_delete(scope);
    ucl_gc();

    input = NULL;
    response = NULL;
    scope = NULL;
}

static struct ucl_object *eval (const char *code) {
    struct ucl_object *sexp = ucl_parse(code);
    return ucl_evaluate(scope, ucl_car(sexp));
}

void test_simple_add(void) {
    response = eval("(+ 2 3)");
    TEST_ASSERT_OBJ_INT_V(response, 5);
}

void test_simple_let(void) {
    response = eval("(let ((x 2)) (+ x 3))");
    TEST_ASSERT_OBJ_INT_V(response, 5);
}

void test_let_assignment_error(void) {
    response = eval("(let ((x (error \"foo\"))) (+ x 3))");
    TEST_ASSERT_OBJ_ERROR(response);
}

void test_nested_let(void) {
    response = eval("(let ((x 2)) (let ((y 5)) (+ x y)))");
    TEST_ASSERT_OBJ_INT_V(response, 7);
}

void test_nested_let_scope(void) {
    response = eval("(let ((x 2)) (let ((y 5)) (+ x y)) y)");
    TEST_ASSERT_OBJ_ERROR(response);
}

void test_multi_let(void) {
    response = eval("(let ((x 2)(y 5)) (+ x y))");
    TEST_ASSERT_OBJ_INT_V(response, 7);
}

void test_let_return_sym(void) {
    response = eval("(let ((x 2)) x)");
    TEST_ASSERT_OBJ_INT_V(response, 2);
}

void test_eval_int(void) {
    response = eval("2");
    TEST_ASSERT_OBJ_INT_V(response, 2);
}

void test_eval_string(void) {
    response = eval("\"foo\"");
    TEST_ASSERT_OBJ_STRING_V(response, "foo");
}

void test_eval_sym_defined(void) {
    ucl_scope_put(scope, "foo", ucl_int_create(2));
    response = eval("foo");
    TEST_ASSERT_OBJ_INT_V(response, 2);
}

void test_eval_sym_undefined(void) {
    response = eval("foo");
    TEST_ASSERT_OBJ_ERROR(response);
}

void test_eval_nested_error(void) {
    response = eval("(+ (error \"foo\") 3)");
    TEST_ASSERT_OBJ_ERROR(response);
}

void test_eval_list(void) {
    response = eval("(list 1 2 3)");
    TEST_ASSERT_OBJ_LIST(response);
    TEST_ASSERT_EQUAL(ucl_list_length(response)->integer, 3);

    int i = 1;
    FOREACH_LIST(response, iter, item) {
        TEST_ASSERT_EQUAL(item->integer, i);
        i++;
    }
}

void test_eval_defun(void) {
    response = eval("(defun foo (a b) (+ a b))");

    TEST_ASSERT_OBJ_SYMBOL_V(response, "foo");
}

void test_call_function(void) {
    response = eval("(defun foo (a b) (+ a b))");

    TEST_ASSERT_OBJ_SYMBOL(response);
    TEST_ASSERT_EQUAL_STRING(response->symbol, "foo");

    response = eval("(foo 2 3)");

    TEST_ASSERT_OBJ_INT(response);
    TEST_ASSERT_EQUAL(response->integer, 5);
}

void test_setq(void) {
    response = eval("(setq bar 123)");

    TEST_ASSERT_OBJ_INT_V(response, 123);

    response = eval("bar");

    TEST_ASSERT_OBJ_INT_V(response, 123);
}

void test_setq_from_function(void) {
    response = eval("(setq bar 1)");

    TEST_ASSERT_OBJ_INT_V(response, 1);

    response = eval("(defun foo (a) (setq bar a))");

    TEST_ASSERT_OBJ_SYMBOL_V(response, "foo");

    response = eval("(foo 2)");

    TEST_ASSERT_OBJ_INT_V(response, 2);

    response = eval("bar");

    TEST_ASSERT_OBJ_INT_V(response, 1);
}

void test_lambda(void) {
    response = eval("((lambda (x) (+ x 2)) 2)");

    TEST_ASSERT_OBJ_INT_V(response, 4);
}

void test_mapcar_lambda(void) {
    response = eval("(car (mapcar (quote (lambda (x) (+ x 2))) (list 5)))");

    TEST_ASSERT_OBJ_INT_V(response, 7);
}

void test_mapcar_function(void) {
    response = eval("(defun foo (a) (+ a 2))");

    TEST_ASSERT_OBJ_SYMBOL(response);
    TEST_ASSERT_EQUAL_STRING(response->symbol, "foo");

    response = eval("(car (mapcar (quote foo) (list 5)))");

    TEST_ASSERT_OBJ_INT_V(response, 7);
}

void test_car(void) {
    response = eval("(car (list 2 3 4))");

    TEST_ASSERT_OBJ_INT_V(response, 2);
}

void test_cdr(void) {
    response = eval("(car (cdr (list 2 3 4)))");

    TEST_ASSERT_OBJ_INT_V(response, 3);
}

void test_nth_0(void) {
    response = eval("(nth 0 (list 2 3 4))");

    TEST_ASSERT_OBJ_INT_V(response, 2);
}

void test_nth_1(void) {
    response = eval("(nth 1 (list 2 3 4))");

    TEST_ASSERT_OBJ_INT_V(response, 3);
}

void test_nth_oob(void) {
    response = eval("(nth 3 (list 2 3 4))");

    TEST_ASSERT_OBJ_ERROR(response);
}

void test_eval_defun_gc(void) {
    response = eval("(defun foo (a b) (+ a b))");

    TEST_ASSERT_OBJ_SYMBOL_V(response, "foo");

    ucl_gc();
    response = eval("(foo 10 15)");

    TEST_ASSERT_OBJ_INT_V(response, 25);
}

void test_complex(void) {
    response = eval("(defun sum (elems) (reduce (quote +) elems 0))");

    TEST_ASSERT_OBJ_SYMBOL_V(response, "sum");

    response = eval("(sum (filter (quote (lambda (x) (% x 2))) (list 1 2 3 4 5)))");

    TEST_ASSERT_OBJ_INT_V(response, 9);
}

int main(void) {
    UNITY_BEGIN();

    RUN_TEST(test_simple_add);
    RUN_TEST(test_simple_let);
    RUN_TEST(test_let_assignment_error);
    RUN_TEST(test_nested_let);
    RUN_TEST(test_nested_let_scope);
    RUN_TEST(test_multi_let);
    RUN_TEST(test_let_return_sym);
    RUN_TEST(test_eval_int);
    RUN_TEST(test_eval_string);
    RUN_TEST(test_eval_sym_defined);
    RUN_TEST(test_eval_sym_undefined);
    RUN_TEST(test_eval_nested_error);
    RUN_TEST(test_eval_list);
    RUN_TEST(test_eval_defun);
    RUN_TEST(test_call_function);
    RUN_TEST(test_setq);
    RUN_TEST(test_setq_from_function);
    RUN_TEST(test_lambda);
    RUN_TEST(test_mapcar_lambda);
    RUN_TEST(test_car);
    RUN_TEST(test_cdr);
    RUN_TEST(test_nth_0);
    RUN_TEST(test_nth_1);
    RUN_TEST(test_nth_oob);
    RUN_TEST(test_eval_defun_gc);
    RUN_TEST(test_complex);

    return UNITY_END();
}