48 lines
892 B
Ruby
48 lines
892 B
Ruby
|
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
|