xbar 0.0.1 → 0.4.0

data/spec/xbar/association_spec.rb

@@ -0,0 +1,660 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe XBar::Association do
+  before(:all) do
+    set_xbar_env("default", "test")
+  end
+  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
+      @keyboard_master.computer.should == @computer_master
+      @keyboard_brazil.computer.should == @computer_brazil
+    end
+
+    it "should read correctly the relationed model" do
+      new_computer_brazil = Computer.using(:brazil).create!(:name => "New Computer Brazil")
+      new_computer_master = Computer.create!(:name => "New Computer Brazil")
+      @keyboard_brazil.computer = new_computer_brazil
+      @keyboard_brazil.save
+      @keyboard_brazil.reload
+      @keyboard_brazil.computer_id.should ==  new_computer_brazil.id
+      @keyboard_brazil.computer.should ==  new_computer_brazil
+      new_computer_brazil.save
+      new_computer_brazil.reload
+      new_computer_brazil.keyboard.should == @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
+      c.keyboard.should == k
+      k.computer_id.should == c.id
+      k.computer.should == 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
+      c.keyboard.should == k
+      k.computer_id.should == c.id
+      k.computer.should == 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
+      @brazil_role.permission_ids.should == [@permission_brazil.id]
+      @brazil_role.permissions.should == [@permission_brazil]
+    end
+
+    it "should finds the client that the item belongs" do
+      @permission_brazil.role_ids.should == [@brazil_role.id]
+      @permission_brazil.roles.should == [@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]
+      @permission_brazil.roles.should == [new_brazil_role]
+      @permission_brazil.save
+      @permission_brazil.reload
+      @permission_brazil.role_ids.should == [new_brazil_role.id]
+      @permission_brazil.roles.should == [new_brazil_role]
+    end
+
+    it "should works 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()
+      c.roles().should == [new_brazil_role]
+      new_brazil_role.permissions.should == [c]
+    end
+
+    describe "it should works 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_attributes" do
+        @permission_brazil_2.update_attributes(:role_ids => [@role.id])
+        @permission_brazil_2.roles.to_set.should == [@role].to_set
+      end
+
+      if !XBar.rails3?
+        it "update_attribute" do
+          @permission_brazil_2.update_attribute(:role_ids, [@role.id])
+          @permission_brazil_2.roles.to_set.should == [@role].to_set
+        end
+      end
+
+      it "<<" do
+        @permission_brazil_2.roles << @role
+        @role.save()
+        @permission_brazil_2.save()
+        @permission_brazil_2.reload
+        @permission_brazil_2.roles.to_set.should == [@role].to_set
+      end
+
+      it "build" do
+        role = @permission_brazil_2.roles.build(:name => "Builded Role")
+        @permission_brazil_2.save()
+        @permission_brazil_2.roles.to_set.should == [role].to_set
+      end
+
+      it "create" do
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.to_set.should == [role].to_set
+      end
+
+      it "create" do
+        role = @permission_brazil_2.roles.create!(:name => "Builded Role")
+        @permission_brazil_2.roles.to_set.should == [role].to_set
+      end
+
+      it "count" do
+        @permission_brazil_2.roles.count.should == 0
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.count.should == 1
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.count.should == 2
+      end
+
+      it "size" do
+        @permission_brazil_2.roles.size.should == 0
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.size.should == 1
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.size.should == 2
+      end
+
+      it "length" do
+        @permission_brazil_2.roles.length.should == 0
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.length.should == 1
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.length.should == 2
+      end
+
+
+      it "empty?" do
+        @permission_brazil_2.roles.empty?.should be_true
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.empty?.should be_false
+      end
+
+      it "delete_all" do
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.empty?.should be_false
+        @permission_brazil_2.roles.delete_all
+        @permission_brazil_2.roles.empty?.should be_true
+      end
+
+      it "destroy_all" do
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.empty?.should be_false
+        @permission_brazil_2.roles.destroy_all
+        @permission_brazil_2.roles.empty?.should be_true
+      end
+
+      it "find" do
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.find(:first).should == role
+        @permission_brazil_2.roles.destroy_all
+        @permission_brazil_2.roles.find(:first).should be_nil
+      end
+
+      it "exists?" do
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.exists?(role).should be_true
+        @permission_brazil_2.roles.destroy_all
+        @permission_brazil_2.roles.exists?(role).should be_false
+      end
+
+      it "clear" do
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.empty?.should be_false
+        @permission_brazil_2.roles.clear
+        @permission_brazil_2.roles.empty?.should be_true
+      end
+
+      it "delete" do
+        role = @permission_brazil_2.roles.create(:name => "Builded Role")
+        @permission_brazil_2.roles.empty?.should be_false
+        @permission_brazil_2.roles.delete(role)
+        @permission_brazil_2.reload
+        @role.reload
+        @role.permissions.should == []
+        @permission_brazil_2.roles.should == []
+      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
+      @programmer.project_ids().should == [ @project.id]
+      @programmer.projects().should == [@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]
+      @project.programmers.should == [new_brazil_programmer]
+      @project.save()
+      @project.reload
+      @project.programmer_ids.should == [new_brazil_programmer.id]
+      @project.programmers().should == [new_brazil_programmer]
+    end
+
+    it "should works 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()
+      c.programmers().should == [new_brazil_programmer]
+      new_brazil_programmer.projects.should == [c]
+    end
+
+    describe "it should works 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_attributes" do
+        @new_brazil_programmer.update_attributes(:project_ids => [@project.id])
+        @new_brazil_programmer.projects.to_set.should == [@project].to_set
+      end
+
+      if !XBar.rails3?
+        it "update_attribute" do
+          @new_brazil_programmer.update_attribute(:project_ids, [@project.id])
+          @new_brazil_programmer.projects.to_set.should == [@project].to_set
+        end
+      end
+
+      it "<<" do
+        @new_brazil_programmer.projects << @project
+        @project.save()
+        @new_brazil_programmer.save()
+        @new_brazil_programmer.reload
+        @new_brazil_programmer.projects.to_set.should == [@project].to_set
+      end
+
+      it "build" do
+        role = @new_brazil_programmer.projects.build(:name => "New VB App :-/")
+        @new_brazil_programmer.save()
+        @new_brazil_programmer.projects.to_set.should == [role].to_set
+      end
+
+      it "create" do
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.to_set.should == [role].to_set
+      end
+
+      it "create" do
+        role = @new_brazil_programmer.projects.create!(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.to_set.should == [role].to_set
+      end
+
+      it "count" do
+        @new_brazil_programmer.projects.count.should == 0
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.count.should == 1
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.count.should == 2
+      end
+
+      it "size" do
+        @new_brazil_programmer.projects.size.should == 0
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.size.should == 1
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.size.should == 2
+      end
+
+      it "length" do
+        @new_brazil_programmer.projects.length.should == 0
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.length.should == 1
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.length.should == 2
+      end
+
+
+      it "empty?" do
+        @new_brazil_programmer.projects.empty?.should be_true
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.empty?.should be_false
+      end
+
+      it "delete_all" do
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.empty?.should be_false
+        @new_brazil_programmer.projects.delete_all
+        @new_brazil_programmer.projects.empty?.should be_true
+      end
+
+      it "destroy_all" do
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.empty?.should be_false
+        @new_brazil_programmer.projects.destroy_all
+        @new_brazil_programmer.projects.empty?.should be_true
+      end
+
+      it "find" do
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.find(:first).should == role
+        @new_brazil_programmer.projects.destroy_all
+        @new_brazil_programmer.projects.find(:first).should be_nil
+      end
+
+      it "exists?" do
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.exists?(role).should be_true
+        @new_brazil_programmer.projects.destroy_all
+        @new_brazil_programmer.projects.exists?(role).should be_false
+      end
+
+      it "clear" do
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.empty?.should be_false
+        @new_brazil_programmer.projects.clear
+        @new_brazil_programmer.projects.empty?.should be_true
+      end
+
+      it "delete" do
+        role = @new_brazil_programmer.projects.create(:name => "New VB App :-/")
+        @new_brazil_programmer.projects.empty?.should be_false
+        @new_brazil_programmer.projects.delete(role)
+        @new_brazil_programmer.reload
+        @project.reload
+        @project.programmers.should == []
+        @new_brazil_programmer.projects.should == []
+      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
+      @brazil_client.item_ids.should == [@item_brazil.id]
+      @brazil_client.items().should == [@item_brazil]
+    end
+
+    it "should finds the client that the item belongs" do
+      @item_brazil.client.should == @brazil_client
+    end
+
+    it "should raise error if you try to add a record from a different shard" do
+      lambda do
+        @brazil_client.items << Item.using(:canada).create!(:name => "New User")
+      end.should raise_error("Association Error: Records are from different shards")
+    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
+      @item_brazil.client.should == new_brazil_client
+      @item_brazil.save()
+      @item_brazil.reload
+      @item_brazil.client_id.should == new_brazil_client.id
+      @item_brazil.client().should == new_brazil_client
+    end
+
+    it "should works for build method" do
+      item2 = Item.using(:brazil).create!(:name => "Brazil Item")
+      c = item2.create_client(:name => "new Client")
+      c.save()
+      item2.save()
+      item2.client.should == c
+      c.items().should == [item2]
+    end
+
+    describe "it should works when using" do
+      before(:each) do
+        @item_brazil_2 = Item.using(:brazil).create!(:name => "Brazil Item 2")
+        @brazil_client.items.to_set.should == [@item_brazil].to_set
+      end
+
+
+      it "update_attributes" do
+        @brazil_client.update_attributes(:item_ids => [@item_brazil_2.id, @item_brazil.id])
+        @brazil_client.items.to_set.should == [@item_brazil, @item_brazil_2].to_set
+      end
+
+      if !XBar.rails3?
+        it "update_attribute" do
+          @brazil_client.update_attribute(:item_ids, [@item_brazil_2.id, @item_brazil.id])
+          @brazil_client.items.to_set.should == [@item_brazil, @item_brazil_2].to_set
+        end
+      end
+
+      it "<<" do
+        @brazil_client.items << @item_brazil_2
+        @brazil_client.items.to_set.should == [@item_brazil, @item_brazil_2].to_set
+      end
+
+      it "build" do
+        item = @brazil_client.items.build(:name => "Builded Item")
+        item.save()
+        @brazil_client.items.to_set.should == [@item_brazil, item].to_set
+      end
+
+      it "create" do
+        item = @brazil_client.items.create(:name => "Builded Item")
+        @brazil_client.items.to_set.should == [@item_brazil, item].to_set
+      end
+
+      it "count" do
+        @brazil_client.items.count.should == 1
+        item = @brazil_client.items.create(:name => "Builded Item")
+        @brazil_client.items.count.should == 2
+      end
+
+      it "size" do
+        @brazil_client.items.size.should == 1
+        item = @brazil_client.items.create(:name => "Builded Item")
+        @brazil_client.items.size.should == 2
+      end
+
+      it "create!" do
+        item = @brazil_client.items.create!(:name => "Builded Item")
+        @brazil_client.items.to_set.should == [@item_brazil, item].to_set
+      end
+
+      it "length" do
+        @brazil_client.items.length.should == 1
+        item = @brazil_client.items.create(:name => "Builded Item")
+        @brazil_client.items.length.should == 2
+      end
+
+      it "empty?" do
+        @brazil_client.items.empty?.should be_false
+        c = Client.create!(:name => "Client1")
+        c.items.empty?.should be_true
+      end
+
+      it "delete" do
+        @brazil_client.items.empty?.should be_false
+        @brazil_client.items.delete(@item_brazil)
+        @brazil_client.reload
+        @item_brazil.reload
+        @item_brazil.client.should be_nil
+        @brazil_client.items.should == []
+        @brazil_client.items.empty?.should be_true
+      end
+
+      it "delete_all" do
+        @brazil_client.items.empty?.should be_false
+        @brazil_client.items.delete_all
+        @brazil_client.items.empty?.should be_true
+      end
+
+      it "destroy_all" do
+        @brazil_client.items.empty?.should be_false
+        @brazil_client.items.destroy_all
+        @brazil_client.items.empty?.should be_true
+      end
+
+      it "find" do
+        @brazil_client.items.find(:first).should == @item_brazil
+        @brazil_client.items.destroy_all
+        @brazil_client.items.find(:first).should be_nil
+      end
+
+      it "exists?" do
+        @brazil_client.items.exists?(@item_brazil).should be_true
+        @brazil_client.items.destroy_all
+        @brazil_client.items.exists?(@item_brazil).should be_false
+      end
+
+      it "uniq" do
+        @brazil_client.items.uniq.should == [@item_brazil]
+      end
+
+      it "clear" do
+        @brazil_client.items.empty?.should be_false
+        @brazil_client.items.clear
+        @brazil_client.items.empty?.should 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
+      @brazil_client.comment_ids.should == [@comment_brazil.id]
+      @brazil_client.comments().should == [@comment_brazil]
+    end
+
+    it "should finds the client that the comment belongs" do
+      @comment_brazil.commentable.should == @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
+      @comment_brazil.commentable.should == new_brazil_client
+      @comment_brazil.save()
+      @comment_brazil.reload
+      @comment_brazil.commentable_id.should == new_brazil_client.id
+      @comment_brazil.commentable().should == new_brazil_client
+    end
+
+    describe "it should works when using" do
+      before(:each) do
+        @comment_brazil_2 = Comment.using(:brazil).create!(:name => "Brazil Comment 2")
+        @brazil_client.comments.to_set.should == [@comment_brazil].to_set
+      end
+
+      it "update_attributes" do
+        @brazil_client.update_attributes(:comment_ids => [@comment_brazil_2.id, @comment_brazil.id])
+        @brazil_client.comments.to_set.should == [@comment_brazil, @comment_brazil_2].to_set
+      end
+
+      if !XBar.rails3?
+        it "update_attribute" do
+          @brazil_client.update_attribute(:comment_ids, [@comment_brazil_2.id, @comment_brazil.id])
+          @brazil_client.comments.to_set.should == [@comment_brazil, @comment_brazil_2].to_set
+        end
+      end
+
+      it "<<" do
+        @brazil_client.comments << @comment_brazil_2
+        @brazil_client.comments.to_set.should == [@comment_brazil, @comment_brazil_2].to_set
+      end
+
+      it "build" do
+        comment = @brazil_client.comments.build(:name => "Builded Comment")
+        comment.save()
+        @brazil_client.comments.to_set.should == [@comment_brazil, comment].to_set
+      end
+
+      it "create" do
+        comment = @brazil_client.comments.create(:name => "Builded Comment")
+        @brazil_client.comments.to_set.should == [@comment_brazil, comment].to_set
+      end
+
+      it "count" do
+        @brazil_client.comments.count.should == 1
+        comment = @brazil_client.comments.create(:name => "Builded Comment")
+        @brazil_client.comments.count.should == 2
+      end
+
+      it "size" do
+        @brazil_client.comments.size.should == 1
+        comment = @brazil_client.comments.create(:name => "Builded Comment")
+        @brazil_client.comments.size.should == 2
+      end
+
+      it "create!" do
+        comment = @brazil_client.comments.create!(:name => "Builded Comment")
+        @brazil_client.comments.to_set.should == [@comment_brazil, comment].to_set
+      end
+
+      it "length" do
+        @brazil_client.comments.length.should == 1
+        comment = @brazil_client.comments.create(:name => "Builded Comment")
+        @brazil_client.comments.length.should == 2
+      end
+
+      it "empty?" do
+        @brazil_client.comments.empty?.should be_false
+        c = Client.create!(:name => "Client1")
+        c.comments.empty?.should be_true
+      end
+
+      it "delete" do
+        @brazil_client.comments.empty?.should be_false
+        @brazil_client.comments.delete(@comment_brazil)
+        @brazil_client.reload
+        @comment_brazil.reload
+        @comment_brazil.commentable.should be_nil
+        @brazil_client.comments.should == []
+        @brazil_client.comments.empty?.should be_true
+      end
+
+      it "delete_all" do
+        @brazil_client.comments.empty?.should be_false
+        @brazil_client.comments.delete_all
+        @brazil_client.comments.empty?.should be_true
+      end
+
+      it "destroy_all" do
+        @brazil_client.comments.empty?.should be_false
+        @brazil_client.comments.destroy_all
+        @brazil_client.comments.empty?.should be_true
+      end
+
+      it "find" do
+        @brazil_client.comments.find(:first).should == @comment_brazil
+        @brazil_client.comments.destroy_all
+        @brazil_client.comments.find(:first).should be_nil
+      end
+
+      it "exists?" do
+        @brazil_client.comments.exists?(@comment_brazil).should be_true
+        @brazil_client.comments.destroy_all
+        @brazil_client.comments.exists?(@comment_brazil).should be_false
+      end
+
+      it "uniq" do
+        @brazil_client.comments.uniq.should == [@comment_brazil]
+      end
+
+      it "clear" do
+        @brazil_client.comments.empty?.should be_false
+        @brazil_client.comments.clear
+        @brazil_client.comments.empty?.should be_true
+      end
+    end
+  end
+end

data/spec/xbar/controller_spec.rb

@@ -0,0 +1,40 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe "Rails Controllers" do
+  before(:all) do
+    set_xbar_env('default', 'test')
+  end
+  it "should use #using method to in all requests" do
+    class UsersController < ActionController::Base
+      around_filter :select_shard
+      def create
+        User.create!(:name => "ActionController")
+        render :nothing => true
+      end
+
+      def select_shard(&block)
+        XBar.using(:brazil, &block)
+      end
+
+      def self._routes
+        ActionDispatch::Routing::RouteSet.new
+      end
+    end
+
+    UsersController.action_methods.include?("create").should be_true
+
+    if XBar.rails3?
+      a = UsersController.new
+      a.stub!(:request).and_return(
+        mock({:fullpath => "", :filtered_parameters => {}, :env => {},
+        :formats => [mock(:to_sym => :xml, :ref => "xml")],
+        :method => "GET", :format => nil, :content_mime_type => nil}))
+      a.instance_variable_set(:@_response, mock(:content_type => "xml", :body= => "", :status => 401))
+      a.process(:create)
+      User.using(:brazil).find_by_name("ActionController").should_not be_nil
+      User.using(:master).find_by_name("ActionController").should be_nil
+    else
+      pending()
+    end
+  end
+end