Add scheduler/queue_heuristic unit test

1 files changed, 37 insertions, 0 deletions

diff --git a/src/test/test_scheduler.c b/src/test/test_scheduler.c
index e8bceeb2d4..02426f3723 100644
--- a/src/test/test_scheduler.c
+++ b/src/test/test_scheduler.c
@@ -15,6 +15,7 @@
 #define TOR_CHANNEL_INTERNAL_
 #include "or.h"
 #include "compat_libevent.h"
+#define SCHEDULER_PRIVATE_
 #include "scheduler.h"
 
 /* Test suite stuff */
@@ -134,8 +135,44 @@ test_scheduler_initfree(void *arg)
   return;
 }
 
+static void
+test_scheduler_queue_heuristic(void *arg)
+{
+  time_t now = approx_time();
+  uint64_t qh;
+
+  (void)arg;
+
+  queue_heuristic = 0;
+  queue_heuristic_timestamp = 0;
+
+  /* Not yet inited case */
+  scheduler_update_queue_heuristic(now - 180);
+  test_eq(queue_heuristic, 0);
+  test_eq(queue_heuristic_timestamp, now - 180);
+
+  queue_heuristic = 1000000000L;
+  queue_heuristic_timestamp = now - 120;
+
+  scheduler_update_queue_heuristic(now - 119);
+  test_eq(queue_heuristic, 500000000L);
+  test_eq(queue_heuristic_timestamp, now - 119);
+
+  scheduler_update_queue_heuristic(now - 116);
+  test_eq(queue_heuristic, 62500000L);
+  test_eq(queue_heuristic_timestamp, now - 116);
+
+  qh = scheduler_get_queue_heuristic();
+  test_eq(qh, 0);
+
+ done:
+  return;
+}
+
 struct testcase_t scheduler_tests[] = {
   { "initfree", test_scheduler_initfree, TT_FORK, NULL, NULL },
+  { "queue_heuristic", test_scheduler_queue_heuristic,
+    TT_FORK, NULL, NULL },
   END_OF_TESTCASES
 };