165 lines
4.5 KiB
C++
165 lines
4.5 KiB
C++
|
|
// Copyright Oliver Kowalke 2013.
|
|
// 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)
|
|
//
|
|
// This test is based on the tests of Boost.Thread
|
|
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <map>
|
|
#include <mutex>
|
|
#include <stdexcept>
|
|
#include <vector>
|
|
|
|
#include <boost/atomic.hpp>
|
|
#include <boost/bind.hpp>
|
|
#include <boost/chrono.hpp>
|
|
#include <boost/cstdint.hpp>
|
|
#include <boost/function.hpp>
|
|
#include <boost/ref.hpp>
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <boost/thread.hpp>
|
|
#include <boost/utility.hpp>
|
|
|
|
#include <boost/fiber/all.hpp>
|
|
|
|
typedef boost::chrono::milliseconds ms;
|
|
|
|
boost::atomic< int > value1;
|
|
|
|
void wait_fn( boost::barrier & b,
|
|
boost::fibers::mutex & mtx,
|
|
boost::fibers::condition_variable & cond,
|
|
bool & flag) {
|
|
b.wait();
|
|
std::unique_lock< boost::fibers::mutex > lk( mtx);
|
|
cond.wait( lk, [&flag](){ return flag; });
|
|
++value1;
|
|
}
|
|
|
|
void notify_one_fn( boost::barrier & b,
|
|
boost::fibers::mutex & mtx,
|
|
boost::fibers::condition_variable & cond,
|
|
bool & flag) {
|
|
b.wait();
|
|
std::unique_lock< boost::fibers::mutex > lk( mtx);
|
|
flag = true;
|
|
lk.unlock();
|
|
cond.notify_one();
|
|
}
|
|
|
|
void notify_all_fn( boost::barrier & b,
|
|
boost::fibers::mutex & mtx,
|
|
boost::fibers::condition_variable & cond,
|
|
bool & flag) {
|
|
b.wait();
|
|
std::unique_lock< boost::fibers::mutex > lk( mtx);
|
|
flag = true;
|
|
lk.unlock();
|
|
cond.notify_all();
|
|
}
|
|
|
|
void fn1( boost::barrier & b,
|
|
boost::fibers::mutex & mtx,
|
|
boost::fibers::condition_variable & cond,
|
|
bool & flag) {
|
|
boost::fibers::fiber(
|
|
boost::fibers::launch::dispatch,
|
|
wait_fn,
|
|
std::ref( b),
|
|
std::ref( mtx),
|
|
std::ref( cond),
|
|
std::ref( flag) ).join();
|
|
}
|
|
|
|
void fn2( boost::barrier & b,
|
|
boost::fibers::mutex & mtx,
|
|
boost::fibers::condition_variable & cond,
|
|
bool & flag) {
|
|
boost::fibers::fiber(
|
|
boost::fibers::launch::dispatch,
|
|
notify_one_fn,
|
|
std::ref( b),
|
|
std::ref( mtx),
|
|
std::ref( cond),
|
|
std::ref( flag) ).join();
|
|
}
|
|
|
|
void fn3( boost::barrier & b,
|
|
boost::fibers::mutex & mtx,
|
|
boost::fibers::condition_variable & cond,
|
|
bool & flag) {
|
|
boost::fibers::fiber(
|
|
boost::fibers::launch::dispatch,
|
|
notify_all_fn,
|
|
std::ref( b),
|
|
std::ref( mtx),
|
|
std::ref( cond),
|
|
std::ref( flag) ).join();
|
|
}
|
|
|
|
void test_one_waiter_notify_one() {
|
|
for ( int i = 0; i < 10; ++i) {
|
|
boost::barrier b( 2);
|
|
|
|
bool flag = false;
|
|
value1 = 0;
|
|
boost::fibers::mutex mtx;
|
|
boost::fibers::condition_variable cond;
|
|
|
|
BOOST_CHECK( 0 == value1);
|
|
|
|
boost::thread t1(std::bind( fn1, std::ref( b), std::ref( mtx), std::ref( cond), std::ref( flag) ) );
|
|
boost::thread t2(std::bind( fn2, std::ref( b), std::ref( mtx), std::ref( cond), std::ref( flag) ) );
|
|
|
|
t1.join();
|
|
t2.join();
|
|
|
|
BOOST_CHECK( 1 == value1);
|
|
}
|
|
}
|
|
|
|
void test_two_waiter_notify_all() {
|
|
for ( int i = 0; i < 10; ++i) {
|
|
boost::barrier b( 3);
|
|
|
|
bool flag = false;
|
|
value1 = 0;
|
|
boost::fibers::mutex mtx;
|
|
boost::fibers::condition_variable cond;
|
|
|
|
BOOST_CHECK( 0 == value1);
|
|
|
|
boost::thread t1(std::bind( fn1, std::ref( b), std::ref( mtx), std::ref( cond), std::ref( flag) ) );
|
|
boost::thread t2(std::bind( fn1, std::ref( b), std::ref( mtx), std::ref( cond), std::ref( flag) ) );
|
|
boost::thread t3(std::bind( fn3, std::ref( b), std::ref( mtx), std::ref( cond), std::ref( flag) ) );
|
|
|
|
t1.join();
|
|
t2.join();
|
|
t3.join();
|
|
|
|
BOOST_CHECK( 2 == value1);
|
|
}
|
|
}
|
|
|
|
void test_dummy() {
|
|
}
|
|
|
|
boost::unit_test::test_suite * init_unit_test_suite( int, char* [])
|
|
{
|
|
boost::unit_test::test_suite * test =
|
|
BOOST_TEST_SUITE("Boost.Fiber: multithreaded condition_variable test suite");
|
|
|
|
#if ! defined(BOOST_FIBERS_NO_ATOMICS)
|
|
test->add( BOOST_TEST_CASE( & test_one_waiter_notify_one) );
|
|
test->add( BOOST_TEST_CASE( & test_two_waiter_notify_all) );
|
|
#else
|
|
test->add( BOOST_TEST_CASE( & test_dummy) );
|
|
#endif
|
|
|
|
return test;
|
|
}
|