tests: add tests for wait_for() with timer objects

Reviewed-by: Nadav Har'El <nyh@cloudius-systems.com>
Signed-off-by: Avi Kivity <avi@cloudius-systems.com>

build.mk

@@ -150,6 +150,7 @@ boost-tests := tests/tst-rename.so \
 	tests/tst-promise.so \
 	tests/tst-dlfcn.so \
 	tests/tst-stat.so
+boost-tests += tests/tst-wait-for.so
 
 java_tests := tests/hello/Hello.class
 

tests/tst-wait-for.cc

+/*
+ * Copyright (C) 2014 Cloudius Systems, Ltd.
+ *
+ * This work is open source software, licensed under the terms of the
+ * BSD license as described in the LICENSE file in the top-level directory.
+ */
+
+
+#define BOOST_TEST_MODULE tst-wait-for
+
+#include <boost/test/unit_test.hpp>
+#include <sched.hh>
+#include <osv/waitqueue.hh>
+#include <drivers/clock.hh>
+#include <cstdlib>
+
+BOOST_AUTO_TEST_CASE(test_wait_for_one_timer)
+{
+    auto now = clock::get()->time();
+    sched::timer tmr(*sched::thread::current());
+    tmr.set(now + 1_s);
+    sched::thread::wait_for(tmr);
+    auto later = clock::get()->time();
+    BOOST_REQUIRE(std::abs(later - (now + 1_s)) < 20_ms);
+    BOOST_REQUIRE(tmr.expired());
+}
+
+BOOST_AUTO_TEST_CASE(test_wait_for_two_timers)
+{
+    auto now = clock::get()->time();
+    sched::timer tmr1(*sched::thread::current());
+    sched::timer tmr2(*sched::thread::current());
+    tmr1.set(now + 2_s);
+    tmr2.set(now + 1_s);
+    sched::thread::wait_for(tmr1, tmr2);
+    BOOST_REQUIRE(!tmr1.expired() && tmr2.expired());
+    tmr2.cancel();
+    sched::thread::wait_for(tmr1, tmr2);
+    BOOST_REQUIRE(tmr1.expired() && !tmr2.expired());
+}
+
+BOOST_AUTO_TEST_CASE(test_waitqueue_1)
+{
+    waitqueue wq;
+    mutex mtx;
+    int counter = 0;
+    WITH_LOCK(mtx) {
+        sched::thread waker([&] {
+            WITH_LOCK(mtx) {
+                ++counter;
+                wq.wake_one(mtx);
+            }
+        });
+        waker.start();
+        wq.wait(mtx);
+        waker.join();
+    }
+    BOOST_REQUIRE(counter == 1);
+}
+
+BOOST_AUTO_TEST_CASE(test_waitqueue_2)
+{
+    waitqueue wq;
+    mutex mtx;
+    int counter = 0;
+    sched::timer tmr(*sched::thread::current());
+    WITH_LOCK(mtx) {
+        tmr.set(clock::get()->time() + 500_ms);
+        sched::thread waker([&] {
+            sched::thread::sleep_until(clock::get()->time() + 1_s);
+            WITH_LOCK(mtx) {
+                ++counter;
+                wq.wake_one(mtx);
+            }
+        });
+        waker.start();
+        // timer wait
+        sched::thread::wait_for(mtx, wq, tmr);
+        BOOST_REQUIRE(tmr.expired());
+        BOOST_REQUIRE(counter == 0);
+        // null wait
+        sched::thread::wait_for(mtx, wq, tmr);
+        BOOST_REQUIRE(tmr.expired());
+        BOOST_REQUIRE(counter == 0);
+        // wait for wq
+        tmr.cancel();
+        sched::thread::wait_for(mtx, wq, tmr);
+        BOOST_REQUIRE(counter == 1);
+        waker.join();
+    }
+}