check_add_email.py

#!/usr/bin/env python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

import re

import pytest

from .. import BrowserID
from base import BaseTest
import restmail


@pytest.mark.nondestructive
class TestSignIn(BaseTest):

    @pytest.mark.travis
    def test_add_email(self, mozwebqa):
        user = self.create_verified_user(mozwebqa.selenium, mozwebqa.timeout)
        user.additional_emails.append('%s_1@restmail.net' % user.id)

        # log out
        mozwebqa.selenium.get('%s/' % mozwebqa.base_url)
        self.log_out(mozwebqa.selenium, mozwebqa.timeout)

        # initiate add email
        mozwebqa.selenium.find_element_by_css_selector('#loggedout button').click()
        from .. pages.sign_in import SignIn
        signin = SignIn(mozwebqa.selenium, mozwebqa.timeout, expect='returning')

        signin.click_add_another_email_address()
        signin.new_email = user.additional_emails[0]
        signin.click_add_new_email()
        signin.close_window()
        signin.switch_to_main_window()

        mail = restmail.get_mail(user.additional_emails[0],
                                 timeout=mozwebqa.timeout)
        assert 'Click to confirm this email address' in mail[0]['text']
        confirm_url = re.search(BrowserID.CONFIRM_URL_REGEX,
            mail[0]['text']).group(0)

        from .. pages.complete_registration import CompleteRegistration
        complete_registration = CompleteRegistration(mozwebqa.selenium,
            mozwebqa.timeout, confirm_url, expect='redirect')

        self.log_out(mozwebqa.selenium, mozwebqa.timeout)

        mozwebqa.selenium.find_element_by_css_selector('#loggedout button').click()
        signin = SignIn(mozwebqa.selenium, mozwebqa.timeout, expect='returning')

        assert user.additional_emails[0] in signin.emails
        assert signin.selected_email == user.additional_emails[0]