81 lines
1.8 KiB
C++
81 lines
1.8 KiB
C++
#include <catch2/catch_test_macros.hpp>
|
|
|
|
#include <matador/sql/connection_pool.hpp>
|
|
|
|
#include "DummyConnection.hpp"
|
|
|
|
using namespace matador::sql;
|
|
|
|
TEST_CASE("Create connection pool", "[connection pool]") {
|
|
using pool_t = connection_pool<matador::test::DummyConnection>;
|
|
|
|
pool_t pool("sqlite://sqlite.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("Acquire connection by id", "[connection pool]") {
|
|
using pool_t = connection_pool<matador::test::DummyConnection>;
|
|
|
|
pool_t pool("sqlite://sqlite.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.acquire(ptr.id().value());
|
|
|
|
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);
|
|
} |