vendor/rails/activerecord/test/cases/connection_pool_test.rb

   1 require "cases/helper"
   2
   3 class ConnectionManagementTest < ActiveRecord::TestCase
   4   def setup
   5     @env = {}
   6     @app = stub('App')
   7     @management = ActiveRecord::ConnectionAdapters::ConnectionManagement.new(@app)
   8
   9     @connections_cleared = false
  10     ActiveRecord::Base.stubs(:clear_active_connections!).with { @connections_cleared = true }
  11   end
  12
  13   test "clears active connections after each call" do
  14     @app.expects(:call).with(@env)
  15     @management.call(@env)
  16     assert @connections_cleared
  17   end
  18
  19   test "doesn't clear active connections when running in a test case" do
  20     @env['rack.test'] = true
  21     @app.expects(:call).with(@env)
  22     @management.call(@env)
  23     assert !@connections_cleared
  24   end
  25 end