0
@@ -336,25 +336,31 @@ module ThoughtBot # :nodoc:
0
           should_set_the_flash_to nil
0
         end
0
         
0
-        # Macro that creates a test asserting that the controller assigned to @name
0
+        # Macro that creates a test asserting that the controller assigned to
0
+        # each of the named instance variable(s).
0
         #
0
         # Example:
0
         #
0
-        #   should_assign_to :user
0
-        def should_assign_to(name)
0
-          should "assign @#{name}" do
0
-            assert assigns(name.to_sym), "The action isn't assigning to @#{name}"
0
+        #   should_assign_to :user, :posts
0
+        def should_assign_to(*names)
0
+          names.each do |name|
0
+            should "assign @#{name}" do
0
+              assert assigns(name.to_sym), "The action isn't assigning to @#{name}"
0
+            end
0
           end
0
         end
0
 
0
-        # Macro that creates a test asserting that the controller did not assign to @name
0
+        # Macro that creates a test asserting that the controller did not assign to
0
+        # any of the named instance variable(s).
0
         #
0
         # Example:
0
         #
0
-        #   should_not_assign_to :user
0
-        def should_not_assign_to(name)
0
-          should "not assign to @#{name}" do
0
-            assert !assigns(name.to_sym), "@#{name} was visible"
0
+        #   should_not_assign_to :user, :posts
0
+        def should_not_assign_to(*names)
0
+          names.each do |name|
0
+            should "not assign to @#{name}" do
0
+              assert !assigns(name.to_sym), "@#{name} was visible"
0
+            end
0
           end
0
         end
0
 

0
@@ -39,5 +39,15 @@ class PostsControllerTest < Test::Unit::TestCase
0
       resource.create.params = { :title => "first post", :body => 'blah blah blah'}
0
       resource.update.params = { :title => "changed" }
0
     end
0
+
0
+    context "viewing posts for a user" do
0
+      setup do
0
+        get :index, :user_id => users(:first)
0
+      end
0
+      should_respond_with :success
0
+      should_assign_to :user, :posts
0
+      should_not_assign_to :foo, :bar
0
+    end
0
   end
0
+
0
 end