Added assert_select* for CSS selector-based testing (deprecates assert_tag)...

Added assert_select* for CSS selector-based testing (deprecates assert_tag) #5936 [assaf.arkin@gmail.com]

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@4929 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

actionpack/CHANGELOG

 *SVN*
 
+* Added assert_select* for CSS selector-based testing (deprecates assert_tag) #5936 [assaf.arkin@gmail.com]
+
 * radio_button_tag generates unique id attributes.  #3353 [Bob Silva, somekool@gmail.com]
 
 * strip_tags returns nil for a blank arg such as nil or "".  #2229 [duncan@whomwah.com]

actionpack/lib/action_controller/assert_tag.rb

+require 'test/unit'
+require 'test/unit/assertions'
+require 'rexml/document'
+require File.dirname(__FILE__) + "/vendor/html-scanner/html/document"
+
+module ActionController
+  module Assertions
+    module TagAssertions
+      # Asserts that there is a tag/node/element in the body of the response
+      # that meets all of the given conditions. The +conditions+ parameter must
+      # be a hash of any of the following keys (all are optional):
+      #
+      # * <tt>:tag</tt>: the node type must match the corresponding value
+      # * <tt>:attributes</tt>: a hash. The node's attributes must match the
+      #   corresponding values in the hash.
+      # * <tt>:parent</tt>: a hash. The node's parent must match the
+      #   corresponding hash.
+      # * <tt>:child</tt>: a hash. At least one of the node's immediate children
+      #   must meet the criteria described by the hash.
+      # * <tt>:ancestor</tt>: a hash. At least one of the node's ancestors must
+      #   meet the criteria described by the hash.
+      # * <tt>:descendant</tt>: a hash. At least one of the node's descendants
+      #   must meet the criteria described by the hash.
+      # * <tt>:sibling</tt>: a hash. At least one of the node's siblings must
+      #   meet the criteria described by the hash.
+      # * <tt>:after</tt>: a hash. The node must be after any sibling meeting
+      #   the criteria described by the hash, and at least one sibling must match.
+      # * <tt>:before</tt>: a hash. The node must be before any sibling meeting
+      #   the criteria described by the hash, and at least one sibling must match.
+      # * <tt>:children</tt>: a hash, for counting children of a node. Accepts
+      #   the keys:
+      #   * <tt>:count</tt>: either a number or a range which must equal (or
+      #     include) the number of children that match.
+      #   * <tt>:less_than</tt>: the number of matching children must be less
+      #     than this number.
+      #   * <tt>:greater_than</tt>: the number of matching children must be
+      #     greater than this number.
+      #   * <tt>:only</tt>: another hash consisting of the keys to use
+      #     to match on the children, and only matching children will be
+      #     counted.
+      # * <tt>:content</tt>: the textual content of the node must match the
+      #     given value. This will not match HTML tags in the body of a
+      #     tag--only text.
+      #
+      # Conditions are matched using the following algorithm:
+      #
+      # * if the condition is a string, it must be a substring of the value.
+      # * if the condition is a regexp, it must match the value.
+      # * if the condition is a number, the value must match number.to_s.
+      # * if the condition is +true+, the value must not be +nil+.
+      # * if the condition is +false+ or +nil+, the value must be +nil+.
+      #
+      # Usage:
+      #
+      #   # assert that there is a "span" tag
+      #   assert_tag :tag => "span"
+      #
+      #   # assert that there is a "span" tag with id="x"
+      #   assert_tag :tag => "span", :attributes => { :id => "x" }
+      #
+      #   # assert that there is a "span" tag using the short-hand
+      #   assert_tag :span
+      #
+      #   # assert that there is a "span" tag with id="x" using the short-hand
+      #   assert_tag :span, :attributes => { :id => "x" }
+      #
+      #   # assert that there is a "span" inside of a "div"
+      #   assert_tag :tag => "span", :parent => { :tag => "div" }
+      #
+      #   # assert that there is a "span" somewhere inside a table
+      #   assert_tag :tag => "span", :ancestor => { :tag => "table" }
+      #
+      #   # assert that there is a "span" with at least one "em" child
+      #   assert_tag :tag => "span", :child => { :tag => "em" }
+      #
+      #   # assert that there is a "span" containing a (possibly nested)
+      #   # "strong" tag.
+      #   assert_tag :tag => "span", :descendant => { :tag => "strong" }
+      #
+      #   # assert that there is a "span" containing between 2 and 4 "em" tags
+      #   # as immediate children
+      #   assert_tag :tag => "span",
+      #              :children => { :count => 2..4, :only => { :tag => "em" } } 
+      #
+      #   # get funky: assert that there is a "div", with an "ul" ancestor
+      #   # and an "li" parent (with "class" = "enum"), and containing a 
+      #   # "span" descendant that contains text matching /hello world/
+      #   assert_tag :tag => "div",
+      #              :ancestor => { :tag => "ul" },
+      #              :parent => { :tag => "li",
+      #                           :attributes => { :class => "enum" } },
+      #              :descendant => { :tag => "span",
+      #                               :child => /hello world/ }
+      #
+      # <strong>Please note</strong: #assert_tag and #assert_no_tag only work
+      # with well-formed XHTML. They recognize a few tags as implicitly self-closing
+      # (like br and hr and such) but will not work correctly with tags
+      # that allow optional closing tags (p, li, td). <em>You must explicitly
+      # close all of your tags to use these assertions.</em>
+      def assert_tag(*opts)
+        clean_backtrace do
+          opts = opts.size > 1 ? opts.last.merge({ :tag => opts.first.to_s }) : opts.first
+          tag = find_tag(opts)
+          assert tag, "expected tag, but no tag found matching #{opts.inspect} in:\n#{@response.body.inspect}"
+        end
+      end
+      
+      # Identical to #assert_tag, but asserts that a matching tag does _not_
+      # exist. (See #assert_tag for a full discussion of the syntax.)
+      def assert_no_tag(*opts)
+        clean_backtrace do
+          opts = opts.size > 1 ? opts.last.merge({ :tag => opts.first.to_s }) : opts.first
+          tag = find_tag(opts)
+          assert !tag, "expected no tag, but found tag matching #{opts.inspect} in:\n#{@response.body.inspect}"
+        end
+      end
+    end
+  end
+end
+
+Test::Unit::TestCase.send :include, ActionController::Assertions::TagAssertions
\ No newline at end of file

