lake/test/laketest.c

/**
 * laketest.c
 * Lake Scheme
 *
 * Copyright 2011 Sami Samhuri
 * MIT License
 *
 * Based on MinUnit: http://www.jera.com/techinfo/jtns/jtn002.html
 *
 */

#include "lake.h"
#include "eval.h"
#include "laketest.h"
#include "parse.h"
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>

static int captured = 0;

static void capture_output(void)
{
    if (captured) return;
    captured = 1;

    int fd = open("./tmp", O_WRONLY);
    close(2);
    int newfd = dup(fd);
    close(fd);

    fd = open("./tmp", O_WRONLY);
    close(1);
    newfd = dup(fd);
    close(fd);
}

void restore_output(void)
{
    if (!captured) return;
    captured = 0;

    freopen("/dev/tty", "a", stdout);
    freopen("/dev/tty", "a", stderr);
    unlink("./tmp");
}

int lt_run_tests(char *title, test_fn *tests)
{
    int pass = 0;
    int n_tests = 0;
    char *message = 0;
    test_fn test;
    printf("-- %s --\n", title);
    capture_output();
    while ((test = *(tests++)))
    {
        if ((message = test())) break;
        n_tests++;
    }
    restore_output();
    pass = message == 0;
    if (pass)
    {
        fprintf(stderr, "PASS: %d test%s\n", n_tests, n_tests == 1 ? "" : "s");
    }
    else
    {
        fprintf(stderr, "FAIL: %s\n", message);
    }
    return pass;
}

LakeVal *lt_eval(LakeCtx *lake, char *s)
{
    LakeVal *v = parse_expr(lake, s, strlen(s));
    return eval(lake, lake->toplevel, v);
}