0
@@ -99,13 +99,21 @@ module ThoughtBot # :nodoc:
0
         # Macro that creates a test asserting that the controller assigned to
0
         # each of the named instance variable(s).
0
         #
0
+        # Options:
0
+        # * <tt>:class</tt> - The expected class of the instance variable being checked.
0
+        #
0
         # Example:
0
         #
0
         #   should_assign_to :user, :posts
0
+        #   should_assign_to :user, :class => User
0
         def should_assign_to(*names)
0
+          opts = names.extract_options!
0
           names.each do |name|
0
             should "assign @#{name}" do
0
               assert assigns(name.to_sym), "The action isn't assigning to @#{name}"
0
+              unless opts[:class].nil?
0
+                assert_kind_of opts[:class], assigns(name.to_sym)
0
+              end
0
             end
0
           end
0
         end

0
@@ -64,7 +64,11 @@ class PostsControllerTest < Test::Unit::TestCase
0
         get :index, :user_id => users(:first)
0
       end
0
       should_respond_with :success
0
-      should_assign_to :user, :posts
0
+      should_assign_to :user, :class => User
0
+      should_fail do
0
+        should_assign_to :user, :class => Post
0
+      end
0
+      should_assign_to :posts
0
       should_not_assign_to :foo, :bar
0
     end
0