actionpack/lib/action_controller/assertions.rb

@@ -221,115 +221,6 @@ def assert_routing(path, options, defaults={}, extras={}, message=nil)
         assert_generates(path, options, defaults, extras, message)
       end
 
-      # Asserts that there is a tag/node/element in the body of the response
-      # that meets all of the given conditions. The +conditions+ parameter must
-      # be a hash of any of the following keys (all are optional):
-      #
-      # * <tt>:tag</tt>: the node type must match the corresponding value
-      # * <tt>:attributes</tt>: a hash. The node's attributes must match the
-      #   corresponding values in the hash.
-      # * <tt>:parent</tt>: a hash. The node's parent must match the
-      #   corresponding hash.
-      # * <tt>:child</tt>: a hash. At least one of the node's immediate children
-      #   must meet the criteria described by the hash.
-      # * <tt>:ancestor</tt>: a hash. At least one of the node's ancestors must
-      #   meet the criteria described by the hash.
-      # * <tt>:descendant</tt>: a hash. At least one of the node's descendants
-      #   must meet the criteria described by the hash.
-      # * <tt>:sibling</tt>: a hash. At least one of the node's siblings must
-      #   meet the criteria described by the hash.
-      # * <tt>:after</tt>: a hash. The node must be after any sibling meeting
-      #   the criteria described by the hash, and at least one sibling must match.
-      # * <tt>:before</tt>: a hash. The node must be before any sibling meeting
-      #   the criteria described by the hash, and at least one sibling must match.
-      # * <tt>:children</tt>: a hash, for counting children of a node. Accepts
-      #   the keys:
-      #   * <tt>:count</tt>: either a number or a range which must equal (or
-      #     include) the number of children that match.
-      #   * <tt>:less_than</tt>: the number of matching children must be less
-      #     than this number.
-      #   * <tt>:greater_than</tt>: the number of matching children must be
-      #     greater than this number.
-      #   * <tt>:only</tt>: another hash consisting of the keys to use
-      #     to match on the children, and only matching children will be
-      #     counted.
-      # * <tt>:content</tt>: the textual content of the node must match the
-      #     given value. This will not match HTML tags in the body of a
-      #     tag--only text.
-      #
-      # Conditions are matched using the following algorithm:
-      #
-      # * if the condition is a string, it must be a substring of the value.
-      # * if the condition is a regexp, it must match the value.
-      # * if the condition is a number, the value must match number.to_s.
-      # * if the condition is +true+, the value must not be +nil+.
-      # * if the condition is +false+ or +nil+, the value must be +nil+.
-      #
-      # Usage:
-      #
-      #   # assert that there is a "span" tag
-      #   assert_tag :tag => "span"
-      #
-      #   # assert that there is a "span" tag with id="x"
-      #   assert_tag :tag => "span", :attributes => { :id => "x" }
-      #
-      #   # assert that there is a "span" tag using the short-hand
-      #   assert_tag :span
-      #
-      #   # assert that there is a "span" tag with id="x" using the short-hand
-      #   assert_tag :span, :attributes => { :id => "x" }
-      #
-      #   # assert that there is a "span" inside of a "div"
-      #   assert_tag :tag => "span", :parent => { :tag => "div" }
-      #
-      #   # assert that there is a "span" somewhere inside a table
-      #   assert_tag :tag => "span", :ancestor => { :tag => "table" }
-      #
-      #   # assert that there is a "span" with at least one "em" child
-      #   assert_tag :tag => "span", :child => { :tag => "em" }
-      #
-      #   # assert that there is a "span" containing a (possibly nested)
-      #   # "strong" tag.
-      #   assert_tag :tag => "span", :descendant => { :tag => "strong" }
-      #
-      #   # assert that there is a "span" containing between 2 and 4 "em" tags
-      #   # as immediate children
-      #   assert_tag :tag => "span",
-      #              :children => { :count => 2..4, :only => { :tag => "em" } } 
-      #
-      #   # get funky: assert that there is a "div", with an "ul" ancestor
-      #   # and an "li" parent (with "class" = "enum"), and containing a 
-      #   # "span" descendant that contains text matching /hello world/
-      #   assert_tag :tag => "div",
-      #              :ancestor => { :tag => "ul" },
-      #              :parent => { :tag => "li",
-      #                           :attributes => { :class => "enum" } },
-      #              :descendant => { :tag => "span",
-      #                               :child => /hello world/ }
-      #
-      # <strong>Please note</strong: #assert_tag and #assert_no_tag only work
-      # with well-formed XHTML. They recognize a few tags as implicitly self-closing
-      # (like br and hr and such) but will not work correctly with tags
-      # that allow optional closing tags (p, li, td). <em>You must explicitly
-      # close all of your tags to use these assertions.</em>
-      def assert_tag(*opts)
-        clean_backtrace do
-          opts = opts.size > 1 ? opts.last.merge({ :tag => opts.first.to_s }) : opts.first
-          tag = find_tag(opts)
-          assert tag, "expected tag, but no tag found matching #{opts.inspect} in:\n#{@response.body.inspect}"
-        end
-      end
-      
-      # Identical to #assert_tag, but asserts that a matching tag does _not_
-      # exist. (See #assert_tag for a full discussion of the syntax.)
-      def assert_no_tag(*opts)
-        clean_backtrace do
-          opts = opts.size > 1 ? opts.last.merge({ :tag => opts.first.to_s }) : opts.first
-          tag = find_tag(opts)
-          assert !tag, "expected no tag, but found tag matching #{opts.inspect} in:\n#{@response.body.inspect}"
-        end
-      end
-
       # test 2 html strings to be equivalent, i.e. identical up to reordering of attributes
       def assert_dom_equal(expected, actual, message="")
         clean_backtrace do
