discourse/spec/components/freedom_patches/pool_drainer_spec.rb

require 'spec_helper'

describe 'pool drainer' do
  let(:pool) do
    ActiveRecord::Base.connection_pool
  end

  it 'can correctly drain the connection pool' do
    pool.drain
    old = pool.connections.length
    old.should == 1

    Thread.new do
      conn = pool.checkout
      pool.checkin conn
    end.join

    pool.connections.length.should == (old+1)
    pool.drain
    pool.connections.length.should == old
  end

  it 'can drain with idle time setting' do
    pool.drain
    old = pool.connections.length
    old.should == 1


    Thread.new do
      conn = pool.checkout
      pool.checkin conn
    end.join

    pool.connections.length.should == (old+1)
    pool.drain(1.minute)
    pool.connections.length.should == (old+1)

    # make sure we don't corrupt internal state
    20.times do
      conn = pool.checkout
      pool.checkin conn
      pool.drain
    end

  end

end