prefer the html_part in a multipart email and throw away the host:port

part of the extracted urls.

Author: jlsync

lib/pickle/email.rb

@@ -20,12 +20,13 @@ def email_has_fields?(email, fields)
     end
 
     def visit_in_email(email, link_text)
-      visit(parse_email_for_link(email, link_text))
+      link = parse_email_for_link(email, link_text)
+      visit URI.parse(link).path
     end
 
     def click_first_link_in_email(email)
       link = links_in_email(email).first
-      visit link
+      visit URI.parse(link).path
     end
 
   protected
@@ -64,13 +65,23 @@ def parse_email_for_explicit_link(email, regex)
 
     # e.g. Click here in  <a href="http://confirm">Click here</a>
     def parse_email_for_anchor_text_link(email, link_text)
-      if match_data = email.body.match(%r{<a[^>]*href=['"]?([^'"]*)['"]?[^>]*?>[^<]*?#{link_text}[^<]*?</a>})
+      if email.multipart?
+        body = email.html_part.body
+      else
+        body = email.body
+      end
+      if match_data = body.match(%r{<a[^>]*href=['"]?([^'"]*)['"]?[^>]*?>[^<]*?#{link_text}[^<]*?</a>})
         match_data[1]
       end
     end
 
     def links_in_email(email, protos=['http', 'https'])
-      URI.extract(email.body.to_s, protos)
+      if email.multipart?
+        body = email.html_part.body
+      else
+        body = email.body
+      end
+      URI.extract(body.to_s, protos)
     end
 
   end