Merge lp:~elopio/osqa-sst-helpers/mail into lp:~online-services-qa/osqa-sst-helpers/canonical-identity-provider

[Leo Arias]

Added get_verification_code_from_email functions from sso acceptance tests.

Added get_verification_code_from_email functions from sso acceptance tests.

[Leo Arias (elopio)]

This was proposed by bloodearnest and reviewed by me, but in the wrong branch.

review: Approve

=== added file 'sso_sst_helpers/utils/mail.py'
--- sso_sst_helpers/utils/mail.py	1970-01-01 00:00:00 +0000
+++ sso_sst_helpers/utils/mail.py	2012-09-18 21:38:18 +0000
@@ -0,0 +1,50 @@
+import quopri
+import re
+
+from canonical.isd.tests.sst import mail
+
+
+def _get_verification_data_for_address(email_address):
+    """A private helper for public helpers below.
+
+    Note: We have two different public helpers here for verification
+    code and link so that functional tests don't need to deal with
+    idioms like:
+        vcode, ignored = get_verification_for_address(email_address).
+    """
+    email_msg = mail.get_latest_email_sent_to(email_address)
+    vcode = link = None
+    if email_msg:
+        # The body is encoded as quoted-printable.  This affects any
+        # line longer than a certain length.  Decode now to not have
+        # to worry about it in the regexen.
+        body = quopri.decodestring(email_msg.get_payload())
+        match = re.search(
+            'Here is your confirmation code:(.*)(Enter|If you made)',
+            body, re.S)
+        if match:
+            vcode = match.group(1).strip()
+        else:
+            raise AssertionError("No verification code found in email.")
+        match = re.search(
+            'confirm your (?:account|email address|reset):(.*)If',
+            body, re.S)
+        if match:
+            link = match.group(1).strip()
+        else:
+            raise AssertionError("No verification link found in email.")
+    return vcode, link
+
+
+def get_verification_code_for_address(email_address):
+    print("Retrieving verification code for %s." % email_address)
+    vcode, link = _get_verification_data_for_address(email_address)
+    print("Verification code retrieved: %s." % vcode)
+    return vcode
+
+
+def get_verification_link_for_address(email_address):
+    print("Retrieving verification link for %s." % email_address)
+    vcode, link = _get_verification_data_for_address(email_address)
+    print("Verification link retrieved: %s." % link)
+    return link