Add tests for UCTimers

test/test_runner.c

@@ -18,6 +18,7 @@ extern Suite *buffer_suite(void);
 extern Suite *clock_suite(void);
 extern Suite *seq_suite(void);
 extern Suite *sat_math_suite(void);
+extern Suite *timers_suite(void);
 
 static suite_func suites[] = {
     bitvec_suite,
@@ -31,7 +32,8 @@ static suite_func suites[] = {
     buffer_suite,
     clock_suite,
     seq_suite,
-    sat_math_suite
+    sat_math_suite,
+    timers_suite
 };
 
 int

test/test_timers.c

@@ -0,0 +1,176 @@
+#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;
+}
+