Fixed the assert_select on the highlghted header macro

thoughtbot · Feb 5, 2009 · 7eee6f2 · 7eee6f2
1 parent 430528d
commit 7eee6f2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/shoulda_macros/sortable_table.rb b/shoulda_macros/sortable_table.rb
@@ -56,7 +56,9 @@ def should_display_sortable_table_header_for(*valid_sorts)
 
     def should_display_highlighted_sortable_table_header_for column, order = "ascending"
       should "have the #{order} class on the #{column} table header" do
-        assert_select 'a[href*=?][class*=?]', "sort=#{column}", order
+        assert_select('th[class=?]', order) do |tr|
+          assert_select 'a[href*=?]', "sort=#{column}"
+        end
       end
     end