Strip whitespace from css selectors.

Author: GabeIsman

app/models/page.rb

@@ -37,11 +37,11 @@ def document
   end
 
   def match_text
-    @match = document.css(self.css_selector)
+    @match = document.css(self.css_selector.strip)
 
     if self.exclude_selector.present?
       # Set the content of the exclude selector to the empty string
-      @match.css(self.exclude_selector).each do |node|
+      @match.css(self.exclude_selector.strip).each do |node|
         node.content = ""
       end
     end
@@ -50,7 +50,7 @@ def match_text
   end
 
   def match_html
-    document.css(self.css_selector).to_html
+    document.css(self.css_selector.strip).to_html
   end
 
   def sha2_hash
@@ -59,6 +59,8 @@ def sha2_hash
 
   def sanitize
     self.url = url.strip
+    self.css_selector = css_selector.strip unless self.css_selector.nil?
+    self.exclude_selector = exclude_selector.strip unless self.exclude_selector.nil?
   end
 
   def update_subscriptions

app/models/page_snapshot.rb

@@ -12,11 +12,11 @@ def document
   end
 
   def match_text
-    @match = document.css(self.page.css_selector)
+    @match = document.css(self.page.css_selector.strip)
 
     if self.page.exclude_selector.present?
       # Set the content of the exclude selector to the empty string
-      @match.css(self.page.exclude_selector).each do |node|
+      @match.css(self.page.exclude_selector.strip).each do |node|
         node.content = ""
       end
     end

spec/models/page_spec.rb

@@ -70,6 +70,11 @@
     expect(page.url).to eq url
   end
 
+  it "strips whitespace from the selectors" do
+    selector = ".test"
+    page = create(:page, css_selector: " #{selector}    ", exclude_selector: " #{selector} ")
+  end
+
   it "gracefully handles parsing an invalid uri" do
     weird_url = " bad:///site.com"
     page = build(:page, url: weird_url) # our sanitizer is on before_save, which does not run here