static void test_basic_operations()
{
IOQueue<int> queue;
assert(!queue);
queue.push(1);
assert(queue);
assert(queue.front() == 1);
assert(queue.size() == 1);
queue.pop();
assert(!queue);
assert(queue.size() == 0);
queue.push(2);
assert(queue);
assert(queue.front() == 2);
assert(queue.size() == 1);
queue.push(3);
assert(queue);
assert(queue.front() == 2);
assert(queue.size() == 2);
queue.pop();
assert(queue);
assert(queue.front() == 3);
assert(queue.size() == 1);
queue.pop();
assert(!queue);
assert(queue.size() == 0);
}
static void test_thread_safety()
{
std::atomic<int> count(0);
std::atomic<int> push_count(0);
IOQueue<std::function<void()>> queue;
auto consumer = std::thread(
[&queue, &count]
{
while (count.load() < 9000000)
{
if (queue)
{
queue.front()();
queue.pop();
}
}
}
);
auto producer_func =
[&queue, &push_count, &count]
{
while (push_count.fetch_add(1) < 9000000)
{
queue.push([&count]
{
++count;
});
}
};
auto producer_1 = std::thread([&producer_func]{ producer_func(); });
auto producer_2 = std::thread([&producer_func]{ producer_func(); });
auto producer_3 = std::thread([&producer_func]{ producer_func(); });
auto producer_4 = std::thread([&producer_func]{ producer_func(); });
auto producer_5 = std::thread([&producer_func]{ producer_func(); });
auto producer_6 = std::thread([&producer_func]{ producer_func(); });
auto producer_7 = std::thread([&producer_func]{ producer_func(); });
producer_1.join();
producer_2.join();
producer_3.join();
producer_4.join();
producer_5.join();
producer_6.join();
producer_7.join();
consumer.join();
}