| 1 | #include <atomic> |
|---|---|
| 2 | #include <iostream> |
| 3 | #include <Common/ThreadPool.h> |
| 4 | |
| 5 | #include <gtest/gtest.h> |
| 6 | |
| 7 | |
| 8 | TEST(ThreadPool, Loop) |
| 9 | { |
| 10 | std::atomic<int> res{0}; |
| 11 | |
| 12 | for (size_t i = 0; i < 1000; ++i) |
| 13 | { |
| 14 | size_t threads = 16; |
| 15 | ThreadPool pool(threads); |
| 16 | for (size_t j = 0; j < threads; ++j) |
| 17 | pool.scheduleOrThrowOnError([&] { ++res; }); |
| 18 | pool.wait(); |
| 19 | } |
| 20 | |
| 21 | EXPECT_EQ(res, 16000); |
| 22 | } |
| 23 |