From 3136db04e67b148f6fa04223512ecc24f739bb6b Mon Sep 17 00:00:00 2001
From: Avi Kivity <avi@cloudius-systems.com>
Date: Thu, 2 Jan 2014 12:05:45 +0200
Subject: [PATCH] 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 | 1 +
tests/tst-wait-for.cc | 91 +++++++++++++++++++++++++++++++++++++++++++
2 files changed, 92 insertions(+)
create mode 100644 tests/tst-wait-for.cc
diff --git a/build.mk b/build.mk
index b8ccf656b..db4a79fcf 100644
--- a/build.mk
+++ b/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
diff --git a/tests/tst-wait-for.cc b/tests/tst-wait-for.cc
new file mode 100644
index 000000000..0920f6b0f
--- /dev/null
+++ b/tests/tst-wait-for.cc
@@ -0,0 +1,91 @@
+/*
+ * 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();
+ }
+}
--
GitLab