@@ -382,4 +273,4 @@ def parameterize(value)
         end
     end
   end
-end
+end
\ No newline at end of file

actionpack/lib/action_controller/test_process.rb

 require File.dirname(__FILE__) + '/assertions'
+require File.dirname(__FILE__) + '/assert_select'
+require File.dirname(__FILE__) + '/assert_tag'
 require File.dirname(__FILE__) + '/deprecated_assertions'
 
 module ActionController #:nodoc:

actionpack/lib/action_controller/vendor/html-scanner/html/document.rb

 require File.dirname(__FILE__) + '/tokenizer'
 require File.dirname(__FILE__) + '/node'
+require File.dirname(__FILE__) + '/selector'
 
 module HTML #:nodoc:
   

actionpack/test/controller/assert_select_test.rb

+#--
+# Copyright (c) 2006 Assaf Arkin (http://labnotes.org)
+# Under MIT and/or CC By license.
+#++
+
+require File.dirname(__FILE__) + '/../abstract_unit'
+require File.dirname(__FILE__) + '/fake_controllers'
+require "action_mailer"
+
+class AssertSelectTest < Test::Unit::TestCase
+  class AssertSelectController < ActionController::Base
+    def response_with=(content)
+      @content = content
+    end
+
+    def response_with(&block)
+      @update = block
+    end
+
+    def html()
+      render :text=>@content, :layout=>false, :content_type=>Mime::HTML
+      @content = nil
+    end
+
+    def rjs()
+      render :update do |page|
+        @update.call page
+      end
+      @update = nil
+    end
+
+    def xml()
+      render :text=>@content, :layout=>false, :content_type=>Mime::XML
+      @content = nil
+    end
+
+    def rescue_action(e)
+      raise e
+    end
+  end
+
+  class AssertSelectMailer < ActionMailer::Base
+    def test(html)
+      recipients "test <test@test.host>"
+      from "test@test.host"
+      subject "Test e-mail"
+      part :content_type=>"text/html", :body=>html
+    end
+  end
+
+  AssertionFailedError = Test::Unit::AssertionFailedError
+
+  def setup
+    @controller = AssertSelectController.new
+    @request    = ActionController::TestRequest.new
+    @response   = ActionController::TestResponse.new
+    ActionMailer::Base.delivery_method = :test
+    ActionMailer::Base.perform_deliveries = true
+    ActionMailer::Base.deliveries = []
+  end
+
+
+  def teardown
+    ActionMailer::Base.deliveries.clear
+  end
+
+
+  #
+  # Test assert select.
+  #
+
+  def test_assert_select
+    render_html %Q{<div id="1"></div><div id="2"></div>}
+    assert_select "div", 2
+    assert_raises(AssertionFailedError) { assert_select "div", 3 }
+    assert_raises(AssertionFailedError){ assert_select "p" }
+  end
+
+
+  def test_equality_true_false
+    render_html %Q{<div id="1"></div><div id="2"></div>}
+    assert_nothing_raised               { assert_select "div" }
+    assert_raises(AssertionFailedError) { assert_select "p" }
+    assert_nothing_raised               { assert_select "div", true }
+    assert_raises(AssertionFailedError) { assert_select "p", true }
+    assert_raises(AssertionFailedError) { assert_select "div", false }
+    assert_nothing_raised               { assert_select "p", false }
+  end
+
+
+  def test_equality_string_and_regexp
+    render_html %Q{<div id="1">foo</div><div id="2">foo</div>}
+    assert_nothing_raised               { assert_select "div", "foo" }
+    assert_raises(AssertionFailedError) { assert_select "div", "bar" }
+    assert_nothing_raised               { assert_select "div", :text=>"foo" }
+    assert_raises(AssertionFailedError) { assert_select "div", :text=>"bar" }
+    assert_nothing_raised               { assert_select "div", /(foo|bar)/ }
+    assert_raises(AssertionFailedError) { assert_select "div", /foobar/ }
+    assert_nothing_raised               { assert_select "div", :text=>/(foo|bar)/ }
+    assert_raises(AssertionFailedError) { assert_select "div", :text=>/foobar/ }
+    assert_raises(AssertionFailedError) { assert_select "p", :text=>/foobar/ }
+  end
+
+
+  def test_equality_of_html
+    render_html %Q{<p>\n<em>"This is <strong>not</strong> a big problem,"</em> he said.\n</p>}
+    text = "\"This is not a big problem,\" he said."
+    html = "<em>\"This is <strong>not</strong> a big problem,\"</em> he said."
+    assert_nothing_raised               { assert_select "p", text }
+    assert_raises(AssertionFailedError) { assert_select "p", html }
+    assert_nothing_raised               { assert_select "p", :html=>html }
+    assert_raises(AssertionFailedError) { assert_select "p", :html=>text }
+    # No stripping for pre.
+    render_html %Q{<pre>\n<em>"This is <strong>not</strong> a big problem,"</em> he said.\n</pre>}
+    text = "\n\"This is not a big problem,\" he said.\n"
+    html = "\n<em>\"This is <strong>not</strong> a big problem,\"</em> he said.\n"
+    assert_nothing_raised               { assert_select "pre", text }
+    assert_raises(AssertionFailedError) { assert_select "pre", html }
+    assert_nothing_raised               { assert_select "pre", :html=>html }
+    assert_raises(AssertionFailedError) { assert_select "pre", :html=>text }
+  end
+
+
+  def test_equality_of_instances
+    render_html %Q{<div id="1">foo</div><div id="2">foo</div>}
+    assert_nothing_raised               { assert_select "div", 2 }
+    assert_raises(AssertionFailedError) { assert_select "div", 3 }
+    assert_nothing_raised               { assert_select "div", 1..2 }
+    assert_raises(AssertionFailedError) { assert_select "div", 3..4 }
+    assert_nothing_raised               { assert_select "div", :count=>2 }
+    assert_raises(AssertionFailedError) { assert_select "div", :count=>3 }
+    assert_nothing_raised               { assert_select "div", :minimum=>1 }
+    assert_nothing_raised               { assert_select "div", :minimum=>2 }
+    assert_raises(AssertionFailedError) { assert_select "div", :minimum=>3 }
+    assert_nothing_raised               { assert_select "div", :maximum=>2 }
+    assert_nothing_raised               { assert_select "div", :maximum=>3 }
+    assert_raises(AssertionFailedError) { assert_select "div", :maximum=>1 }
+    assert_nothing_raised               { assert_select "div", :minimum=>1, :maximum=>2 }
+    assert_raises(AssertionFailedError) { assert_select "div", :minimum=>3, :maximum=>4 }
+  end
+
+
+  def test_substitution_values
+    render_html %Q{<div id="1">foo</div><div id="2">foo</div>}
+    assert_select "div#?", /\d+/ do |elements|
+      assert_equal 2, elements.size
+    end
+    assert_select "div" do
+      assert_select "div#?", /\d+/ do |elements|
+        assert_equal 2, elements.size
+        assert_select "#1"
+        assert_select "#2"
+      end
+    end
+  end
+
+  
+  def test_nested_assert_select
+    render_html %Q{<div id="1">foo</div><div id="2">foo</div>}
+    assert_select "div" do |elements|
+      assert_equal 2, elements.size
+      assert_select elements[0], "#1"
+      assert_select elements[1], "#2"
+    end
+    assert_select "div" do
+      assert_select "div" do |elements|
+        assert_equal 2, elements.size
+        # Testing in a group is one thing
+        assert_select "#1,#2"
+        # Testing individually is another.
+        assert_select "#1"
+        assert_select "#2"
+        assert_select "#3", false
+      end
+    end
+  end
+
+
+  def test_assert_select_from_rjs
+    render_rjs do |page|
+      page.replace_html "test", "<div id=\"1\">foo</div>\n<div id=\"2\">foo</div>"
+    end
+    assert_select "div" do |elements|
+      assert elements.size == 2
+      assert_select "#1"
+      assert_select "#2"
+    end
+    assert_select "div#?", /\d+/ do |elements|
+      assert_select "#1"
+      assert_select "#2"
+    end
+    # With multiple results.
+    render_rjs do |page|
+      page.replace_html "test", "<div id=\"1\">foo</div>"
+      page.replace_html "test2", "<div id=\"2\">foo</div>"
+    end
+    assert_select "div" do |elements|
+      assert elements.size == 2
+      assert_select "#1"
+      assert_select "#2"
+    end
+  end
+
+
+  #
+  # Test css_select.
+  #
+
+
+  def test_css_select
+    render_html %Q{<div id="1"></div><div id="2"></div>}
+    assert 2, css_select("div").size
+    assert 0, css_select("p").size
+  end
+
+
+  def test_nested_css_select
+    render_html %Q{<div id="1">foo</div><div id="2">foo</div>}
+    assert_select "div#?", /\d+/ do |elements|
+      assert_equal 1, css_select(elements[0], "div").size
+      assert_equal 1, css_select(elements[1], "div").size
+    end
+    assert_select "div" do
+      assert_equal 2, css_select("div").size
+      css_select("div").each do |element|
+        # Testing as a group is one thing
+        assert !css_select("#1,#2").empty?
+        # Testing individually is another
+        assert !css_select("#1").empty?
+        assert !css_select("#2").empty?
+      end
+    end
+  end
+
+
+  def test_css_select_from_rjs
+    # With one result.
+    render_rjs do |page|
+      page.replace_html "test", "<div id=\"1\">foo</div>\n<div id=\"2\">foo</div>"
+    end
+    assert_equal 2, css_select("div").size
+    assert_equal 1, css_select("#1").size
+    assert_equal 1, css_select("#2").size
+    # With multiple results.
+    render_rjs do |page|
+      page.replace_html "test", "<div id=\"1\">foo</div>"
+      page.replace_html "test2", "<div id=\"2\">foo</div>"
+    end
+    assert_equal 2, css_select("div").size
+    assert_equal 1, css_select("#1").size
+    assert_equal 1, css_select("#2").size
+  end
+
+
+  #
+  # Test assert_select_rjs.
+  #
+
+
+  def test_assert_select_rjs
+    # Test that we can pick up all statements in the result.
+    render_rjs do |page|
+      page.replace "test", "<div id=\"1\">foo</div>"
+      page.replace_html "test2", "<div id=\"2\">foo</div>"
+      page.insert_html :top, "test3", "<div id=\"3\">foo</div>"
+    end
+    found = false
+    assert_select_rjs do
+      assert_select "#1"
+      assert_select "#2"
+      assert_select "#3"
+      found = true
+    end
+    assert found
+    # Test that we fail if there is nothing to pick.
+    render_rjs do |page|
+    end
+    assert_raises(AssertionFailedError) { assert_select_rjs }
+  end
+
+
+  def test_assert_select_rjs_with_id
+    # Test that we can pick up all statements in the result.
+    render_rjs do |page|
+      page.replace "test1", "<div id=\"1\">foo</div>"
+      page.replace_html "test2", "<div id=\"2\">foo</div>"
+      page.insert_html :top, "test3", "<div id=\"3\">foo</div>"
+    end
+    assert_select_rjs "test1" do
+      assert_select "div", 1
+      assert_select "#1"
+    end
+    assert_select_rjs "test2" do
+      assert_select "div", 1
+      assert_select "#2"
+    end
+    assert_select_rjs "test3" do
+      assert_select "div", 1
+      assert_select "#3"
+    end
+    assert_raises(AssertionFailedError) { assert_select_rjs "test4" }
+  end
+
+
+  def test_assert_select_rjs_for_replace
+    render_rjs do |page|
+      page.replace "test1", "<div id=\"1\">foo</div>"
+      page.replace_html "test2", "<div id=\"2\">foo</div>"
+      page.insert_html :top, "test3", "<div id=\"3\">foo</div>"
+    end
+    # Replace.
+    assert_select_rjs :replace do
+      assert_select "div", 1
+      assert_select "#1"
+    end
+    assert_select_rjs :replace, "test1" do
+      assert_select "div", 1
+      assert_select "#1"
+    end
+    assert_raises(AssertionFailedError) { assert_select_rjs :replace, "test2" }
+    # Replace HTML.
+    assert_select_rjs :replace_html do
+      assert_select "div", 1
+      assert_select "#2"
+    end
+    assert_select_rjs :replace_html, "test2" do
+      assert_select "div", 1
+      assert_select "#2"
+    end
+    assert_raises(AssertionFailedError) { assert_select_rjs :replace_html, "test1" }
+  end
+
+
+  def test_assert_select_rjs_for_insert
+    render_rjs do |page|
+      page.replace "test1", "<div id=\"1\">foo</div>"
+      page.replace_html "test2", "<div id=\"2\">foo</div>"
+      page.insert_html :top, "test3", "<div id=\"3\">foo</div>"
+    end
+    # Non-positioned.
+    assert_select_rjs :insert_html do
+      assert_select "div", 1
+      assert_select "#3"
+    end
+    assert_select_rjs :insert_html, "test3" do
+      assert_select "div", 1
+      assert_select "#3"
+    end
+    assert_raises(AssertionFailedError) { assert_select_rjs :insert_html, "test1" }
+    # Positioned.
+    render_rjs do |page|
+      page.insert_html :top, "test1", "<div id=\"1\">foo</div>"
+      page.insert_html :bottom, "test2", "<div id=\"2\">foo</div>"
+      page.insert_html :before, "test3", "<div id=\"3\">foo</div>"
+      page.insert_html :after, "test4", "<div id=\"4\">foo</div>"
+    end
+    assert_select_rjs :insert, :top do
+      assert_select "div", 1
+      assert_select "#1"
+    end
+    assert_select_rjs :insert, :bottom do
+      assert_select "div", 1
+      assert_select "#2"
+    end
+    assert_select_rjs :insert, :before do
+      assert_select "div", 1
+      assert_select "#3"
+    end
+    assert_select_rjs :insert, :after do
+      assert_select "div", 1
+      assert_select "#4"
+    end
+    assert_select_rjs :insert_html do
+      assert_select "div", 4
+    end
+  end
+
+
+  def test_nested_assert_select_rjs
+    # Simple selection from a single result.
+    render_rjs do |page|
+      page.replace_html "test", "<div id=\"1\">foo</div>\n<div id=\"2\">foo</div>"
+    end
+    assert_select_rjs "test" do |elements|
+      assert_equal 2, elements.size
+      assert_select "#1"
+      assert_select "#2"
+    end
+    # Deal with two results.
+    render_rjs do |page|
+      page.replace_html "test", "<div id=\"1\">foo</div>"
+      page.replace_html "test2", "<div id=\"2\">foo</div>"
+    end
+    assert_select_rjs "test" do |elements|
+      assert_equal 1, elements.size
+      assert_select "#1"
+    end
+    assert_select_rjs "test2" do |elements|
+      assert_equal 1, elements.size
+      assert_select "#2"
+    end
+  end
+
+
+  def test_feed_item_encoded
+    render_xml <<-EOF
+<rss version="2.0">
+  <channel>
+    <item>
+      <description>
+        <![CDATA[
+          <p>Test 1</p>
+        ]]>
+      </description>
+    </item>
+    <item>
+      <description>
+        <![CDATA[
+          <p>Test 2</p>
+        ]]>
+      </description>
+    </item>
+  </channel>
+</rss>
+EOF
+    assert_select "channel item description" do
+      # Test element regardless of wrapper.
+      assert_select_encoded do
+        assert_select "p", :count=>2, :text=>/Test/
+      end
+      # Test through encoded wrapper.
+      assert_select_encoded do
+        assert_select "encoded p", :count=>2, :text=>/Test/
+      end
+      # Use :root instead (recommended)
+      assert_select_encoded do
+        assert_select ":root p", :count=>2, :text=>/Test/
+      end
+      # Test individually.
+      assert_select "description" do |elements|
+        assert_select_encoded elements[0] do
+          assert_select "p", "Test 1"
+        end
+        assert_select_encoded elements[1] do
+          assert_select "p", "Test 2"
+        end
+      end
+    end
+
+    # Test that we only un-encode element itself.
+    assert_select "channel item" do
+      assert_select_encoded do
+        assert_select "p", 0
+      end
+    end
+  end
+
+
+  #
+  # Test assert_select_email
+  #
+
+  def test_assert_select_email
+    assert_raises(AssertionFailedError) { assert_select_email {} }
+    AssertSelectMailer.deliver_test "<div><p>foo</p><p>bar</p></div>"
+    assert_select_email do
+      assert_select "div:root" do
+        assert_select "p:first-child", "foo"
+        assert_select "p:last-child", "bar"
+      end
+    end
+  end
+
+
+  protected
+    def render_html(html)
+      @controller.response_with = html
+      get :html
+    end
+
+    def render_rjs(&block)
+      @controller.response_with &block
+      get :rjs
+    end
+
+    def render_xml(xml)
+      @controller.response_with = xml
+      get :xml
+    end
+end
\ No newline at end of file