From 932fdab8844b866c55262d890eb808b513fff9ac Mon Sep 17 00:00:00 2001
From: Bryan Helmkamp <bryan@brynary.com>
Date: Sun, 15 Feb 2009 19:32:25 -0500
Subject: [PATCH] Get quoting working for content

---
 lib/webrat/core/matchers/have_tag.rb | 23 ++++++++++++++++++-----
 spec/public/matchers_spec.rb         | 15 +++++++++++++++
 2 files changed, 33 insertions(+), 5 deletions(-)

diff --git a/lib/webrat/core/matchers/have_tag.rb b/lib/webrat/core/matchers/have_tag.rb
index 3f809521..7529fcc3 100644
--- a/lib/webrat/core/matchers/have_tag.rb
+++ b/lib/webrat/core/matchers/have_tag.rb
@@ -36,14 +36,27 @@ def tag_inspect
       def query
         options  = @expected.last.dup
         selector = @expected.first.to_s
-
-        selector << ":contains('#{options.delete(:content)}')" if options[:content]
-
+        
         options.each do |key, value|
+          next if key == :content
           selector << "[#{key}='#{value}']"
         end
-
-        Nokogiri::CSS::Parser.parse(selector).map { |ast| ast.to_xpath }
+        
+        q = Nokogiri::CSS::Parser.parse(selector).map { |ast| ast.to_xpath }.first
+        
+        if options[:content] && options[:content].include?("'") && options[:content].include?('"')
+          parts = options[:content].split("'").map do |part|
+            "'#{part}'"
+          end
+          
+          string = "concat(" + parts.join(", \"'\", ") + ")"
+          q << "[contains(., #{string})]"
+        elsif options[:content] && options[:content].include?("'")
+          q << "[contains(., \"#{options[:content]}\")]"
+        elsif options[:content]
+          q << "[contains(., '#{options[:content]}')]"
+        end
+        q
       end
     end
 
diff --git a/spec/public/matchers_spec.rb b/spec/public/matchers_spec.rb
index 59642768..ed36c6f9 100644
--- a/spec/public/matchers_spec.rb
+++ b/spec/public/matchers_spec.rb
@@ -8,6 +8,9 @@
     @body = <<-HTML
       <div id='main'>
         <div class='inner'>hello, world!</div>
+        <h2>Welcome "Bryan"</h2>
+        <h3>Welcome 'Bryan'</h3>
+        <h4>Welcome 'Bryan"</h4>
         <ul>
           <li>First</li>
           <li>Second</li>
@@ -178,6 +181,18 @@
       @body.should have_tag("div", :content  => "hello, world!")
     end
     
+    it "should be able to specify the content of the tag with double quotes in it" do
+      @body.should have_tag("h2", :content  => 'Welcome "Bryan"')
+    end
+    
+    it "should be able to specify the content of the tag with single quotes in it" do
+      @body.should have_tag("h3", :content  => "Welcome 'Bryan'")
+    end
+    
+    it "should be able to specify the content of the tag with both kinds of quotes" do
+      @body.should have_tag("h4", :content  => "Welcome 'Bryan\"")
+    end
+    
     it "should be able to specify the attributes of the tag" do
       @body.should have_tag("div", :class => "inner")
     end