#include <array>
#include <vector>
#include "gtest/gtest.h"
#include "queue.h"
using namespace std;
TEST(CUnitQueue, Increase)
{
const int buffer_size_pkts = 4;
CUnitQueue unit_queue;
unit_queue.init(buffer_size_pkts, 1500, AF_INET);
vector<CUnit*> taken_units;
for (int i = 0; i < 5 * buffer_size_pkts; ++i)
{
CUnit* unit = unit_queue.getNextAvailUnit();
ASSERT_NE(unit, nullptr);
unit_queue.makeUnitGood(unit);
taken_units.push_back(unit);
}
}
TEST(CUnitQueue, IncreaseAndFree)
{
const int buffer_size_pkts = 4;
CUnitQueue unit_queue;
unit_queue.init(buffer_size_pkts, 1500, AF_INET);
CUnit* taken_unit = nullptr;
for (int i = 0; i < 5 * buffer_size_pkts; ++i)
{
CUnit* unit = unit_queue.getNextAvailUnit();
ASSERT_NE(unit, nullptr);
unit_queue.makeUnitGood(unit);
if (taken_unit)
unit_queue.makeUnitFree(taken_unit);
taken_unit = unit;
}
}
TEST(CUnitQueue, IncreaseAndFreeGrouped)
{
const int buffer_size_pkts = 4;
CUnitQueue unit_queue;
unit_queue.init(buffer_size_pkts, 1500, AF_INET);
vector<CUnit*> taken_units;
for (int i = 0; i < 5 * buffer_size_pkts; ++i)
{
CUnit* unit = unit_queue.getNextAvailUnit();
ASSERT_NE(unit, nullptr);
unit_queue.makeUnitGood(unit);
if (taken_units.size() >= buffer_size_pkts)
{
for_each(taken_units.begin(), taken_units.end(),
[&unit_queue](CUnit* u) { unit_queue.makeUnitFree(u); });
taken_units.clear();
}
taken_units.push_back(unit);
EXPECT_LE(unit_queue.capacity(), 2 * buffer_size_pkts)
<< "Buffer capacity should not exceed two queues of 4 units";
}
}