Added find_css to email driver

DavyJonesLocker · Nov 18, 2014 · 17bf7b4 · 17bf7b4
1 parent 9ece658
commit 17bf7b4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/lib/capybara/email/driver.rb b/lib/capybara/email/driver.rb
@@ -35,6 +35,9 @@ def find(selector)
 
   alias_method :find_xpath, :find
 
+  def find_css(selector)
+    dom.css(selector).map { |node| Capybara::Email::Node.new(self, node) }
+  end
 
   # String version of email HTML source
   #

diff --git a/spec/email/driver_spec.rb b/spec/email/driver_spec.rb
@@ -19,6 +19,7 @@ def self.call(env)
     current_email.click_link 'example'
     expect(page).to have_content 'Hello world!'
     expect(current_email).to have_content 'This is only a html test'
+    expect(current_email).to have_css 'a'
 
     expect(all_emails.first).to eq email