active_record_host_pool 0.3.1

data/test/test_arhp.rb

@@ -0,0 +1,108 @@
+require File.expand_path('helper', File.dirname(__FILE__))
+
+class ActiveRecordHostPoolTest < ActiveSupport::TestCase
+  def setup
+    arhp_create_databases
+    arhp_create_models
+  end
+
+  def test_models_with_matching_hosts_should_share_a_connection
+    assert(Test1.connection.raw_connection == Test2.connection.raw_connection)
+    assert(Test3.connection.raw_connection == Test4.connection.raw_connection)
+  end
+
+  def test_models_without_matching_hosts_should_not_share_a_connection
+    assert(Test1.connection.raw_connection != Test4.connection.raw_connection)
+  end
+
+  def test_should_select_on_correct_database
+    action_should_use_correct_database(:select_all, "select 1")
+  end
+
+  def test_should_insert_on_correct_database
+    action_should_use_correct_database(:insert, "insert into tests values(NULL, 'foo')")
+  end
+
+  def test_connection_returns_a_proxy
+    assert Test1.connection.is_a?(ActiveRecordHostPool::ConnectionProxy)
+  end
+
+  def test_object_creation
+    Test1.create(:val => 'foo')
+    assert_equal("arhp_test_1", current_database(Test1))
+
+    Test3.create(:val => 'bar')
+    assert_equal("arhp_test_1", current_database(Test1))
+    assert_equal("arhp_test_3", current_database(Test3))
+
+    Test2.create(:val => 'bar')
+    assert_equal("arhp_test_2", current_database(Test2))
+    assert Test2.find_by_val('bar')
+    assert !Test1.find_by_val('bar')
+  end
+
+  def test_disconnect
+    Test1.create(:val => 'foo')
+    unproxied = Test1.connection.unproxied
+    Test1.connection_handler.clear_all_connections!
+    Test1.create(:val => 'foo')
+    assert(unproxied != Test1.connection.unproxied)
+  end
+
+  def test_checkout
+    connection = ActiveRecord::Base.connection_pool.checkout
+    assert(connection.is_a?(ActiveRecordHostPool::ConnectionProxy))
+    ActiveRecord::Base.connection_pool.checkin(connection)
+    c2 = ActiveRecord::Base.connection_pool.checkout
+    assert(c2 == connection)
+  end
+
+
+  def test_underlying_assumption_about_test_db
+    debug_me = false
+    # ensure connection
+    Test1.first
+
+    # which is the "default" DB to connect to?
+    first_db = Test1.connection.unproxied.instance_variable_get("@connection_options")[3]
+    puts "\nOk, we started on #{first_db}" if debug_me
+
+    switch_to_klass = case first_db
+      when "arhp_test_2"
+        Test1
+      when "arhp_test_1"
+        Test2
+    end
+    expected_database = switch_to_klass.connection.instance_variable_get("@database")
+
+    # switch to the other database
+    switch_to_klass.first
+    puts "\nAnd now we're on #{current_database(switch_to_klass)}" if debug_me
+
+    # get the current thread id so we can shoot ourselves in the head
+    thread_id = switch_to_klass.connection.select_value("select @@pseudo_thread_id")
+
+    # now, disable our auto-switching and trigger a mysql reconnect
+    switch_to_klass.connection.unproxied.stubs(:_switch_connection).returns(true)
+    switch_to_klass.connection.execute("KILL #{thread_id}")
+
+    # and finally, did mysql reconnect correctly?
+    puts "\nAnd now we end up on #{current_database(switch_to_klass)}" if debug_me
+    assert_equal expected_database, current_database(switch_to_klass)
+  end
+
+  def teardown
+    arhp_drop_databases
+  end
+
+  private
+
+  def action_should_use_correct_database(action, sql)
+    (1..4).each { |i|
+      klass = eval "Test#{i}"
+      desired_db = "arhp_test_#{i}"
+      klass.connection.send(action, sql)
+      assert_equal desired_db, current_database(klass)
+    }
+  end
+end

metadata

@@ -0,0 +1,166 @@
+--- !ruby/object:Gem::Specification 
+name: active_record_host_pool
+version: !ruby/object:Gem::Version 
+  hash: 17
+  prerelease: 
+  segments: 
+  - 0
+  - 3
+  - 1
+  version: 0.3.1
+platform: ruby
+authors: 
+- Ben Osheroff
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-02 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activerecord
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 2
+        - 3
+        - 5
+        version: 2.3.5
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: shoulda
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: mocha
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id005
+- !ruby/object:Gem::Dependency 
+  name: ruby-debug
+  prerelease: false
+  requirement: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id006
+description: ""
+email: 
+- ben@gimbo.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/active_record_host_pool/connection_adapter_mixin.rb
+- lib/active_record_host_pool/connection_proxy.rb
+- lib/active_record_host_pool/pool_proxy.rb
+- lib/active_record_host_pool.rb
+- README.md
+- test/database.yml
+- test/helper.rb
+- test/schema.rb
+- test/test.log
+- test/test_arhp.rb
+has_rdoc: true
+homepage: http://github.com/zendesk/active_record_host_pool
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.5.2
+signing_key: 
+specification_version: 3
+summary: When connecting to databases on one host, use just one connection
+test_files: 
+- test/database.yml
+- test/helper.rb
+- test/schema.rb
+- test/test.log
+- test/test_arhp.rb