ET9_WebGL_Dev/Cpp/Platform/Thread/ThreadPoolTest.cc

#include <boost/function.hpp>
#include <boost/bind.hpp>
#include <boost/threadpool.hpp>
#include <gtest/gtest.h>
#include <glog/logging.h>
#include <gflags/gflags.h>

namespace Egametang {

class ThreadPoolTest: public testing::Test
{
protected:
	boost::threadpool::fifo_pool pool;

public:
	ThreadPoolTest() : pool(10)
	{
	}

	virtual ~ThreadPoolTest()
	{
	}

	void Max(int a, int b, int* z)
	{
		*z = a > b? a : b;
	}
};

TEST_F(ThreadPoolTest, Test1)
{
	std::vector<int> x(100, 8);
	std::vector<int> y(100, 9);
	std::vector<int> z(100, 0);
	for (int i = 0; i < 100; ++i)
	{
		pool.schedule(
				boost::bind(&ThreadPoolTest::Max,
						this, x[i], y[i], &z[i]));
	}
	pool.wait();
	for (int i = 0; i < 100; ++i)
	{
		ASSERT_EQ(9, z[i]);
	}
}

} // 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();
}