Merge pull request #5 from eodin/master

grabBodyFromEmail(): allow grab email MIME-part with Content-part alternatives

Author: ericmartel

README.md

@@ -231,6 +231,14 @@ Returns the string containing all of the recipients, such as To, CC and if provi
 grabBodyFromEmail
 ```
 Returns the email body as string
+```
+$email_body = $I->grabBodyFromEmail('text/html');
+$doc = new DOMDocument();
+$doc->loadHTML($email_body);
+$login = $doc->getElementById('login')->textContent;
+$password = $doc->getElementById('password')->textContent;
+```
+Fetch login & password from 'text/html' email MIME-part.
 
 ### To Do
 While this framework should be sufficient for most email testing scenarios, the remaining features need to be implemented
@@ -265,5 +273,5 @@ THE SOFTWARE.
    [MailHogModule]: http://github.com/ericmartel/codeception-email-mailhog
    [MailCatcherModule]: http://github.com/ericmartel/codeception-email-mailcatcher
    [MailtrapModule]: http://github.com/ericmartel/codeception-email-mailtrap
-   
+
 

src/TestsEmails.php

@@ -563,11 +563,23 @@ public function dontSeeInEmailPriority($email, $expected)
   }
 
     /**
+
+     * @param string $content_type_alternative      MIME-part Content-Type
      * @return string Body
      */
-    public function grabBodyFromEmail()
+    public function grabBodyFromEmail($content_type_alternative = null)
     {
         $email = $this->getOpenedEmail();
-        return $this->getDecodedEmailProperty($email, $email->Content->Body);
+
+        if( isset($content_type_alternative) ) {
+            foreach ($email->MIME->Parts as $part) {
+                if (!empty($part->Headers->{'Content-Type'}[0]) &&
+                    strpos($part->Headers->{'Content-Type'}[0], $content_type_alternative) !== false) {
+                    return $this->getDecodedEmailProperty($part, $part->Body);  //return first entry
+                }
+            }
+            return null;                                                        //not found
+        }
+        return $this->getDecodedEmailProperty($email, $email->Content->Body);   //return full email
     }
 };