@@ -336,25 +336,31 @@ module ThoughtBot # :nodoc: should_set_the_flash_to nil end - # Macro that creates a test asserting that the controller assigned to @name + # Macro that creates a test asserting that the controller assigned to + # each of the named instance variable(s). # # Example: # - # should_assign_to :user - def should_assign_to(name) - should "assign @#{name}" do - assert assigns(name.to_sym), "The action isn't assigning to @#{name}" + # should_assign_to :user, :posts + def should_assign_to(*names) + names.each do |name| + should "assign @#{name}" do + assert assigns(name.to_sym), "The action isn't assigning to @#{name}" + end end end - # Macro that creates a test asserting that the controller did not assign to @name + # Macro that creates a test asserting that the controller did not assign to + # any of the named instance variable(s). # # Example: # - # should_not_assign_to :user - def should_not_assign_to(name) - should "not assign to @#{name}" do - assert !assigns(name.to_sym), "@#{name} was visible" + # should_not_assign_to :user, :posts + def should_not_assign_to(*names) + names.each do |name| + should "not assign to @#{name}" do + assert !assigns(name.to_sym), "@#{name} was visible" + end end end lib/shoulda/controller_tests/controller_tests.rb @@ -39,5 +39,15 @@ class PostsControllerTest < Test::Unit::TestCase resource.create.params = { :title => "first post", :body => 'blah blah blah'} resource.update.params = { :title => "changed" } end + + context "viewing posts for a user" do + setup do + get :index, :user_id => users(:first) + end + should_respond_with :success + should_assign_to :user, :posts + should_not_assign_to :foo, :bar + end end + end test/functional/posts_controller_test.rb 0de2794be2820ea22824aa6a6c3e4a940ab8f693 Bob Showalter showaltb@gmail.com http://github.com/showaltb/shoulda/commit/cefcfe1205f3636615e0562bba0ff60d3bf02922 cefcfe1205f3636615e0562bba0ff60d3bf02922 2008-07-31T08:26:27-07:00 2008-07-31T08:26:27-07:00 Change should_assign_to and should_not_assign_to to allow multiple names (#22) 3ad87dc0f2392f87c23b39ce2f0b2af8d93fa1a8 Bob Showalter showaltb@gmail.com