query/test/orm/sql/ConnectionPoolTest.cpp

#include <catch2/catch_test_macros.hpp>

#include "matador/sql/backend_provider.hpp"
#include "matador/sql/connection.hpp"
#include "matador/sql/connection_pool.hpp"

#include "../backend/test_connection.hpp"
#include "../backend/test_backend_service.hpp"

#include "../utils/auto_reset_event.hpp"

using namespace matador::sql;
using namespace matador::test::utils;
using namespace matador::test::orm;

namespace matador::test::orm {

class ConnectionPoolFixture {
public:
    ConnectionPoolFixture() {
        backend_provider::instance().register_backend("noop", std::make_unique<test_backend_service>());

        db = std::make_unique<sql::connection>("noop://noop.db");
    }
    ~ConnectionPoolFixture() = default;

protected:
    std::unique_ptr<connection> db;
};

}

TEST_CASE_METHOD(ConnectionPoolFixture, "Create connection pool", "[connection pool]") {
  using pool_t = connection_pool<test_connection>;

  pool_t 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 pool_t = connection_pool<test_connection>;

  pool_t 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.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);
}

TEST_CASE("Try acquire connection", "[connection pool][try acquire]") {
  using pool_t = connection_pool<test_connection>;

  pool_t 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();
}