#include <check.h>
#include <ucore/timers.h>
#include <ucore/timers.h>
#include <ucore/clock.h>

struct TimerTest {
    int cnt;
};
#define TEST_TIMER_START 1000000
static void timer_test_cb(struct UCTimers *timers, struct UCTimer *timer)
{
    struct TimerTest *tt = timer->cookie_ptr;
    (void)timers; //get rid of unused param warning
    tt->cnt++;
}

static void timer_test_cb_add(struct UCTimers *timers, struct UCTimer *timer)
{
    struct TimerTest *tt = timer->cookie_ptr;
    tt->cnt++;
    uc_timer_add(timers, timer, 10, 0);
}

static void timer_test_cb_remove(struct UCTimers *timers, struct UCTimer *timer)
{
    struct UCTimer *timer2 = timer->cookie_ptr;
    uc_timer_remove(timers, timer2);
}

static void common_init(struct UCoreSettableClock *clk, 
                        struct TimerTest *tt, 
                        struct UCTimers *timers,
                        struct UCTimer *timer) 
{
    uc_settable_clock_init(clk);
    uc_settable_clock_set(clk, TEST_TIMER_START, 0);
    uc_timers_init_ex(timers, &clk->clock);


    tt->cnt = 0;
    memset(timer, 0, sizeof *timer);
    timer->cookie_ptr = tt;
}

START_TEST (test_timers_simple)
    struct UCoreSettableClock clk;
    struct TimerTest tt;
    struct UCTimers timers;
    struct UCTimer timer;
    int t = TEST_TIMER_START;
    int rc;
    
    common_init(&clk, &tt, &timers, &timer);

    timer.callback = timer_test_cb;

    uc_timer_add(&timers, &timer, 10, 0);

    //5 seconds into the future, no timer should fire
    t += 5;
    uc_settable_clock_set(&clk, t, 0);
    rc = uc_timers_run(&timers);
    fail_if(rc != 0);
    fail_if(tt.cnt != 0);
    fail_if(uc_timer_count(&timers) != 1);

    //10 seconds into the future, our timer should fire
    t += 5;
    uc_settable_clock_set(&clk, t, 0);
    rc = uc_timers_run(&timers);
    fail_if(rc != 1);
    fail_if(tt.cnt != 1);

    //20 seconds into the future, our oneshot timer shouldn't fire
    t += 10;
    uc_settable_clock_set(&clk, t, 0);
    rc = uc_timers_run(&timers);
    fail_if(rc != 0);
    fail_if(tt.cnt != 1);
    fail_if(uc_timer_count(&timers) != 0);
    fail_if(uc_timer_running(&timer) != 0);

END_TEST

START_TEST (test_timers_add_from_callback)
    struct UCoreSettableClock clk;
    struct TimerTest tt;
    struct UCTimers timers;
    struct UCTimer timer;
    int t = TEST_TIMER_START;
    int rc;
    
    common_init(&clk, &tt, &timers, &timer);

    timer.callback = timer_test_cb_add;

    uc_timer_add(&timers, &timer, 10, 0);

    //10 seconds into the future, our timer should fire
    t += 10;
    uc_settable_clock_set(&clk, t, 0);
    rc = uc_timers_run(&timers);
    fail_if(rc != 1);
    fail_if(tt.cnt != 1);


    //20 seconds into the future, our timer should fire again
    t += 10;
    uc_settable_clock_set(&clk, t, 0);
    rc = uc_timers_run(&timers);
    fail_if(rc != 1);
    fail_if(tt.cnt != 2);

    fail_if(uc_timer_running(&timer) == 0);
    fail_if(uc_timer_count(&timers) != 1);

END_TEST

START_TEST (test_timers_remove_from_callback)
    struct UCoreSettableClock clk;
    struct TimerTest tt;
    struct UCTimers timers;
    struct UCTimer timer;
    struct UCTimer timer2;
    int t = TEST_TIMER_START;
    int rc;
    struct timeval tv = {0, 0};
    
    common_init(&clk, &tt, &timers, &timer);

    timer.callback = timer_test_cb_remove;
    timer.cookie_ptr = &timer2;
    memset(&timer2, 0, sizeof timer2);
    timer2.callback = timer_test_cb;

    uc_timer_add(&timers, &timer, 5, 0);
    uc_timer_add(&timers, &timer2, 10, 0);

    fail_if(uc_timer_count(&timers) != 2);
    rc = uc_timers_first(&timers, &tv);
    fail_if(rc != 0);
    fail_if(tv.tv_sec != TEST_TIMER_START + 5);
    fail_if(uc_timer_running(&timer) == 0);
    fail_if(uc_timer_running(&timer2) == 0);

    //5 seconds into the future, our timer should fire
    t += 5;
    uc_settable_clock_set(&clk, t, 0);
    rc = uc_timers_run(&timers);
    fail_if(rc != 1);
    fail_if(tt.cnt != 0);
    //and our timer2 should be removed, and timer has fired so it
    //should be removed too

    fail_if(uc_timer_count(&timers) != 0);
    fail_if(uc_timer_running(&timer) != 0);
    fail_if(uc_timer_running(&timer2) != 0);
    rc = uc_timers_first(&timers, &tv);
    fail_if(rc == 0);

END_TEST


Suite *timers_suite(void)
{
    Suite *s = suite_create("timers");
    TCase *tc = tcase_create("timers tests");
    tcase_add_test(tc, test_timers_simple);
    tcase_add_test(tc, test_timers_add_from_callback);
    tcase_add_test(tc, test_timers_remove_from_callback);

    suite_add_tcase(s, tc);

    return s;
}