octoball 0.1.0

data/lib/octoball/persistence.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+class Octoball
+  module ShardedPersistence
+    # cover methods using `self.class.connection`.
+    # NOTE: decrement! is implemented using increment!
+    [:update_columns, :increment!, :reload, :_delete_row, :_touch_row, :_update_row, :_create_record,
+     :transaction, :with_transaction_returning_status].each do |method|
+      class_eval <<-"END", __FILE__, __LINE__ + 1
+        def #{method}(*args, &block)
+          return super if !current_shard || current_shard == ActiveRecord::Base.current_shard
+          ActiveRecord::Base.connected_to(shard: current_shard, role: Octoball.current_role) do
+            super
+          end
+        end
+        ruby2_keywords(:#{method}) if respond_to?(:ruby2_keywords, true)
+      END
+    end
+
+    private
+
+    def init_internals
+      super
+      @current_shard = self.class.connection.current_shard
+    end
+  end
+
+  ::ActiveRecord::Base.prepend(ShardedPersistence)
+end

data/lib/octoball/relation_proxy.rb

@@ -0,0 +1,99 @@
+# frozen_string_literal: true
+
+class Octoball
+  class RelationProxy < BasicObject
+    attr_reader :current_shard
+
+    def initialize(rel, shard)
+      @rel = rel
+      self.current_shard = shard
+    end
+
+    def current_shard=(shard)
+      @current_shard = shard
+      @rel.current_shard = shard unless @rel.is_a?(::Enumerator)
+    end
+
+    def using(shard)
+      self.current_shard = shard
+      self
+    end
+
+    def ar_relation
+      @rel
+    end
+
+    def respond_to?(method, include_all = false)
+      return true if [:ar_relation, :current_shard, :current_shard=, :using].include?(method)
+      @rel.respond_to?(method, include_all)
+    end
+
+    ENUM_METHODS = (::Enumerable.instance_methods - ::Object.instance_methods).reject do |m|
+      ::ActiveRecord::Relation.instance_method(m).source_location rescue nil
+    end + [:each, :map, :index_by]
+    ENUM_WITH_BLOCK_METHODS = [:find, :select, :none?, :any?, :one?, :many?, :sum]
+
+    def method_missing(method, *args, &block)
+      # raise NoMethodError unless the method is defined in @rel
+      return @rel.public_send(method, *args, &block) unless @rel.respond_to?(method)
+
+      preamble = <<-EOS
+        def #{method}(*margs, &mblock)
+          return @rel.#{method}(*margs, &mblock) unless @current_shard
+      EOS
+      postamble = <<-EOS
+          return ret unless ret.is_a?(::ActiveRecord::Relation) || ret.is_a?(::ActiveRecord::QueryMethods::WhereChain) || ret.is_a?(::Enumerator)
+          ::Octoball::RelationProxy.new(ret, @current_shard)
+        end
+        ruby2_keywords(:#{method}) if respond_to?(:ruby2_keywords, true)
+      EOS
+      connected_to = '::ActiveRecord::Base.connected_to(role: ::Octoball.current_role, shard: @current_shard)'
+
+      if ENUM_METHODS.include?(method)
+        ::Octoball::RelationProxy.class_eval <<-EOS, __FILE__, __LINE__ - 1
+          #{preamble}
+          ret = #{connected_to} { @rel.to_a }.#{method}(*margs, &mblock)
+          #{postamble}
+        EOS
+      elsif ENUM_WITH_BLOCK_METHODS.include?(method)
+        ::Octoball::RelationProxy.class_eval <<-EOS, __FILE__, __LINE__ - 1
+          #{preamble}
+          ret = nil
+          if mblock
+            ret = #{connected_to} { @rel.to_a }.#{method}(*margs, &mblock)
+          else
+            #{connected_to} { ret = @rel.#{method}(*margs, &mblock); nil } # return nil avoid loading relation
+          end
+          #{postamble}
+        EOS
+      else
+        ::Octoball::RelationProxy.class_eval <<-EOS, __FILE__, __LINE__ - 1
+          #{preamble}
+          ret = nil
+          #{connected_to} { ret = @rel.#{method}(*margs, &mblock); nil } # return nil to avoid loading relation
+          #{postamble}
+        EOS
+      end
+
+      public_send(method, *args, &block)
+    end
+    ruby2_keywords(:method_missing) if respond_to?(:ruby2_keywords, true)
+
+    def inspect
+      return @rel.inspect unless @current_shard
+      ::ActiveRecord::Base.connected_to(shard: @current_shard, role: ::Octoball.current_role) { @rel.inspect }
+    end
+
+    def ==(obj)
+      return false if obj.respond_to?(:current_shard) && obj.current_shard != @current_shard
+      return @rel == obj unless @current_shard
+      ::ActiveRecord::Base.connected_to(shard: @current_shard, role: ::Octoball.current_role) { @rel == obj }
+    end
+
+    def ===(obj)
+      return false if obj.respond_to?(:current_shard) && obj.current_shard != @current_shard
+      return @rel === obj unless @current_shard
+      ::ActiveRecord::Base.connected_to(shard: @current_shard, role: ::Octoball.current_role) { @rel === obj }
+    end
+  end
+end

data/lib/octoball/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+class Octoball
+  VERSION = '0.1.0'
+end

data/octoball.gemspec

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+$LOAD_PATH.push File.expand_path('../lib', __FILE__)
+require 'octoball/version'
+
+Gem::Specification.new do |s|
+  s.name        = 'octoball'
+  s.version     = Octoball::VERSION
+  s.licenses    = ['MIT']
+  s.summary     = "Octopus-like Database Sharding Helper for ActiveRecord 6.1+"
+  s.description = "Octoball provides Octopus-like database sharding helper methods for ActiveRecord 6.1 or later, using Rails' native horizontal sharding handling. This provides migration path to Rails 6.1+ for applications using Octopus gem with older Rails."
+  s.authors     = ["Tomoki Sekiyama"]
+  s.email       = 'tomoki.sekiyama@aktsk.jp'
+  s.files       = `git ls-files`.split("\n")
+  s.test_files  = `git ls-files -- {spec}/*`.split("\n")
+  s.homepage    = 'https://github.com/aktsk/octoball'
+  s.require_paths = ['lib']
+
+  s.required_ruby_version = '>= 2.7.0'
+
+  s.add_dependency 'activerecord', '>= 6.1'
+  s.add_dependency 'activesupport', '>= 6.1'
+
+  s.add_development_dependency 'mysql2'
+  s.add_development_dependency 'rake'
+  s.add_development_dependency 'rspec', '>= 3'
+  s.add_development_dependency 'rubocop'
+  s.add_development_dependency 'byebug'
+end

data/spec/migration/1_test_tables.rb

@@ -0,0 +1,84 @@
+class TestTables < ActiveRecord::Migration[6.1]
+  def change
+    create_table(:users) do |u|
+      u.string :name
+      u.integer :number
+      u.boolean :admin
+      u.datetime :created_at
+      u.datetime :updated_at
+    end
+
+    create_table(:clients) do |u|
+      u.string :country
+      u.string :name
+    end
+
+    create_table(:cats) do |u|
+      u.string :name
+    end
+
+    create_table(:items) do |u|
+      u.string :name
+      u.integer :client_id
+    end
+
+    create_table(:computers) do |u|
+      u.string :name
+    end
+
+    create_table(:keyboards) do |u|
+      u.string :name
+      u.integer :computer_id
+    end
+
+    create_table(:roles) do |u|
+      u.string :name
+    end
+
+    create_table(:permissions) do |u|
+      u.string :name
+    end
+
+    create_table(:permissions_roles, :id => false) do |u|
+      u.integer :role_id
+      u.integer :permission_id
+    end
+
+    create_table(:assignments) do |u|
+      u.integer :programmer_id
+      u.integer :project_id
+    end
+
+    create_table(:programmers) do |u|
+      u.string :name
+    end
+
+    create_table(:projects) do |u|
+      u.string :name
+    end
+
+    create_table(:comments) do |u|
+      u.string :name
+      u.string :commentable_type
+      u.integer :commentable_id
+      u.boolean :open, default: false
+    end
+
+    create_table(:parts) do |u|
+      u.string :name
+      u.integer :item_id
+    end
+
+    create_table(:yummy) do |u|
+      u.string :name
+    end
+
+    create_table(:adverts) do |u|
+      u.string :name
+    end
+
+    create_table(:custom) do |u|
+      u.string :value
+    end
+  end
+end

data/spec/migration/2_alone_shard_tables.rb

@@ -0,0 +1,19 @@
+class AloneShardTables < ActiveRecord::Migration[6.1]
+  def change
+    create_table(:mmorpg_players) do |u|
+      u.string :player_name
+    end
+
+    create_table(:weapons) do |u|
+      u.integer :mmorpg_player_id
+      u.string :name
+      u.string :hand
+    end
+
+    create_table(:skills) do |u|
+      u.integer :mmorpg_player_id
+      u.integer :weapon_id
+      u.string :name
+    end
+  end
+end

data/spec/models/application_record.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+class ApplicationRecord < ActiveRecord::Base
+  self.abstract_class = true
+
+  connects_to shards: {
+    master:      { writing: :shard1 },
+    brazil:      { writing: :shard2 },
+    canada:      { writing: :shard3 },
+    russia:      { writing: :shard4 },
+    alone_shard: { writing: :shard5 },
+  }
+end

data/spec/octoball/association_shard_tracking_spec.rb

@@ -0,0 +1,1024 @@
+require 'spec_helper'
+
+describe Octoball::ShardedSingularAssociation, :shards => [:brazil, :master, :canada] do
+  describe 'when you have a 1 x 1 relationship' do
+    before(:each) do
+      @computer_brazil = Computer.using(:brazil).create!(:name => 'Computer Brazil')
+      @computer_master = Computer.create!(:name => 'Computer Brazil')
+      @keyboard_brazil = Keyboard.using(:brazil).create!(:name => 'Keyboard Brazil', :computer => @computer_brazil)
+      @keyboard_master = Keyboard.create!(:name => 'Keyboard Master', :computer => @computer_master)
+    end
+
+    it 'should find the models' do
+      expect(@keyboard_master.computer).to eq(@computer_master)
+      expect(@keyboard_brazil.computer).to eq(@computer_brazil)
+    end
+
+    it 'should read correctly the relationed model' do
+      new_computer_brazil = Computer.using(:brazil).create!(:name => 'New Computer Brazil')
+      _new_computer_hello = Computer.create!(:name => 'New Computer Brazil')
+      @keyboard_brazil.computer = new_computer_brazil
+      @keyboard_brazil.save
+      @keyboard_brazil.reload
+      expect(@keyboard_brazil.computer_id).to eq(new_computer_brazil.id)
+      expect(@keyboard_brazil.computer).to eq(new_computer_brazil)
+      new_computer_brazil.save
+      new_computer_brazil.reload
+      expect(new_computer_brazil.keyboard).to eq(@keyboard_brazil)
+    end
+
+    it 'should join/merge correctly with the relationed model' do
+      new_computer_brazil = Computer.using(:brazil).create!(:name => 'New Computer Brazil')
+      @keyboard_brazil.computer = new_computer_brazil
+      @keyboard_brazil.save
+      expect(Keyboard.using(:brazil).joins(:computer).merge(Computer.where(name: 'New Computer Brazil'))).to eq([@keyboard_brazil])
+      expect(Keyboard.using(:brazil).joins(:computer).merge(Computer.using(:master).where(name: 'New Computer Brazil'))).to eq([@keyboard_brazil])
+    end
+
+    it 'should work when using #build_computer or #build_keyboard' do
+      c = Computer.using(:brazil).create!(:name => 'Computer Brazil')
+      k = c.build_keyboard(:name => 'Building keyboard')
+      c.save
+      k.save
+      expect(c.keyboard).to eq(k)
+      expect(k.computer_id).to eq(c.id)
+      expect(k.computer).to eq(c)
+    end
+
+    it 'should work when using #create_computer or #create_keyboard' do
+      c = Computer.using(:brazil).create!(:name => 'Computer Brazil')
+      k = c.create_keyboard(:name => 'Building keyboard')
+      c.save
+      k.save
+      expect(c.keyboard).to eq(k)
+      expect(k.computer_id).to eq(c.id)
+      expect(k.computer).to eq(c)
+    end
+
+    it 'should include models' do
+      c = Computer.using(:brazil).create!(:name => 'Computer Brazil')
+      k = c.create_keyboard(:name => 'Building keyboard')
+      c.save
+      k.save
+
+      expect(Computer.using(:brazil).includes(:keyboard).find(c.id)).to eq(c)
+    end
+  end
+
+  describe 'when you have a N x N relationship' do
+    before(:each) do
+      @brazil_role = Role.using(:brazil).create!(:name => 'Brazil Role')
+      @master_role = Role.create!(:name => 'Master Role')
+      @permission_brazil = Permission.using(:brazil).create!(:name => 'Brazil Permission')
+      @permission_master = Permission.using(:master).create!(:name => 'Master Permission')
+      @brazil_role.permissions << @permission_brazil
+      @brazil_role.save
+      Client.using(:master).create!(:name => 'teste')
+    end
+
+    it 'should find all models in the specified shard' do
+      expect(@brazil_role.permission_ids).to eq([@permission_brazil.id])
+      expect(@brazil_role.permissions).to eq([@permission_brazil])
+
+      expect(@brazil_role.permissions.first).to eq(@permission_brazil)
+      expect(@brazil_role.permissions.first!).to eq(@permission_brazil)
+      expect(@brazil_role.permissions.last).to eq(@permission_brazil)
+    end
+
+    it 'should finds the client that the item belongs' do
+      expect(@permission_brazil.role_ids).to eq([@brazil_role.id])
+      expect(@permission_brazil.roles).to eq([@brazil_role])
+
+      expect(@permission_brazil.roles.first).to eq(@brazil_role)
+      expect(@permission_brazil.roles.first!).to eq(@brazil_role)
+      expect(@permission_brazil.roles.last).to eq(@brazil_role)
+    end
+
+    it 'should update the attribute for the item' do
+      new_brazil_role = Role.using(:brazil).create!(:name => 'new Role')
+      @permission_brazil.roles = [new_brazil_role]
+      expect(@permission_brazil.roles).to eq([new_brazil_role])
+      @permission_brazil.save
+      @permission_brazil.reload
+      expect(@permission_brazil.role_ids).to eq([new_brazil_role.id])
+      expect(@permission_brazil.roles).to eq([new_brazil_role])
+    end
+
+    it 'should work for build method' do
+      new_brazil_role = Role.using(:brazil).create!(:name => 'Brazil Role')
+      c = new_brazil_role.permissions.create(:name => 'new Permission')
+      c.save
+      new_brazil_role.save
+      expect(c.roles).to eq([new_brazil_role])
+      expect(new_brazil_role.permissions).to eq([c])
+    end
+
+    describe 'it should work when using' do
+      before(:each) do
+        @permission_brazil_2 = Permission.using(:brazil).create!(:name => 'Brazil Item 2')
+        @role = Role.using(:brazil).create!(:name => 'testes')
+      end
+
+      it 'update' do
+        @permission_brazil_2.update(:role_ids => [@role.id])
+        expect(@permission_brazil_2.roles.to_set).to eq([@role].to_set)
+      end
+
+      it '<<' do
+        @permission_brazil_2.roles << @role
+        @role.save
+        @permission_brazil_2.save
+        @permission_brazil_2.reload
+        expect(@permission_brazil_2.roles.to_set).to eq([@role].to_set)
+      end
+
+      it 'build' do
+        role = @permission_brazil_2.roles.build(:name => 'Builded Role')
+        @permission_brazil_2.save
+        expect(@permission_brazil_2.roles.to_set).to eq([role].to_set)
+      end
+
+      it 'create' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.to_set).to eq([role].to_set)
+      end
+
+      it 'create' do
+        role = @permission_brazil_2.roles.create!(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.to_set).to eq([role].to_set)
+      end
+
+      it 'count' do
+        expect(@permission_brazil_2.roles.count).to eq(0)
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.count).to eq(1)
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.count).to eq(2)
+      end
+
+      it 'size' do
+        expect(@permission_brazil_2.roles.size).to eq(0)
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.size).to eq(1)
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.size).to eq(2)
+      end
+
+      it 'length' do
+        expect(@permission_brazil_2.roles.length).to eq(0)
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.length).to eq(1)
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.length).to eq(2)
+      end
+
+      it 'empty?' do
+        expect(@permission_brazil_2.roles.empty?).to be true
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.empty?).to be false
+      end
+
+      it 'delete_all' do
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.empty?).to be false
+        @permission_brazil_2.roles.delete_all
+        expect(@permission_brazil_2.roles.empty?).to be true
+      end
+
+      it 'destroy_all' do
+        _role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.empty?).to be false
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.empty?).to be true
+      end
+
+      it 'find' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.first).to eq(role)
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.first).to be_nil
+      end
+
+      it 'where' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1')).to eq([role])
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1')).to be_empty
+      end
+
+      it 'map' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.map(&:id)).to eq([role.id])
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.map(&:id)).to be_empty
+      end
+
+      it 'where + map' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').map(&:id)).to eq([role.id])
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').map(&:id)).to be_empty
+      end
+
+      # each_with_index is not listed in active_record/relation/delegation.rb
+      it 'where + each_with_index + map (enum method chain)' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').each_with_index.map { |r, i| [r.id, i]}).to eq([[role.id, 0]])
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').each_with_index.map { |r, i| [r.id, i]}).to be_empty
+      end
+
+      # sum & index_by is specialized in active_support/core_ext/enumerable.rb
+      it 'where + sum' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').sum(&:id)).to eq(role.id)
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').sum(&:id)).to eq(0)
+      end
+
+      it 'where + index_by' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').index_by(&:id)).to eq(role.id => role)
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').index_by(&:id)).to be_empty
+      end
+
+      it 'where + find' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').find([role.id])).to eq([role])
+        @permission_brazil_2.roles.destroy_all
+        expect { @permission_brazil_2.roles.where('1=1').find([role.id]) }.to raise_error ActiveRecord::RecordNotFound
+      end
+
+      it 'where + find with block' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').find { |r| r.id == role.id }).to eq(role)
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').find { |r| r.id == role.id }).to be_nil
+      end
+
+      it 'where + select' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').select(:name).first.name).to eq(role.name)
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').select(:name)).to be_empty
+      end
+
+      it 'where + select with block' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').select { |r| r.id == role.id }).to eq([role])
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').select { |r| r.id == role.id }).to be_empty
+      end
+
+      it 'where + any?' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').any?).to be true
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').any?).to be false
+      end
+
+      it 'where + any? with block' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.where('1=1').any? { |r| r.id == role.id }).to be true
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.where('1=1').any? { |r| r.id == role.id }).to be false
+      end
+
+      it 'exists?' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.exists?(role.id)).to be true
+        @permission_brazil_2.roles.destroy_all
+        expect(@permission_brazil_2.roles.exists?(role.id)).to be false
+      end
+
+      it 'clear' do
+        _rol = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.empty?).to be false
+        @permission_brazil_2.roles.clear
+        expect(@permission_brazil_2.roles.empty?).to be true
+      end
+
+      it 'delete' do
+        role = @permission_brazil_2.roles.create(:name => 'Builded Role')
+        expect(@permission_brazil_2.roles.empty?).to be false
+        @permission_brazil_2.roles.delete(role)
+        @permission_brazil_2.reload
+        @role.reload
+        expect(@role.permissions).to eq([])
+        expect(@permission_brazil_2.roles).to eq([])
+      end
+    end
+  end
+
+  describe 'when you have has_many :through' do
+    before(:each) do
+      @programmer = Programmer.using(:brazil).create!(:name => 'Thiago')
+      @project = Project.using(:brazil).create!(:name => 'RubySoc')
+      @project2 = Project.using(:brazil).create!(:name => 'Cobol Application')
+      @programmer.projects << @project
+      @programmer.save
+      Project.using(:master).create!(:name => 'Project Master')
+    end
+
+    it 'should find all models in the specified shard' do
+      expect(@programmer.project_ids).to eq([@project.id])
+      expect(@programmer.projects).to eq([@project])
+
+      expect(@programmer.projects.first).to eq(@project)
+      expect(@programmer.projects.last).to eq(@project)
+    end
+
+    it 'should update the attribute for the item' do
+      new_brazil_programmer = Programmer.using(:brazil).create!(:name => 'Joao')
+      @project.programmers = [new_brazil_programmer]
+      expect(@project.programmers).to eq([new_brazil_programmer])
+      @project.save
+      @project.reload
+      expect(@project.programmer_ids).to eq([new_brazil_programmer.id])
+      expect(@project.programmers).to eq([new_brazil_programmer])
+    end
+
+    it 'should work for create method' do
+      new_brazil_programmer = Programmer.using(:brazil).create!(:name => 'Joao')
+      c = new_brazil_programmer.projects.create(:name => 'new Project')
+      c.save
+      new_brazil_programmer.save
+      expect(c.programmers).to eq([new_brazil_programmer])
+      expect(new_brazil_programmer.projects).to eq([c])
+    end
+
+    describe 'it should work when using' do
+      before(:each) do
+        @new_brazil_programmer = Programmer.using(:brazil).create!(:name => 'Jose')
+        @project = Project.using(:brazil).create!(:name => 'VB Application :-(')
+      end
+
+      it 'update' do
+        @new_brazil_programmer.update(:project_ids => [@project.id])
+        expect(@new_brazil_programmer.projects.to_set).to eq([@project].to_set)
+      end
+
+      it '<<' do
+        @new_brazil_programmer.projects << @project
+        @project.save
+        @new_brazil_programmer.save
+        @new_brazil_programmer.reload
+        expect(@new_brazil_programmer.projects.to_set).to eq([@project].to_set)
+      end
+
+      it 'build' do
+        role = @new_brazil_programmer.projects.build(:name => 'New VB App :-/')
+        @new_brazil_programmer.save
+        expect(@new_brazil_programmer.projects.to_set).to eq([role].to_set)
+      end
+
+      it 'create' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.to_set).to eq([role].to_set)
+      end
+
+      it 'create' do
+        role = @new_brazil_programmer.projects.create!(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.to_set).to eq([role].to_set)
+      end
+
+      it 'count' do
+        expect(@new_brazil_programmer.projects.count).to eq(0)
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.count).to eq(1)
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.count).to eq(2)
+      end
+
+      it 'size' do
+        expect(@new_brazil_programmer.projects.size).to eq(0)
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.size).to eq(1)
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.size).to eq(2)
+      end
+
+      it 'length' do
+        expect(@new_brazil_programmer.projects.length).to eq(0)
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.length).to eq(1)
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.length).to eq(2)
+      end
+
+      it 'empty?' do
+        expect(@new_brazil_programmer.projects.empty?).to be true
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.empty?).to be false
+      end
+
+      it 'delete_all' do
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.empty?).to be false
+        @new_brazil_programmer.projects.delete_all
+        expect(@new_brazil_programmer.projects.empty?).to be true
+      end
+
+      it 'destroy_all' do
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.empty?).to be false
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.empty?).to be true
+      end
+
+      it 'find' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.first).to eq(role)
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.first).to be_nil
+      end
+
+      it 'where' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1')).to eq([role])
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1')).to be_empty
+      end
+
+      it 'map' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.map(&:id)).to eq([role.id])
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.map(&:id)).to be_empty
+      end
+
+      it 'where + map' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').map(&:id)).to eq([role.id])
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').map(&:id)).to be_empty
+      end
+
+      it 'where + each_with_index + map (enum method chain)' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').each_with_index.map { |r, i| [r.id, i] }).to eq([[role.id, 0]])
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').each_with_index.map { |r, i| [r.id, i] }).to be_empty
+      end
+
+      it 'where + sum' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').sum(&:id)).to eq(role.id)
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').sum(&:id)).to eq(0)
+      end
+
+      it 'where + index_by' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').index_by(&:id)).to eq(role.id => role)
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').index_by(&:id)).to be_empty
+      end
+
+      it 'where + find' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').find(role.id)).to eq(role)
+        @new_brazil_programmer.projects.destroy_all
+        expect { @new_brazil_programmer.projects.where('1=1').find(role.id) }.to raise_error ActiveRecord::RecordNotFound
+      end
+
+      it 'where + find with block' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').find { |r| r.id == role.id }).to eq(role)
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').find { |r| r.id == role.id }).to be_nil
+      end
+
+      it 'where + select' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').select(:name).first.name).to eq(role.name)
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').select(:name)).to be_empty
+      end
+
+      it 'where + select with block' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').select { |r| r.id == role.id }).to eq([role])
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').select { |r| r.id == role.id }).to be_empty
+      end
+
+      it 'where + any?' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').any?).to be true
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').any?).to be false
+      end
+
+      it 'where + any? with block' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.where('1=1').any? { |r| r.id == role.id }).to be true
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.where('1=1').any? { |r| r.id == role.id }).to be false
+      end
+
+      it 'exists?' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.exists?(role.id)).to be true
+        @new_brazil_programmer.projects.destroy_all
+        expect(@new_brazil_programmer.projects.exists?(role.id)).to be false
+      end
+
+      it 'clear' do
+        _role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.empty?).to be false
+        @new_brazil_programmer.projects.clear
+        expect(@new_brazil_programmer.projects.empty?).to be true
+      end
+
+      it 'delete' do
+        role = @new_brazil_programmer.projects.create(:name => 'New VB App :-/')
+        expect(@new_brazil_programmer.projects.empty?).to be false
+        @new_brazil_programmer.projects.delete(role)
+        @new_brazil_programmer.reload
+        @project.reload
+        expect(@project.programmers).to eq([])
+        expect(@new_brazil_programmer.projects).to eq([])
+      end
+    end
+  end
+
+  describe 'when you have a 1 x N relationship' do
+    before(:each) do
+      @brazil_client = Client.using(:brazil).create!(:name => 'Brazil Client')
+      @master_client = Client.create!(:name => 'Master Client')
+      @item_brazil = Item.using(:brazil).create!(:name => 'Brazil Item', :client => @brazil_client)
+      @item_master = Item.create!(:name => 'Master Item', :client => @master_client)
+      @brazil_client = Client.using(:brazil).find_by_name('Brazil Client')
+      Client.using(:master).create!(:name => 'teste')
+    end
+
+    it 'should find all models in the specified shard' do
+      expect(@brazil_client.item_ids).to eq([@item_brazil.id])
+      expect(@brazil_client.items).to eq([@item_brazil])
+
+      expect(@brazil_client.items.last).to eq(@item_brazil)
+      expect(@brazil_client.items.first).to eq(@item_brazil)
+    end
+
+    it 'should finds the client that the item belongs' do
+      expect(@item_brazil.client).to eq(@brazil_client)
+    end
+
+    it 'should raise error if you try to add a record from a different shard' do
+      expect do
+        @brazil_client.items << Item.using(:canada).create!(:name => 'New User')
+      end.to raise_error(Octoball::MismatchedShards)
+    end
+
+    it 'should make difference if the shard is a symbol for raising MismatchedShards' do
+      expect do
+        @brazil_client.items << Item.using(:brazil).create!(:name => 'New Brazil Item')
+      end.not_to raise_error
+    end
+
+    it 'should update the attribute for the item' do
+      new_brazil_client = Client.using(:brazil).create!(:name => 'new Client')
+      @item_brazil.client = new_brazil_client
+      expect(@item_brazil.client).to eq(new_brazil_client)
+      @item_brazil.save
+      @item_brazil.reload
+      expect(@item_brazil.client_id).to eq(new_brazil_client.id)
+      expect(@item_brazil.client).to eq(new_brazil_client)
+    end
+
+    it 'should work for build method' do
+      item2 = Item.using(:brazil).create!(:name => 'Brazil Item')
+      c = item2.create_client(:name => 'new Client')
+      c.save
+      item2.save
+      expect(item2.client).to eq(c)
+      expect(c.items).to eq([item2])
+    end
+
+    context 'when calling methods on a collection generated by an association' do
+      let(:collection) { @brazil_client.items }
+      before :each do
+        @brazil_client.items.create(:name => 'Brazil Item #2')
+      end
+
+      it "can call collection indexes directly without resetting the collection's current_shard" do
+        last_item = collection[1]
+        expect(collection.length).to eq(2)
+        expect(collection).to eq([collection[0], last_item])
+      end
+
+      it "can call methods on the collection without resetting the collection's current_shard" do
+        last_item = collection[collection.size - 1]
+        expect(collection.length).to eq(2)
+        expect(collection).to eq([collection[0], last_item])
+      end
+    end
+
+    describe 'it should work when using' do
+      before(:each) do
+        @item_brazil_2 = Item.using(:brazil).create!(:name => 'Brazil Item 2')
+        expect(@brazil_client.items.to_set).to eq([@item_brazil].to_set)
+      end
+
+      it 'update' do
+        @brazil_client.update(:item_ids => [@item_brazil_2.id, @item_brazil.id])
+        expect(@brazil_client.items.to_set).to eq([@item_brazil, @item_brazil_2].to_set)
+      end
+
+      it '<<' do
+        @brazil_client.items << @item_brazil_2
+        expect(@brazil_client.items.to_set).to eq([@item_brazil, @item_brazil_2].to_set)
+      end
+
+      it 'all' do
+        item = @brazil_client.items.build(:name => 'Builded Item')
+        item.save
+        i = @brazil_client.items
+        expect(i.to_set).to eq([@item_brazil, item].to_set)
+        expect(i.reload.all.to_set).to eq([@item_brazil, item].to_set)
+      end
+
+      it 'build' do
+        item = @brazil_client.items.build(:name => 'Builded Item')
+        item.save
+        expect(@brazil_client.items.to_set).to eq([@item_brazil, item].to_set)
+      end
+
+      it 'create' do
+        item = @brazil_client.items.create(:name => 'Builded Item')
+        expect(@brazil_client.items.to_set).to eq([@item_brazil, item].to_set)
+      end
+
+      it 'count' do
+        expect(@brazil_client.items.count).to eq(1)
+        _itm = @brazil_client.items.create(:name => 'Builded Item')
+        expect(@brazil_client.items.count).to eq(2)
+      end
+
+      it 'size' do
+        expect(@brazil_client.items.size).to eq(1)
+        _itm = @brazil_client.items.create(:name => 'Builded Item')
+        expect(@brazil_client.items.size).to eq(2)
+      end
+
+      it 'create!' do
+        item = @brazil_client.items.create!(:name => 'Builded Item')
+        expect(@brazil_client.items.to_set).to eq([@item_brazil, item].to_set)
+      end
+
+      it 'length' do
+        expect(@brazil_client.items.length).to eq(1)
+        _itm = @brazil_client.items.create(:name => 'Builded Item')
+        expect(@brazil_client.items.length).to eq(2)
+      end
+
+      it 'empty?' do
+        expect(@brazil_client.items.empty?).to be false
+        c = Client.create!(:name => 'Client1')
+        expect(c.items.empty?).to be true
+      end
+
+      it 'delete' do
+        expect(@brazil_client.items.empty?).to be false
+        @brazil_client.items.delete(@item_brazil)
+        @brazil_client.reload
+        @item_brazil.reload
+        expect(@item_brazil.client).to be_nil
+        expect(@brazil_client.items).to eq([])
+        expect(@brazil_client.items.empty?).to be true
+      end
+
+      it 'delete_all' do
+        expect(@brazil_client.items.empty?).to be false
+        @brazil_client.items.delete_all
+        expect(@brazil_client.items.empty?).to be true
+      end
+
+      it 'destroy_all' do
+        expect(@brazil_client.items.empty?).to be false
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.empty?).to be true
+      end
+
+      it 'find' do
+        expect(@brazil_client.items.first).to eq(@item_brazil)
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.first).to be_nil
+      end
+
+      it 'where' do
+        expect(@brazil_client.items.where('1=1')).to eq([@item_brazil])
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1')).to be_empty
+      end
+
+      it 'map' do
+        expect(@brazil_client.items.map(&:id)).to eq([@item_brazil.id])
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.map(&:id)).to be_empty
+      end
+
+      it 'where + map' do
+        expect(@brazil_client.items.where('1=1').map(&:id)).to eq([@item_brazil.id])
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').map(&:id)).to be_empty
+      end
+
+      it 'where + each_with_index + map (enum method chain)' do
+        expect(@brazil_client.items.where('1=1').each_with_index.map { |r, i| [r.id, i]}).to eq([[@item_brazil.id, 0]])
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').each_with_index.map { |r, i| [r.id, i]}).to be_empty
+      end
+
+      it 'where + sum' do
+        expect(@brazil_client.items.where('1=1').sum(&:id)).to eq(@item_brazil.id)
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').sum(&:id)).to eq(0)
+      end
+
+      it 'where + index_by' do
+        expect(@brazil_client.items.where('1=1').index_by(&:id)).to eq(@item_brazil.id => @item_brazil)
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').index_by(&:id)).to be_empty
+      end
+
+      it 'where + find' do
+        expect(@brazil_client.items.where('1=1').find(@item_brazil.id)).to eq(@item_brazil)
+        @brazil_client.items.destroy_all
+        expect { @brazil_client.items.where('1=1').find(@item_brazil.id) }.to raise_error ActiveRecord::RecordNotFound
+      end
+
+      it 'where + find with block' do
+        expect(@brazil_client.items.where('1=1').find { |i| i.id == @item_brazil.id }).to eq(@item_brazil)
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').find { |i| i.id == @item_brazil.id }).to be_nil
+      end
+
+      it 'where + select' do
+        expect(@brazil_client.items.where('1=1').select(:name).first.name).to eq(@item_brazil.name)
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').select(:name)).to be_empty
+      end
+
+      it 'where + select with block' do
+        expect(@brazil_client.items.where('1=1').select { |i| i.id == @item_brazil.id }).to eq([@item_brazil])
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').select { |i| i.id == @item_brazil.id }).to be_empty
+      end
+
+      it 'where + any?' do
+        expect(@brazil_client.items.where('1=1').any?).to be true
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').any?).to be false
+      end
+
+      it 'where + any? with block' do
+        expect(@brazil_client.items.where('1=1').any? { |i| i.id == @item_brazil.id }).to be true
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.where('1=1').any? { |i| i.id == @item_brazil.id }).to be false
+      end
+
+      it 'exists?' do
+        expect(@brazil_client.items.exists?(@item_brazil.id)).to be true
+        @brazil_client.items.destroy_all
+        expect(@brazil_client.items.exists?(@item_brazil.id)).to be false
+      end
+
+      it 'uniq' do
+        expect(@brazil_client.items.uniq).to eq([@item_brazil])
+      end
+
+      it 'clear' do
+        expect(@brazil_client.items.empty?).to be false
+        @brazil_client.items.clear
+        expect(@brazil_client.items.empty?).to be true
+      end
+    end
+  end
+
+  describe 'when you have a 1 x N polymorphic relationship' do
+    before(:each) do
+      @brazil_client = Client.using(:brazil).create!(:name => 'Brazil Client')
+      @master_client = Client.create!(:name => 'Master Client')
+      @comment_brazil = Comment.using(:brazil).create!(:name => 'Brazil Comment', :commentable => @brazil_client)
+      @comment_master = Comment.create!(:name => 'Master Comment', :commentable => @master_client)
+      @brazil_client = Client.using(:brazil).find_by_name('Brazil Client')
+      Client.using(:master).create!(:name => 'teste')
+    end
+
+    it 'should find all models in the specified shard' do
+      expect(@brazil_client.comment_ids).to eq([@comment_brazil.id])
+      expect(@brazil_client.comments).to eq([@comment_brazil])
+    end
+
+    it 'should finds the client that the comment belongs' do
+      expect(@comment_brazil.commentable).to eq(@brazil_client)
+    end
+
+    it 'should update the attribute for the comment' do
+      new_brazil_client = Client.using(:brazil).create!(:name => 'new Client')
+      @comment_brazil.commentable = new_brazil_client
+      expect(@comment_brazil.commentable).to eq(new_brazil_client)
+      @comment_brazil.save
+      @comment_brazil.reload
+      expect(@comment_brazil.commentable_id).to eq(new_brazil_client.id)
+      expect(@comment_brazil.commentable).to eq(new_brazil_client)
+    end
+
+    describe 'it should work when using' do
+      before(:each) do
+        @comment_brazil_2 = Comment.using(:brazil).create!(:name => 'Brazil Comment 2')
+        expect(@brazil_client.comments.to_set).to eq([@comment_brazil].to_set)
+      end
+
+      it 'update' do
+        @brazil_client.update(:comment_ids => [@comment_brazil_2.id, @comment_brazil.id])
+        expect(@brazil_client.comments.to_set).to eq([@comment_brazil, @comment_brazil_2].to_set)
+      end
+
+      it '<<' do
+        @brazil_client.comments << @comment_brazil_2
+        expect(@brazil_client.comments.to_set).to eq([@comment_brazil, @comment_brazil_2].to_set)
+      end
+
+      it 'all' do
+        comment = @brazil_client.comments.build(:name => 'Builded Comment')
+        comment.save
+        c = @brazil_client.comments
+        expect(c.to_set).to eq([@comment_brazil, comment].to_set)
+        expect(c.reload.all.to_set).to eq([@comment_brazil, comment].to_set)
+      end
+
+      it 'build' do
+        comment = @brazil_client.comments.build(:name => 'Builded Comment')
+        comment.save
+        expect(@brazil_client.comments.to_set).to eq([@comment_brazil, comment].to_set)
+      end
+
+      it 'create' do
+        comment = @brazil_client.comments.create(:name => 'Builded Comment')
+        expect(@brazil_client.comments.to_set).to eq([@comment_brazil, comment].to_set)
+      end
+
+      it 'count' do
+        expect(@brazil_client.comments.count).to eq(1)
+        _cmt = @brazil_client.comments.create(:name => 'Builded Comment')
+        expect(@brazil_client.comments.count).to eq(2)
+      end
+
+      it 'group + count' do
+        expect(@brazil_client.comments.group(:id).count.length).to eq(1)
+        _cmt = @brazil_client.comments.create(:name => 'Builded Comment')
+        expect(@brazil_client.comments.group(:id).count.length).to eq(2)
+      end
+
+      it 'size' do
+        expect(@brazil_client.comments.size).to eq(1)
+        _cmt = @brazil_client.comments.create(:name => 'Builded Comment')
+        expect(@brazil_client.comments.size).to eq(2)
+      end
+
+      it 'create!' do
+        comment = @brazil_client.comments.create!(:name => 'Builded Comment')
+        expect(@brazil_client.comments.to_set).to eq([@comment_brazil, comment].to_set)
+      end
+
+      it 'length' do
+        expect(@brazil_client.comments.length).to eq(1)
+        _cmt = @brazil_client.comments.create(:name => 'Builded Comment')
+        expect(@brazil_client.comments.length).to eq(2)
+      end
+
+      it 'empty?' do
+        expect(@brazil_client.comments.empty?).to be false
+        c = Client.create!(:name => 'Client1')
+        expect(c.comments.empty?).to be true
+      end
+
+      it 'delete' do
+        expect(@brazil_client.comments.empty?).to be false
+        @brazil_client.comments.delete(@comment_brazil)
+        @brazil_client.reload
+        @comment_brazil.reload
+        expect(@comment_brazil.commentable).to be_nil
+        expect(@brazil_client.comments).to eq([])
+        expect(@brazil_client.comments.empty?).to be true
+      end
+
+      it 'delete_all' do
+        expect(@brazil_client.comments.empty?).to be false
+        @brazil_client.comments.delete_all
+        expect(@brazil_client.comments.empty?).to be true
+      end
+
+      it 'destroy_all' do
+        expect(@brazil_client.comments.empty?).to be false
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.empty?).to be true
+      end
+
+      it 'find' do
+        expect(@brazil_client.comments.first).to eq(@comment_brazil)
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.first).to be_nil
+      end
+
+      it 'where' do
+        expect(@brazil_client.comments.where('1=1')).to eq([@comment_brazil])
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1')).to be_empty
+      end
+
+      it 'map' do
+        expect(@brazil_client.comments.map(&:id)).to eq([@comment_brazil.id])
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.map(&:id)).to be_empty
+      end
+
+      it 'where + map' do
+        expect(@brazil_client.comments.where('1=1').map(&:id)).to eq([@comment_brazil.id])
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').map(&:id)).to be_empty
+      end
+
+      it 'where + each_with_index + map (enum method chain)' do
+        expect(@brazil_client.comments.where('1=1').each_with_index.map { |r, i| [r.id, i]}).to eq([[@comment_brazil.id, 0]])
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').each_with_index.map { |r, i| [r.id, i]}).to be_empty
+      end
+
+      it 'where + sum' do
+        expect(@brazil_client.comments.where('1=1').sum(&:id)).to eq(@comment_brazil.id)
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').sum(&:id)).to eq(0)
+      end
+
+      it 'where + index_by' do
+        expect(@brazil_client.comments.where('1=1').index_by(&:id)).to eq(@comment_brazil.id => @comment_brazil)
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').index_by(&:id)).to be_empty
+      end
+
+      it 'where + find' do
+        expect(@brazil_client.comments.where('1=1').find(@comment_brazil.id)).to eq(@comment_brazil)
+        @brazil_client.comments.destroy_all
+        expect { @brazil_client.comments.where('1=1').find(@comment_brazil.id) }.to raise_error ActiveRecord::RecordNotFound
+      end
+
+      it 'where + find with block' do
+        expect(@brazil_client.comments.where('1=1').find { |c| c.id == @comment_brazil.id }).to eq(@comment_brazil)
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').find { |c| c.id == @comment_brazil.id }).to be_nil
+      end
+
+      it 'where + select' do
+        expect(@brazil_client.comments.where('1=1').select(:name).first.name).to eq(@comment_brazil.name)
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').select(:name)).to be_empty
+      end
+
+      it 'where + select with block' do
+        expect(@brazil_client.comments.where('1=1').select { |c| c.id == @comment_brazil.id }).to eq([@comment_brazil])
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').select { |c| c.id == @comment_brazil.id }).to be_empty
+      end
+
+      it 'where + any?' do
+        expect(@brazil_client.comments.where('1=1').any?).to be true
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').any?).to be false
+      end
+
+      it 'where + any? with block' do
+        expect(@brazil_client.comments.where('1=1').any? { |c| c.id == @comment_brazil.id }).to be true
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.where('1=1').any? { |c| c.id == @comment_brazil.id }).to be false
+      end
+
+      it 'exists?' do
+        expect(@brazil_client.comments.exists?(@comment_brazil.id)).to be true
+        @brazil_client.comments.destroy_all
+        expect(@brazil_client.comments.exists?(@comment_brazil.id)).to be false
+      end
+
+      it 'uniq' do
+        expect(@brazil_client.comments.uniq).to eq([@comment_brazil])
+      end
+
+      it 'clear' do
+        expect(@brazil_client.comments.empty?).to be false
+        @brazil_client.comments.clear
+        expect(@brazil_client.comments.empty?).to be true
+      end
+    end
+  end
+
+  it 'block' do
+    @brazil_role = Role.using(:brazil).create!(:name => 'Brazil Role')
+    expect(@brazil_role.permissions.build(:name => 'ok').name).to eq('ok')
+    expect(@brazil_role.permissions.create(:name => 'ok').name).to eq('ok')
+    expect(@brazil_role.permissions.create!(:name => 'ok').name).to eq('ok')
+  end
+end