| // |
| // strand.cpp |
| // ~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2011 Christopher M. Kohlhoff (chris at kohlhoff dot com) |
| // |
| // Distributed under the Boost Software License, Version 1.0. (See accompanying |
| // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) |
| // |
| |
| // Disable autolinking for unit tests. |
| #if !defined(BOOST_ALL_NO_LIB) |
| #define BOOST_ALL_NO_LIB 1 |
| #endif // !defined(BOOST_ALL_NO_LIB) |
| |
| // Test that header file is self-contained. |
| #include <boost/asio/strand.hpp> |
| |
| #include <sstream> |
| #include <boost/thread/thread.hpp> |
| #include <boost/bind.hpp> |
| #include <boost/asio/deadline_timer.hpp> |
| #include <boost/asio/io_service.hpp> |
| #include "unit_test.hpp" |
| |
| using namespace boost::asio; |
| |
| void increment(int* count) |
| { |
| ++(*count); |
| } |
| |
| void increment_without_lock(strand* s, int* count) |
| { |
| int original_count = *count; |
| |
| s->dispatch(boost::bind(increment, count)); |
| |
| // No other functions are currently executing through the locking dispatcher, |
| // so the previous call to dispatch should have successfully nested. |
| BOOST_CHECK(*count == original_count + 1); |
| } |
| |
| void increment_with_lock(strand* s, int* count) |
| { |
| int original_count = *count; |
| |
| s->dispatch(boost::bind(increment, count)); |
| |
| // The current function already holds the strand's lock, so the |
| // previous call to dispatch should have successfully nested. |
| BOOST_CHECK(*count == original_count + 1); |
| } |
| |
| void sleep_increment(io_service* ios, int* count) |
| { |
| deadline_timer t(*ios, boost::posix_time::seconds(2)); |
| t.wait(); |
| |
| ++(*count); |
| } |
| |
| void start_sleep_increments(io_service* ios, strand* s, int* count) |
| { |
| // Give all threads a chance to start. |
| deadline_timer t(*ios, boost::posix_time::seconds(2)); |
| t.wait(); |
| |
| // Start three increments. |
| s->post(boost::bind(sleep_increment, ios, count)); |
| s->post(boost::bind(sleep_increment, ios, count)); |
| s->post(boost::bind(sleep_increment, ios, count)); |
| } |
| |
| void throw_exception() |
| { |
| throw 1; |
| } |
| |
| void io_service_run(io_service* ios) |
| { |
| ios->run(); |
| } |
| |
| void strand_test() |
| { |
| io_service ios; |
| strand s(ios); |
| int count = 0; |
| |
| ios.post(boost::bind(increment_without_lock, &s, &count)); |
| |
| // No handlers can be called until run() is called. |
| BOOST_CHECK(count == 0); |
| |
| ios.run(); |
| |
| // The run() call will not return until all work has finished. |
| BOOST_CHECK(count == 1); |
| |
| count = 0; |
| ios.reset(); |
| s.post(boost::bind(increment_with_lock, &s, &count)); |
| |
| // No handlers can be called until run() is called. |
| BOOST_CHECK(count == 0); |
| |
| ios.run(); |
| |
| // The run() call will not return until all work has finished. |
| BOOST_CHECK(count == 1); |
| |
| count = 0; |
| ios.reset(); |
| ios.post(boost::bind(start_sleep_increments, &ios, &s, &count)); |
| boost::thread thread1(boost::bind(io_service_run, &ios)); |
| boost::thread thread2(boost::bind(io_service_run, &ios)); |
| |
| // Check all events run one after another even though there are two threads. |
| deadline_timer timer1(ios, boost::posix_time::seconds(3)); |
| timer1.wait(); |
| BOOST_CHECK(count == 0); |
| timer1.expires_at(timer1.expires_at() + boost::posix_time::seconds(2)); |
| timer1.wait(); |
| BOOST_CHECK(count == 1); |
| timer1.expires_at(timer1.expires_at() + boost::posix_time::seconds(2)); |
| timer1.wait(); |
| BOOST_CHECK(count == 2); |
| |
| thread1.join(); |
| thread2.join(); |
| |
| // The run() calls will not return until all work has finished. |
| BOOST_CHECK(count == 3); |
| |
| count = 0; |
| int exception_count = 0; |
| ios.reset(); |
| s.post(throw_exception); |
| s.post(boost::bind(increment, &count)); |
| s.post(boost::bind(increment, &count)); |
| s.post(throw_exception); |
| s.post(boost::bind(increment, &count)); |
| |
| // No handlers can be called until run() is called. |
| BOOST_CHECK(count == 0); |
| BOOST_CHECK(exception_count == 0); |
| |
| for (;;) |
| { |
| try |
| { |
| ios.run(); |
| break; |
| } |
| catch (int) |
| { |
| ++exception_count; |
| } |
| } |
| |
| // The run() calls will not return until all work has finished. |
| BOOST_CHECK(count == 3); |
| BOOST_CHECK(exception_count == 2); |
| |
| count = 0; |
| ios.reset(); |
| |
| // Check for clean shutdown when handlers posted through an orphaned strand |
| // are abandoned. |
| { |
| strand s2(ios); |
| s2.post(boost::bind(increment, &count)); |
| s2.post(boost::bind(increment, &count)); |
| s2.post(boost::bind(increment, &count)); |
| } |
| |
| // No handlers can be called until run() is called. |
| BOOST_CHECK(count == 0); |
| } |
| |
| test_suite* init_unit_test_suite(int, char*[]) |
| { |
| test_suite* test = BOOST_TEST_SUITE("strand"); |
| test->add(BOOST_TEST_CASE(&strand_test)); |
| return test; |
| } |