158 lines
3.5 KiB
C++
158 lines
3.5 KiB
C++
#include <catch2/catch_test_macros.hpp>
|
|
|
|
#include "matador/sql/connection_pool.hpp"
|
|
|
|
#include "../backend/test_connection.hpp"
|
|
|
|
#include "../utils/auto_reset_event.hpp"
|
|
|
|
#include "ConnectionPoolFixture.hpp"
|
|
|
|
#include <thread>
|
|
|
|
using namespace matador::sql;
|
|
using namespace matador::test::utils;
|
|
using namespace matador::test::orm;
|
|
|
|
TEST_CASE_METHOD(ConnectionPoolFixture, "Create connection pool", "[connection pool]") {
|
|
connection_pool pool("noop://noop.db", 4);
|
|
|
|
REQUIRE(pool.size() == 4);
|
|
REQUIRE(pool.idle() == 4);
|
|
REQUIRE(pool.inuse() == 0);
|
|
|
|
auto ptr = pool.acquire();
|
|
REQUIRE(ptr.valid());
|
|
REQUIRE(ptr.id().value() > 0);
|
|
REQUIRE(ptr->is_open());
|
|
|
|
REQUIRE(pool.idle() == 3);
|
|
REQUIRE(pool.inuse() == 1);
|
|
|
|
pool.release(ptr);
|
|
REQUIRE(!ptr.valid());
|
|
REQUIRE(pool.idle() == 4);
|
|
REQUIRE(pool.inuse() == 0);
|
|
|
|
ptr = pool.acquire(3);
|
|
REQUIRE(ptr.valid());
|
|
REQUIRE(ptr.id() == 3);
|
|
REQUIRE(ptr->is_open());
|
|
{
|
|
auto ptr2 = pool.acquire();
|
|
REQUIRE(ptr2.valid());
|
|
REQUIRE(ptr2->is_open());
|
|
|
|
REQUIRE(pool.idle() == 2);
|
|
REQUIRE(pool.inuse() == 2);
|
|
}
|
|
|
|
REQUIRE(pool.idle() == 3);
|
|
REQUIRE(pool.inuse() == 1);
|
|
|
|
pool.release(ptr);
|
|
REQUIRE(!ptr.valid());
|
|
REQUIRE(pool.idle() == 4);
|
|
REQUIRE(pool.inuse() == 0);
|
|
}
|
|
|
|
TEST_CASE_METHOD(ConnectionPoolFixture, "Acquire connection by id", "[connection pool]") {
|
|
using namespace std::chrono_literals;
|
|
|
|
connection_pool pool("noop://noop.db", 4);
|
|
|
|
REQUIRE(pool.size() == 4);
|
|
REQUIRE(pool.idle() == 4);
|
|
REQUIRE(pool.inuse() == 0);
|
|
|
|
auto ptr = pool.acquire();
|
|
REQUIRE(ptr.valid());
|
|
REQUIRE(ptr.id());
|
|
REQUIRE(ptr.id().value() > 0);
|
|
REQUIRE(ptr->is_open());
|
|
|
|
auto same_ptr = pool.try_acquire(ptr.id().value());
|
|
REQUIRE(!same_ptr.valid());
|
|
|
|
same_ptr = pool.acquire(ptr.id().value(), 3000ms);
|
|
REQUIRE(!same_ptr.valid());
|
|
|
|
const auto connection_id = ptr.id().value();
|
|
|
|
pool.release(ptr);
|
|
REQUIRE(!ptr.valid());
|
|
|
|
same_ptr = pool.acquire(connection_id);
|
|
|
|
REQUIRE(same_ptr.valid());
|
|
REQUIRE(same_ptr.id() == connection_id);
|
|
}
|
|
|
|
TEST_CASE("Try acquire connection", "[connection pool][try acquire]") {
|
|
connection_pool pool("noop://noop.db", 1);
|
|
|
|
REQUIRE(pool.size() == 1);
|
|
REQUIRE(pool.idle() == 1);
|
|
REQUIRE(pool.inuse() == 0);
|
|
|
|
auto ptr = pool.try_acquire();
|
|
REQUIRE(ptr.valid());
|
|
REQUIRE(ptr.id());
|
|
REQUIRE(ptr.id().value() > 0);
|
|
REQUIRE(ptr->is_open());
|
|
REQUIRE(pool.size() == 1);
|
|
REQUIRE(pool.idle() == 0);
|
|
REQUIRE(pool.inuse() == 1);
|
|
|
|
auto ptr2 = pool.try_acquire();
|
|
REQUIRE(!ptr2.valid());
|
|
|
|
pool.release(ptr);
|
|
REQUIRE(!ptr.valid());
|
|
REQUIRE(pool.size() == 1);
|
|
REQUIRE(pool.idle() == 1);
|
|
REQUIRE(pool.inuse() == 0);
|
|
|
|
ptr2 = pool.try_acquire();
|
|
REQUIRE(ptr2.valid());
|
|
REQUIRE(ptr2.id());
|
|
REQUIRE(ptr2.id().value() > 0);
|
|
REQUIRE(ptr2->is_open());
|
|
REQUIRE(pool.size() == 1);
|
|
REQUIRE(pool.idle() == 0);
|
|
REQUIRE(pool.inuse() == 1);
|
|
|
|
pool.release(ptr2);
|
|
|
|
auto_reset_event reset_main_event;
|
|
auto_reset_event reset_thread_event;
|
|
|
|
std::thread t([&reset_main_event, &reset_thread_event, &pool] {
|
|
auto c1 = pool.acquire();
|
|
REQUIRE(c1.valid());
|
|
REQUIRE(c1.id());
|
|
REQUIRE(c1.id().value() > 0);
|
|
|
|
reset_main_event.set();
|
|
|
|
reset_thread_event.wait_one();
|
|
|
|
pool.release(c1);
|
|
REQUIRE(!c1.valid());
|
|
reset_main_event.set();
|
|
});
|
|
reset_main_event.wait_one();
|
|
|
|
ptr2 = pool.try_acquire();
|
|
REQUIRE(!ptr2.valid());
|
|
|
|
reset_thread_event.set();
|
|
|
|
reset_main_event.wait_one();
|
|
ptr2 = pool.try_acquire();
|
|
REQUIRE(ptr2.valid());
|
|
REQUIRE(ptr2.id());
|
|
REQUIRE(ptr2.id().value() > 0);
|
|
|
|
t.join();
|
|
} |