diff options
Diffstat (limited to 'libadhocutil/unittests/testResourcePool.cpp')
| -rw-r--r-- | libadhocutil/unittests/testResourcePool.cpp | 242 |
1 files changed, 242 insertions, 0 deletions
diff --git a/libadhocutil/unittests/testResourcePool.cpp b/libadhocutil/unittests/testResourcePool.cpp new file mode 100644 index 0000000..8b4161e --- /dev/null +++ b/libadhocutil/unittests/testResourcePool.cpp @@ -0,0 +1,242 @@ +#define BOOST_TEST_MODULE ResourcePool +#include <boost/test/unit_test.hpp> + +#include <resourcePool.impl.h> + +class MockResource { + public: + MockResource() { count += 1; } + ~MockResource() { count -= 1; } + + MockResource(const MockResource &) = delete; + void operator=(const MockResource &) = delete; + + static std::atomic<unsigned int> count; +}; + +std::atomic<unsigned int> MockResource::count; + +class TRP : public AdHoc::ResourcePool<MockResource> { + public: + TRP() : AdHoc::ResourcePool<MockResource>(10, 10) { } + protected: + MockResource * createResource() const override + { + return new MockResource(); + } +}; + +class TRPSmall : public AdHoc::ResourcePool<MockResource> { + public: + TRPSmall() : AdHoc::ResourcePool<MockResource>(3, 1) { } + protected: + MockResource * createResource() const override + { + return new MockResource(); + } +}; + +BOOST_AUTO_TEST_CASE ( get ) +{ + { + TRP pool; + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + + { + auto r1 = pool.get(); + BOOST_REQUIRE_EQUAL(1, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(1, r1.handleCount()); + BOOST_REQUIRE_EQUAL(1, MockResource::count); + BOOST_REQUIRE(r1.get()); + + auto r2(pool.get()); + BOOST_REQUIRE_EQUAL(2, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(1, r2.handleCount()); + BOOST_REQUIRE_EQUAL(2, MockResource::count); + BOOST_REQUIRE(r2.get()); + + auto r1a = r1; + BOOST_REQUIRE_EQUAL(2, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(2, r1.handleCount()); + BOOST_REQUIRE_EQUAL(2, r1a.handleCount()); + BOOST_REQUIRE_EQUAL(2, MockResource::count); + BOOST_REQUIRE(r1.get()); + BOOST_REQUIRE(r1a.get()); + BOOST_REQUIRE_EQUAL(r1.get(), r1a.get()); + } + + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(2, pool.availableCount()); + } + BOOST_REQUIRE_EQUAL(0, MockResource::count); +} + +BOOST_AUTO_TEST_CASE ( getMine ) +{ + TRP pool; + auto r1 = pool.get(); + BOOST_REQUIRE(r1.get()); + auto r2 = pool.getMine(); + BOOST_REQUIRE_EQUAL(r1.get(), r2.get()); + BOOST_REQUIRE_EQUAL(2, r1.handleCount()); + BOOST_REQUIRE_EQUAL(2, r2.handleCount()); +} + +BOOST_AUTO_TEST_CASE( getMineNoCurrent ) +{ + TRP pool; + BOOST_REQUIRE_THROW(pool.getMine(), AdHoc::NoCurrentResourceT<MockResource>); + { + auto r1 = pool.get(); + auto r2 = pool.getMine(); + BOOST_REQUIRE_EQUAL(r1.get(), r2.get()); + } + BOOST_REQUIRE_THROW(pool.getMine(), AdHoc::NoCurrentResource); +} + +BOOST_AUTO_TEST_CASE( discard ) +{ + TRP pool; + try { + auto r1 = pool.get(); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(1, pool.inUseCount()); + throw std::exception(); + } + catch (...) { + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + } +} + +BOOST_AUTO_TEST_CASE( keepSome1 ) +{ + TRPSmall pool; + { + auto r1 = pool.get(); + { + auto r2 = pool.get(); + { + auto r3 = pool.get(); + BOOST_REQUIRE_EQUAL(3, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(3, MockResource::count); + } + BOOST_REQUIRE_EQUAL(2, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); + BOOST_REQUIRE_EQUAL(3, MockResource::count); + } + BOOST_REQUIRE_EQUAL(1, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); + BOOST_REQUIRE_EQUAL(2, MockResource::count); + { + auto r2 = pool.get(); + BOOST_REQUIRE_EQUAL(2, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(2, MockResource::count); + } + BOOST_REQUIRE_EQUAL(1, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); + BOOST_REQUIRE_EQUAL(2, MockResource::count); + } + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); + BOOST_REQUIRE_EQUAL(1, MockResource::count); +} + +BOOST_AUTO_TEST_CASE( keepSome2 ) +{ + TRPSmall pool; + { + auto r1 = pool.get(); + auto r2 = pool.get(); + auto r3 = pool.get(); + BOOST_REQUIRE_EQUAL(3, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(3, MockResource::count); + } + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); + BOOST_REQUIRE_EQUAL(1, MockResource::count); +} + +BOOST_AUTO_TEST_CASE( idle ) +{ + TRP pool; + { + { + auto r1 = pool.get(); + auto r2 = pool.get(); + } + auto r3 = pool.get(); + BOOST_REQUIRE_EQUAL(1, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); + BOOST_REQUIRE_EQUAL(2, MockResource::count); + pool.idle(); + BOOST_REQUIRE_EQUAL(1, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(1, MockResource::count); + } + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); + BOOST_REQUIRE_EQUAL(1, MockResource::count); + pool.idle(); + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(0, pool.availableCount()); + BOOST_REQUIRE_EQUAL(0, MockResource::count); +} + +BOOST_AUTO_TEST_CASE( threading1 ) +{ + TRPSmall pool; + std::list<std::thread *> threads; + for (int x = 0; x < 100; x += 1) { + threads.push_back(new std::thread([&pool](){ + auto r = pool.get(); + usleep(50000); + })); + usleep(5000); + // pool size never exceeds 3 + BOOST_REQUIRE(pool.inUseCount() <= 3); + } + for(std::thread * thread : threads) { + thread->join(); + delete thread; + } + // pool keep returns to 1 + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); +} + +static +void +acquireAndKeepFor1Second(TRPSmall * pool) +{ + auto r = pool->get(); + sleep(1); +} + +BOOST_AUTO_TEST_CASE( threading2 ) +{ + TRPSmall pool; + std::thread t1([&pool]() { acquireAndKeepFor1Second(&pool); }); + std::thread t2([&pool]() { acquireAndKeepFor1Second(&pool); }); + std::thread t3([&pool]() { acquireAndKeepFor1Second(&pool); }); + + BOOST_REQUIRE_THROW(pool.get(100), AdHoc::TimeOutOnResourcePoolT<MockResource>); + BOOST_REQUIRE_EQUAL(3, pool.inUseCount()); + + t1.join(); + { + auto r = pool.get(0); + t2.join(); + t3.join(); + } + + BOOST_REQUIRE_EQUAL(0, pool.inUseCount()); + BOOST_REQUIRE_EQUAL(1, pool.availableCount()); +} + |