introduce a timer class for reaping connections

1 files changed, 46 insertions, 0 deletions

diff --git a/activerecord/test/cases/reaper_test.rb b/activerecord/test/cases/reaper_test.rb
new file mode 100644
index 0000000000..ed3daa175d
--- /dev/null
+++ b/activerecord/test/cases/reaper_test.rb
@@ -0,0 +1,46 @@
+require "cases/helper"
+
+module ActiveRecord
+  module ConnectionAdapters
+    class ReaperTest < ActiveRecord::TestCase
+      attr_reader :pool
+
+      def setup
+        super
+        @pool = ConnectionPool.new ActiveRecord::Base.connection_pool.spec
+      end
+
+      def teardown
+        super
+        @pool.connections.each(&:close)
+      end
+
+      # A reaper with nil time should never reap connections
+      def test_nil_time
+        conn = pool.checkout
+        pool.timeout = 0
+
+        count = pool.connections.length
+        conn.extend(Module.new { def active?; false; end; })
+
+        reaper = ConnectionPool::Reaper.new(pool, nil)
+        reaper.start
+        sleep 0.0001
+        assert_equal count, pool.connections.length
+      end
+
+      def test_some_time
+        conn = pool.checkout
+        pool.timeout = 0
+
+        count = pool.connections.length
+        conn.extend(Module.new { def active?; false; end; })
+
+        reaper = ConnectionPool::Reaper.new(pool, 0.0001)
+        reaper.start
+        sleep 0.0002
+        assert_equal(count - 1, pool.connections.length)
+      end
+    end
+  end
+end