ET9_WebGL_Base/Cpp/Platform/Thread/CountBarrierTest.cc

#include <boost/bind.hpp>
#include <boost/ref.hpp>
#include <boost/detail/atomic_count.hpp>
#include <boost/date_time.hpp>
#include <boost/threadpool.hpp>
#include <gtest/gtest.h>
#include <glog/logging.h>
#include <gflags/gflags.h>
#include "Thread/CountBarrier.h"

namespace Egametang {

class CountBarrierTest: public testing::Test
{
protected:
	boost::detail::atomic_count count;
public:
	CountBarrierTest(): count(0)
	{
	}
	virtual ~CountBarrierTest()
	{
	}

	void Wait(CountBarrier& barrier)
	{
		barrier.Wait();
	}
	void Signal(CountBarrier& barrier)
	{
		boost::xtime xt;
		boost::xtime_get(&xt, boost::TIME_UTC);
		xt.sec += 1;
		boost::thread::sleep(xt);
		++count;
		barrier.Signal();
	}
};

TEST_F(CountBarrierTest, Count)
{
	CountBarrier barrier(10);
	ASSERT_EQ(10, barrier.Count());
}

TEST_F(CountBarrierTest, WaitAndSignal)
{
	CountBarrier barrier(10);
	boost::threadpool::fifo_pool pool(10);
	for (int i = 0; i < 10; ++i)
	{
		pool.schedule(
				boost::bind(&CountBarrierTest::Signal,
						this, boost::ref(barrier)));
	}
	ASSERT_EQ(0, this->count);
	barrier.Wait();
	ASSERT_EQ(10, this->count);
	pool.wait();
}

} // namespace Egametang

int main(int argc, char* argv[])
{
	testing::InitGoogleTest(&argc, argv);
	google::InitGoogleLogging(argv[0]);
	google::ParseCommandLineFlags(&argc, &argv, true);
	return RUN_ALL_TESTS();
}