2007-02-10 00:12:31 +01:00
#--
# Copyright (c) 2006 Assaf Arkin (http://labnotes.org)
# Under MIT and/or CC By license.
#++
module ActionController
module Assertions
unless const_defined? ( :NO_STRIP )
NO_STRIP = %w{ pre script style textarea }
end
2008-06-02 08:35:38 +02:00
# Adds the +assert_select+ method for use in Rails functional
2007-12-21 08:48:59 +01:00
# test cases, which can be used to make assertions on the response HTML of a controller
2008-06-02 08:35:38 +02:00
# action. You can also call +assert_select+ within another +assert_select+ to
2007-02-10 00:12:31 +01:00
# make assertions on elements selected by the enclosing assertion.
#
2008-06-02 08:35:38 +02:00
# Use +css_select+ to select elements without making an assertions, either
2007-02-10 00:12:31 +01:00
# from the response HTML or elements selected by the enclosing assertion.
2009-12-01 02:38:34 +01:00
#
2007-02-10 00:12:31 +01:00
# In addition to HTML responses, you can make the following assertions:
2008-10-27 07:47:01 +01:00
# * +assert_select_rjs+ - Assertions on HTML content of RJS update and insertion operations.
# * +assert_select_encoded+ - Assertions on HTML encoded inside XML, for example for dealing with feed item descriptions.
2008-05-18 06:22:34 +02:00
# * +assert_select_email+ - Assertions on the HTML body of an e-mail.
2007-02-10 00:12:31 +01:00
#
2007-12-21 08:48:59 +01:00
# Also see HTML::Selector to learn how to use selectors.
2007-02-10 00:12:31 +01:00
module SelectorAssertions
2009-12-01 02:38:34 +01:00
def initialize ( * args )
super
@selected = nil
end
2007-02-10 00:12:31 +01:00
# :call-seq:
# css_select(selector) => array
# css_select(element, selector) => array
#
# Select and return all matching elements.
#
# If called with a single argument, uses that argument as a selector
# to match all elements of the current page. Returns an empty array
# if no match is found.
#
# If called with two arguments, uses the first argument as the base
# element and the second argument as the selector. Attempts to match the
# base element and any of its children. Returns an empty array if no
# match is found.
#
2008-06-02 08:35:38 +02:00
# The selector may be a CSS selector expression (String), an expression
# with substitution values (Array) or an HTML::Selector object.
2007-02-10 00:12:31 +01:00
#
2007-12-21 08:48:59 +01:00
# ==== Examples
# # Selects all div tags
# divs = css_select("div")
#
# # Selects all paragraph tags and does something interesting
# pars = css_select("p")
# pars.each do |par|
# # Do something fun with paragraphs here...
# end
#
# # Selects all list items in unordered lists
2009-12-01 02:38:34 +01:00
# items = css_select("ul>li")
#
2007-12-21 08:48:59 +01:00
# # Selects all form tags and then all inputs inside the form
2007-02-10 00:12:31 +01:00
# forms = css_select("form")
# forms.each do |form|
# inputs = css_select(form, "input")
# ...
# end
2007-12-21 08:48:59 +01:00
#
2007-02-10 00:12:31 +01:00
def css_select ( * args )
# See assert_select to understand what's going on here.
arg = args . shift
if arg . is_a? ( HTML :: Node )
root = arg
arg = args . shift
elsif arg == nil
raise ArgumentError , " First argument is either selector or element to select, but nil found. Perhaps you called assert_select with an element that does not exist? "
elsif @selected
matches = [ ]
2007-12-21 08:48:59 +01:00
2007-02-10 00:12:31 +01:00
@selected . each do | selected |
subset = css_select ( selected , HTML :: Selector . new ( arg . dup , args . dup ) )
subset . each do | match |
matches << match unless matches . any? { | m | m . equal? ( match ) }
end
end
return matches
else
root = response_from_page_or_rjs
end
case arg
when String
selector = HTML :: Selector . new ( arg , args )
when Array
selector = HTML :: Selector . new ( * arg )
when HTML :: Selector
selector = arg
else raise ArgumentError , " Expecting a selector as the first argument "
end
selector . select ( root )
end
# :call-seq:
# assert_select(selector, equality?, message?)
# assert_select(element, selector, equality?, message?)
#
# An assertion that selects elements and makes one or more equality tests.
#
# If the first argument is an element, selects all matching elements
# starting from (and including) that element and all its children in
# depth-first order.
#
2009-02-04 21:26:08 +01:00
# If no element if specified, calling +assert_select+ selects from the
# response HTML unless +assert_select+ is called from within an +assert_select+ block.
#
# When called with a block +assert_select+ passes an array of selected elements
# to the block. Calling +assert_select+ from the block, with no element specified,
# runs the assertion on the complete set of elements selected by the enclosing assertion.
# Alternatively the array may be iterated through so that +assert_select+ can be called
# separately for each element.
#
2007-02-10 00:12:31 +01:00
#
2007-12-21 08:48:59 +01:00
# ==== Example
2009-02-04 21:26:08 +01:00
# If the response contains two ordered lists, each with four list elements then:
# assert_select "ol" do |elements|
2007-02-10 00:12:31 +01:00
# elements.each do |element|
2009-02-04 21:26:08 +01:00
# assert_select element, "li", 4
2007-02-10 00:12:31 +01:00
# end
# end
2007-12-21 08:48:59 +01:00
#
2009-02-04 21:26:08 +01:00
# will pass, as will:
# assert_select "ol" do
# assert_select "li", 8
2007-02-10 00:12:31 +01:00
# end
#
2008-06-02 08:35:38 +02:00
# The selector may be a CSS selector expression (String), an expression
2007-02-10 00:12:31 +01:00
# with substitution values, or an HTML::Selector object.
#
# === Equality Tests
#
# The equality test may be one of the following:
2008-05-18 06:22:34 +02:00
# * <tt>true</tt> - Assertion is true if at least one element selected.
# * <tt>false</tt> - Assertion is true if no element selected.
# * <tt>String/Regexp</tt> - Assertion is true if the text value of at least
2007-02-10 00:12:31 +01:00
# one element matches the string or regular expression.
2008-05-18 06:22:34 +02:00
# * <tt>Integer</tt> - Assertion is true if exactly that number of
2007-02-10 00:12:31 +01:00
# elements are selected.
2008-05-18 06:22:34 +02:00
# * <tt>Range</tt> - Assertion is true if the number of selected
2007-02-10 00:12:31 +01:00
# elements fit the range.
# If no equality test specified, the assertion is true if at least one
# element selected.
#
# To perform more than one equality tests, use a hash with the following keys:
2008-05-18 06:22:34 +02:00
# * <tt>:text</tt> - Narrow the selection to elements that have this text
2007-02-10 00:12:31 +01:00
# value (string or regexp).
2008-05-18 06:22:34 +02:00
# * <tt>:html</tt> - Narrow the selection to elements that have this HTML
2007-02-10 00:12:31 +01:00
# content (string or regexp).
2008-05-18 06:22:34 +02:00
# * <tt>:count</tt> - Assertion is true if the number of selected elements
2007-02-10 00:12:31 +01:00
# is equal to this value.
2008-05-18 06:22:34 +02:00
# * <tt>:minimum</tt> - Assertion is true if the number of selected
2007-02-10 00:12:31 +01:00
# elements is at least this value.
2008-05-18 06:22:34 +02:00
# * <tt>:maximum</tt> - Assertion is true if the number of selected
2007-02-10 00:12:31 +01:00
# elements is at most this value.
#
# If the method is called with a block, once all equality tests are
# evaluated the block is called with an array of all matched elements.
#
2007-12-21 08:48:59 +01:00
# ==== Examples
2007-02-10 00:12:31 +01:00
#
# # At least one form element
# assert_select "form"
#
# # Form element includes four input fields
# assert_select "form input", 4
#
# # Page title is "Welcome"
# assert_select "title", "Welcome"
#
# # Page title is "Welcome" and there is only one title element
# assert_select "title", {:count=>1, :text=>"Welcome"},
# "Wrong title or more than one title element"
#
# # Page contains no forms
# assert_select "form", false, "This page must contain no forms"
#
# # Test the content and style
# assert_select "body div.header ul.menu"
#
# # Use substitution values
# assert_select "ol>li#?", /item-\d+/
#
# # All input fields in the form have a name
# assert_select "form input" do
# assert_select "[name=?]", /.+/ # Not empty
# end
def assert_select ( * args , & block )
# Start with optional element followed by mandatory selector.
arg = args . shift
if arg . is_a? ( HTML :: Node )
# First argument is a node (tag or text, but also HTML root),
# so we know what we're selecting from.
root = arg
arg = args . shift
elsif arg == nil
# This usually happens when passing a node/element that
# happens to be nil.
raise ArgumentError , " First argument is either selector or element to select, but nil found. Perhaps you called assert_select with an element that does not exist? "
elsif @selected
root = HTML :: Node . new ( nil )
root . children . concat @selected
else
# Otherwise just operate on the response document.
root = response_from_page_or_rjs
end
2009-12-01 02:38:34 +01:00
2007-02-10 00:12:31 +01:00
# First or second argument is the selector: string and we pass
# all remaining arguments. Array and we pass the argument. Also
# accepts selector itself.
case arg
when String
selector = HTML :: Selector . new ( arg , args )
when Array
selector = HTML :: Selector . new ( * arg )
when HTML :: Selector
selector = arg
else raise ArgumentError , " Expecting a selector as the first argument "
end
2009-12-01 02:38:34 +01:00
2007-02-10 00:12:31 +01:00
# Next argument is used for equality tests.
equals = { }
case arg = args . shift
when Hash
equals = arg
when String , Regexp
equals [ :text ] = arg
when Integer
equals [ :count ] = arg
when Range
equals [ :minimum ] = arg . begin
equals [ :maximum ] = arg . end
when FalseClass
equals [ :count ] = 0
when NilClass , TrueClass
equals [ :minimum ] = 1
else raise ArgumentError , " I don't understand what you're trying to match "
end
# By default we're looking for at least one match.
if equals [ :count ]
equals [ :minimum ] = equals [ :maximum ] = equals [ :count ]
else
equals [ :minimum ] = 1 unless equals [ :minimum ]
end
# Last argument is the message we use if the assertion fails.
message = args . shift
#- message = "No match made with selector #{selector.inspect}" unless message
if args . shift
raise ArgumentError , " Not expecting that last argument, you either have too many arguments, or they're the wrong type "
end
matches = selector . select ( root )
# If text/html, narrow down to those elements that match it.
content_mismatch = nil
if match_with = equals [ :text ]
matches . delete_if do | match |
text = " "
2008-05-18 06:22:34 +02:00
text . force_encoding ( match_with . encoding ) if text . respond_to? ( :force_encoding )
2007-02-10 00:12:31 +01:00
stack = match . children . reverse
while node = stack . pop
if node . tag?
stack . concat node . children . reverse
else
2008-05-18 06:22:34 +02:00
content = node . content
content . force_encoding ( match_with . encoding ) if content . respond_to? ( :force_encoding )
text << content
2007-02-10 00:12:31 +01:00
end
end
text . strip! unless NO_STRIP . include? ( match . name )
unless match_with . is_a? ( Regexp ) ? ( text =~ match_with ) : ( text == match_with . to_s )
content_mismatch || = build_message ( message , " <?> expected but was \n <?>. " , match_with , text )
true
end
end
elsif match_with = equals [ :html ]
matches . delete_if do | match |
html = match . children . map ( & :to_s ) . join
html . strip! unless NO_STRIP . include? ( match . name )
unless match_with . is_a? ( Regexp ) ? ( html =~ match_with ) : ( html == match_with . to_s )
content_mismatch || = build_message ( message , " <?> expected but was \n <?>. " , match_with , html )
true
end
end
end
# Expecting foo found bar element only if found zero, not if
# found one but expecting two.
message || = content_mismatch if matches . empty?
# Test minimum/maximum occurrence.
2007-12-21 08:48:59 +01:00
min , max = equals [ :minimum ] , equals [ :maximum ]
message = message || %( Expected #{ count_description ( min , max ) } matching " #{ selector . to_s } ", found #{ matches . size } . )
assert matches . size > = min , message if min
assert matches . size < = max , message if max
2007-02-10 00:12:31 +01:00
# If a block is given call that block. Set @selected to allow
# nested assert_select, which can be nested several levels deep.
if block_given? && ! matches . empty?
begin
in_scope , @selected = @selected , matches
yield matches
ensure
@selected = in_scope
end
end
# Returns all matches elements.
matches
end
2009-12-01 02:38:34 +01:00
2007-12-21 08:48:59 +01:00
def count_description ( min , max ) #:nodoc:
pluralize = lambda { | word , quantity | word << ( quantity == 1 ? '' : 's' ) }
2009-12-01 02:38:34 +01:00
2007-12-21 08:48:59 +01:00
if min && max && ( max != min )
" between #{ min } and #{ max } elements "
elsif min && ! ( min == 1 && max == 1 )
" at least #{ min } #{ pluralize [ 'element' , min ] } "
elsif max
" at most #{ max } #{ pluralize [ 'element' , max ] } "
end
end
2009-12-01 02:38:34 +01:00
2007-02-10 00:12:31 +01:00
# :call-seq:
# assert_select_rjs(id?) { |elements| ... }
# assert_select_rjs(statement, id?) { |elements| ... }
# assert_select_rjs(:insert, position, id?) { |elements| ... }
#
# Selects content from the RJS response.
#
# === Narrowing down
#
# With no arguments, asserts that one or more elements are updated or
# inserted by RJS statements.
#
# Use the +id+ argument to narrow down the assertion to only statements
# that update or insert an element with that identifier.
#
# Use the first argument to narrow down assertions to only statements
2009-12-01 02:38:34 +01:00
# of that type. Possible values are <tt>:replace</tt>, <tt>:replace_html</tt>,
2007-12-21 08:48:59 +01:00
# <tt>:show</tt>, <tt>:hide</tt>, <tt>:toggle</tt>, <tt>:remove</tt> and
# <tt>:insert_html</tt>.
2007-02-10 00:12:31 +01:00
#
2007-12-21 08:48:59 +01:00
# Use the argument <tt>:insert</tt> followed by an insertion position to narrow
2007-02-10 00:12:31 +01:00
# down the assertion to only statements that insert elements in that
2007-12-21 08:48:59 +01:00
# position. Possible values are <tt>:top</tt>, <tt>:bottom</tt>, <tt>:before</tt>
# and <tt>:after</tt>.
#
# Using the <tt>:remove</tt> statement, you will be able to pass a block, but it will
# be ignored as there is no HTML passed for this statement.
2007-02-10 00:12:31 +01:00
#
# === Using blocks
#
2008-06-02 08:35:38 +02:00
# Without a block, +assert_select_rjs+ merely asserts that the response
2007-02-10 00:12:31 +01:00
# contains one or more RJS statements that replace or update content.
#
2008-06-02 08:35:38 +02:00
# With a block, +assert_select_rjs+ also selects all elements used in
2007-02-10 00:12:31 +01:00
# these statements and passes them to the block. Nested assertions are
# supported.
#
2008-06-02 08:35:38 +02:00
# Calling +assert_select_rjs+ with no arguments and using nested asserts
2007-02-10 00:12:31 +01:00
# asserts that the HTML content is returned by one or more RJS statements.
2008-06-02 08:35:38 +02:00
# Using +assert_select+ directly makes the same assertion on the content,
2007-02-10 00:12:31 +01:00
# but without distinguishing whether the content is returned in an HTML
# or JavaScript.
#
2007-12-21 08:48:59 +01:00
# ==== Examples
2007-02-10 00:12:31 +01:00
#
# # Replacing the element foo.
# # page.replace 'foo', ...
# assert_select_rjs :replace, "foo"
#
# # Replacing with the chained RJS proxy.
# # page[:foo].replace ...
# assert_select_rjs :chained_replace, 'foo'
#
# # Inserting into the element bar, top position.
# assert_select_rjs :insert, :top, "bar"
#
2007-12-21 08:48:59 +01:00
# # Remove the element bar
# assert_select_rjs :remove, "bar"
#
2007-02-10 00:12:31 +01:00
# # Changing the element foo, with an image.
# assert_select_rjs "foo" do
# assert_select "img[src=/images/logo.gif""
# end
#
# # RJS inserts or updates a list with four items.
# assert_select_rjs do
# assert_select "ol>li", 4
# end
#
# # The same, but shorter.
# assert_select "ol>li", 4
def assert_select_rjs ( * args , & block )
2008-09-07 07:54:05 +02:00
rjs_type = args . first . is_a? ( Symbol ) ? args . shift : nil
id = args . first . is_a? ( String ) ? args . shift : nil
2007-02-10 00:12:31 +01:00
# If the first argument is a symbol, it's the type of RJS statement we're looking
# for (update, replace, insertion, etc). Otherwise, we're looking for just about
# any RJS statement.
2008-09-07 07:54:05 +02:00
if rjs_type
2007-02-10 00:12:31 +01:00
if rjs_type == :insert
2008-09-07 07:54:05 +02:00
position = args . shift
2009-02-04 21:26:08 +01:00
id = args . shift
2008-09-07 07:54:05 +02:00
insertion = " insert_ #{ position } " . to_sym
raise ArgumentError , " Unknown RJS insertion type #{ position } " unless RJS_STATEMENTS [ insertion ]
2007-02-10 00:12:31 +01:00
statement = " ( #{ RJS_STATEMENTS [ insertion ] } ) "
else
raise ArgumentError , " Unknown RJS statement type #{ rjs_type } " unless RJS_STATEMENTS [ rjs_type ]
statement = " ( #{ RJS_STATEMENTS [ rjs_type ] } ) "
end
else
statement = " #{ RJS_STATEMENTS [ :any ] } "
end
# Next argument we're looking for is the element identifier. If missing, we pick
2008-09-07 07:54:05 +02:00
# any element, otherwise we replace it in the statement.
pattern = Regexp . new (
id ? statement . gsub ( RJS_ANY_ID , " \" #{ id } \" " ) : statement
)
2007-02-10 00:12:31 +01:00
# Duplicate the body since the next step involves destroying it.
matches = nil
2007-12-21 08:48:59 +01:00
case rjs_type
when :remove , :show , :hide , :toggle
matches = @response . body . match ( pattern )
else
@response . body . gsub ( pattern ) do | match |
2008-09-07 07:54:05 +02:00
html = unescape_rjs ( match )
2007-12-21 08:48:59 +01:00
matches || = [ ]
matches . concat HTML :: Document . new ( html ) . root . children . select { | n | n . tag? }
" "
end
2007-02-10 00:12:31 +01:00
end
2007-12-21 08:48:59 +01:00
2007-02-10 00:12:31 +01:00
if matches
2007-12-21 08:48:59 +01:00
assert_block ( " " ) { true } # to count the assertion
if block_given? && ! ( [ :remove , :show , :hide , :toggle ] . include? rjs_type )
2007-02-10 00:12:31 +01:00
begin
in_scope , @selected = @selected , matches
yield matches
ensure
@selected = in_scope
end
end
matches
else
# RJS statement not found.
2008-10-27 07:47:01 +01:00
case rjs_type
when :remove , :show , :hide , :toggle
flunk_message = " No RJS statement that #{ rjs_type . to_s } s ' #{ id } ' was rendered. "
else
flunk_message = " No RJS statement that replaces or inserts HTML content. "
end
flunk args . shift || flunk_message
2007-02-10 00:12:31 +01:00
end
end
# :call-seq:
# assert_select_encoded(element?) { |elements| ... }
#
# Extracts the content of an element, treats it as encoded HTML and runs
# nested assertion on it.
#
# You typically call this method within another assertion to operate on
# all currently selected elements. You can also pass an element or array
# of elements.
#
# The content of each element is un-encoded, and wrapped in the root
# element +encoded+. It then calls the block with all un-encoded elements.
#
2007-12-21 08:48:59 +01:00
# ==== Examples
# # Selects all bold tags from within the title of an ATOM feed's entries (perhaps to nab a section name prefix)
# assert_select_feed :atom, 1.0 do
# # Select each entry item and then the title item
# assert_select "entry>title" do
# # Run assertions on the encoded title elements
# assert_select_encoded do
# assert_select "b"
# end
# end
# end
2009-12-01 02:38:34 +01:00
#
2007-02-10 00:12:31 +01:00
#
2007-12-21 08:48:59 +01:00
# # Selects all paragraph tags from within the description of an RSS feed
2007-02-10 00:12:31 +01:00
# assert_select_feed :rss, 2.0 do
# # Select description element of each feed item.
# assert_select "channel>item>description" do
# # Run assertions on the encoded elements.
# assert_select_encoded do
# assert_select "p"
# end
# end
# end
def assert_select_encoded ( element = nil , & block )
case element
when Array
elements = element
when HTML :: Node
elements = [ element ]
when nil
unless elements = @selected
raise ArgumentError , " First argument is optional, but must be called from a nested assert_select "
end
else
raise ArgumentError , " Argument is optional, and may be node or array of nodes "
end
fix_content = lambda do | node |
# Gets around a bug in the Rails 1.1 HTML parser.
node . content . gsub ( / <! \ [CDATA \ [(.*)( \ ] \ ]>)? /m ) { CGI . escapeHTML ( $1 ) }
end
selected = elements . map do | element |
text = element . children . select { | c | not c . tag? } . map { | c | fix_content [ c ] } . join
root = HTML :: Document . new ( CGI . unescapeHTML ( " <encoded> #{ text } </encoded> " ) ) . root
css_select ( root , " encoded:root " , & block ) [ 0 ]
end
begin
old_selected , @selected = @selected , selected
assert_select " :root " , & block
ensure
@selected = old_selected
end
end
# :call-seq:
# assert_select_email { }
#
# Extracts the body of an email and runs nested assertions on it.
#
# You must enable deliveries for this assertion to work, use:
# ActionMailer::Base.perform_deliveries = true
#
2007-12-21 08:48:59 +01:00
# ==== Examples
#
# assert_select_email do
# assert_select "h1", "Email alert"
# end
#
# assert_select_email do
# items = assert_select "ol>li"
# items.each do
# # Work with items here...
# end
# end
2007-02-10 00:12:31 +01:00
#
def assert_select_email ( & block )
deliveries = ActionMailer :: Base . deliveries
assert ! deliveries . empty? , " No e-mail in delivery list "
for delivery in deliveries
for part in delivery . parts
if part [ " Content-Type " ] . to_s =~ / ^text \/ html \ W /
root = HTML :: Document . new ( part . body ) . root
assert_select root , " :root " , & block
end
end
end
end
protected
unless const_defined? ( :RJS_STATEMENTS )
2008-09-07 07:54:05 +02:00
RJS_PATTERN_HTML = " \" (( \\ \\ \" |[^ \" ])*) \" "
RJS_ANY_ID = " \" ([^ \" ])* \" "
RJS_STATEMENTS = {
:chained_replace = > " \\ $ \\ ( #{ RJS_ANY_ID } \\ ) \\ .replace \\ ( #{ RJS_PATTERN_HTML } \\ ) " ,
:chained_replace_html = > " \\ $ \\ ( #{ RJS_ANY_ID } \\ ) \\ .update \\ ( #{ RJS_PATTERN_HTML } \\ ) " ,
:replace_html = > " Element \\ .update \\ ( #{ RJS_ANY_ID } , #{ RJS_PATTERN_HTML } \\ ) " ,
:replace = > " Element \\ .replace \\ ( #{ RJS_ANY_ID } , #{ RJS_PATTERN_HTML } \\ ) "
2007-02-10 00:12:31 +01:00
}
2008-09-07 07:54:05 +02:00
[ :remove , :show , :hide , :toggle ] . each do | action |
RJS_STATEMENTS [ action ] = " Element \\ . #{ action } \\ ( #{ RJS_ANY_ID } \\ ) "
end
RJS_INSERTIONS = [ " top " , " bottom " , " before " , " after " ]
2007-02-10 00:12:31 +01:00
RJS_INSERTIONS . each do | insertion |
2008-09-07 07:54:05 +02:00
RJS_STATEMENTS [ " insert_ #{ insertion } " . to_sym ] = " Element.insert \\ ( #{ RJS_ANY_ID } , \\ { #{ insertion } : #{ RJS_PATTERN_HTML } \\ } \\ ) "
2007-02-10 00:12:31 +01:00
end
2008-09-07 07:54:05 +02:00
RJS_STATEMENTS [ :insert_html ] = " Element.insert \\ ( #{ RJS_ANY_ID } , \\ { ( #{ RJS_INSERTIONS . join ( '|' ) } ): #{ RJS_PATTERN_HTML } \\ } \\ ) "
2007-02-10 00:12:31 +01:00
RJS_STATEMENTS [ :any ] = Regexp . new ( " ( #{ RJS_STATEMENTS . values . join ( '|' ) } ) " )
RJS_PATTERN_UNICODE_ESCAPED_CHAR = / \\ u([0-9a-zA-Z]{4}) /
end
2008-06-02 08:35:38 +02:00
# +assert_select+ and +css_select+ call this to obtain the content in the HTML
2007-02-10 00:12:31 +01:00
# page, or from all the RJS statements, depending on the type of response.
def response_from_page_or_rjs ( )
2007-12-21 08:48:59 +01:00
content_type = @response . content_type
2009-02-04 21:26:08 +01:00
if content_type && Mime :: JS =~ content_type
2007-02-10 00:12:31 +01:00
body = @response . body . dup
root = HTML :: Node . new ( nil )
2007-12-21 08:48:59 +01:00
2007-02-10 00:12:31 +01:00
while true
2008-09-07 07:54:05 +02:00
next if body . sub! ( RJS_STATEMENTS [ :any ] ) do | match |
html = unescape_rjs ( match )
2007-02-10 00:12:31 +01:00
matches = HTML :: Document . new ( html ) . root . children . select { | n | n . tag? }
root . children . concat matches
" "
end
break
end
2007-12-21 08:48:59 +01:00
2007-02-10 00:12:31 +01:00
root
else
html_document . root
end
end
# Unescapes a RJS string.
def unescape_rjs ( rjs_string )
# RJS encodes double quotes and line breaks.
unescaped = rjs_string . gsub ( '\"' , '"' )
2007-12-21 08:48:59 +01:00
unescaped . gsub! ( / \\ \/ / , '/' )
2007-02-10 00:12:31 +01:00
unescaped . gsub! ( '\n' , " \n " )
2007-10-15 19:16:54 +02:00
unescaped . gsub! ( '\076' , '>' )
unescaped . gsub! ( '\074' , '<' )
2007-02-10 00:12:31 +01:00
# RJS encodes non-ascii characters.
unescaped . gsub! ( RJS_PATTERN_UNICODE_ESCAPED_CHAR ) { | u | [ $1 . hex ] . pack ( 'U*' ) }
unescaped
end
end
end
end