slave_pools 0.1.0

data/spec/slave_pool_spec.rb

@@ -0,0 +1,43 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe SlavePoolsModule::SlavePool do
+  
+  context "Multiple slaves" do
+    before do
+      @slaves = ["db1", "db2", "db3"]
+      @slave_pool = SlavePoolsModule::SlavePool.new("name", @slaves.clone)
+    end
+    specify {@slave_pool.pool_size.should == 3}
+
+    it "should return items in a round robin fashion" do
+      first = @slaves.shift
+      @slave_pool.current.should == first
+      @slaves.each do |item|
+        @slave_pool.next.should == item
+      end
+      @slave_pool.next.should == first
+    end
+  end
+  
+  context "Single Slave" do
+    before do
+      @slaves = ["db1"]
+      @slave_pool = SlavePoolsModule::SlavePool.new("name", @slaves.clone)
+    end
+    specify {@slave_pool.pool_size.should == 1}
+
+    it "should return items in a round robin fashion" do
+      @slave_pool.current.should == "db1"
+      @slave_pool.next.should == "db1"
+      @slave_pool.next.should == "db1"
+    end
+    
+    it "shouldn't call next_reader! if there is only one slave" do
+      @slave_pool.should_not_receive(:next_index!)
+      @slave_pool.current.should == "db1"
+      @slave_pool.next.should == "db1"
+      @slave_pool.next.should == "db1"
+    end
+  end  
+end
+

data/spec/slave_pools_spec.rb

@@ -0,0 +1,69 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe SlavePools do
+
+  before(:all) do
+    ActiveRecord::Base.configurations = SLAVE_POOLS_SPEC_CONFIG
+    ActiveRecord::Base.establish_connection :test
+    @sql = 'SELECT NOW()'
+  end
+  
+  context "with no setup" do
+    it "should not error out if next slave is called and SlavePools is not set up" do
+      SlavePools.should_receive(:active?).and_return(false)
+      SlavePools.next_slave!.should be_nil
+    end
+    
+    it "should not error out if current is called and SlavePools is not set up" do
+      SlavePools.should_receive(:active?).and_return(false)
+      SlavePools.current.should be_nil
+    end
+    
+    it "should yield on a with_pool call if slave_pools is not active" do
+      SlavePools.should_receive(:active?).and_return(false)
+      ActiveRecord::Base.connection.should_receive(:execute)
+      SlavePools.with_pool('admin') {ActiveRecord::Base.connection.execute(@sql)}
+    end
+
+    it "should yield on a with_master call if slave_pools is not active" do
+      SlavePools.should_receive(:active?).and_return(false)
+      ActiveRecord::Base.connection.should_receive(:execute)
+      SlavePools.with_master {ActiveRecord::Base.connection.execute(@sql)}
+    end
+  end
+  
+  describe "Slave Pool Wrapper calls" do
+    before(:each) do
+      SlavePools.setup!
+      @proxy = ActiveRecord::Base.connection_proxy
+    end
+    
+    it 'should send next_slave! call to connection proxy' do
+      ActiveRecord::Base.should_receive(:respond_to?).exactly(1)
+      SlavePools.active?
+    end
+  
+    it 'should send next_slave! call to connection proxy' do
+      @proxy.should_receive(:next_slave!).exactly(1)
+      SlavePools.next_slave!
+    end
+    
+    it 'should send with_pool call to connection proxy' do
+      @proxy.should_receive(:with_pool).exactly(1)
+      SlavePools.with_pool('test')
+    end
+    
+    it 'should send with_master call to connection proxy' do
+      @proxy.should_receive(:with_master).exactly(1)
+      SlavePools.with_master
+    end
+
+    it 'should send current call to connection proxy' do
+      @proxy.should_receive(:current).exactly(1)
+      SlavePools.current
+    end
+
+  end
+
+end
+

data/spec/spec_helper.rb

@@ -0,0 +1,17 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'logger'
+
+require 'slave_pools'
+
+module Rails
+  def self.env
+    ActiveSupport::StringInquirer.new("test")
+  end
+end
+
+SLAVE_POOLS_SPEC_DIR = File.dirname(__FILE__)
+SLAVE_POOLS_SPEC_CONFIG = YAML::load(File.open(SLAVE_POOLS_SPEC_DIR + '/config/database.yml'))
+
+ActiveRecord::Base.logger = Logger.new(SLAVE_POOLS_SPEC_DIR + "/debug.log")
+ActiveRecord::Base.configurations = SLAVE_POOLS_SPEC_CONFIG

metadata

@@ -0,0 +1,132 @@
+--- !ruby/object:Gem::Specification
+name: slave_pools
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Dan Drabik
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.12
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.12
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.3.11
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.3.11
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Connection proxy for ActiveRecord for single master / multiple slave
+  database groups
+email: dan@kickstarter.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE
+- README.rdoc
+files:
+- lib/slave_pools.rb
+- lib/slave_pools/active_record_extensions.rb
+- lib/slave_pools/connection_proxy.rb
+- lib/slave_pools/observer_extensions.rb
+- lib/slave_pools/query_cache_compat.rb
+- lib/slave_pools/slave_pool.rb
+- LICENSE
+- README.rdoc
+- spec/config/database.yml
+- spec/connection_proxy_spec.rb
+- spec/slave_pool_spec.rb
+- spec/slave_pools_spec.rb
+- spec/spec_helper.rb
+- slave_pools.gemspec
+homepage: https://github.com/kickstarter/slave_pools
+licenses: []
+post_install_message: 
+rdoc_options:
+- --line-numbers
+- --inline-source
+- --title
+- slave_pools
+- --main
+- README.rdoc
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '1.2'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Connection proxy for ActiveRecord for single master / multiple slave database
+  groups
+test_files: []