Merge lp:~elopio/+junk/referrals-create-account into lp:ubuntuone-testing

[Leo Arias]

Added the first referrals test: Create account from referrals link.

Added the first referrals test: Create account from referrals link.
Please note that the new account creation is failing. A separate branch will fix that.

[lp:~elopio/+junk/referrals-create-account]

117. By Leo Arias on 2012-08-07

Fixed the calls to assert_equal.

118. By Leo Arias on 2012-08-07

Return floats.

119. By Leo Arias on 2012-08-07

Added the missing logout action call.

Unmerged revisions

119. By Leo Arias on 2012-08-07

Added the missing logout action call.

118. By Leo Arias on 2012-08-07

Return floats.

117. By Leo Arias on 2012-08-07

Fixed the calls to assert_equal.

116. By Leo Arias on 2012-08-07

Removed the print statements.

115. By Leo Arias on 2012-08-07

Added the first referrals test: Create account from referrals link.

114. By Leo Arias on 2012-08-07

Merged with trunk.

113. By Leo Arias on 2012-08-07

Added a couple of actions to account.

112. By Leo Arias on 2012-08-06

Cleaned a little the account actions.

=== added directory 'ubuntuone/web/tests/sst/referrals'
=== added file 'ubuntuone/web/tests/sst/referrals/u1webref123_createaccountfromreferralslink.py'
--- ubuntuone/web/tests/sst/referrals/u1webref123_createaccountfromreferralslink.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/referrals/u1webref123_createaccountfromreferralslink.py	2012-08-07 18:48:18 +0000
@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2012 Canonical Ltd.
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License version 3, as published
+# by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranties of
+# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
+# PURPOSE.  See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from sst.actions import *
+import actions.setup as setup_actions
+import actions.account as account_actions
+import actions.newdashboard as dashboard_actions
+import actions.header as header_actions
+
+# Log in with the referrer.
+setup_actions.setup()
+account_actions.open()
+referrer_available_space_before_referral = \
+    account_actions.get_available_storage_space_in_gigabytes()
+referrer_storage_space_from_referrals_before_referral = \
+    account_actions.get_storage_space_from_referrals_in_gigabytes()
+dashboard_actions.open()
+referrals_link = dashboard_actions.get_referrals_link()
+header_actions.logout()
+# Log in with the recipient.
+setup_actions.setup_new_user_with_referrals_link(referrals_link)
+account_actions.open()
+recipient_available_space = \
+    account_actions.get_available_storage_space_in_gigabytes()
+assert_equal(recipient_available_space, 5.5)
+recipient_storage_space_from_referrals = \
+    account_actions.get_storage_space_from_referrals_in_gigabytes()
+assert_equal(recipient_storage_space_from_referrals, 0.5)
+header_actions.logout()
+# Log in again with the referrer.
+setup_actions.setup()
+account_actions.open()
+referrer_available_space_after_referral = \
+    account_actions.get_available_storage_space_in_gigabytes()
+assert_equal(referrer_available_space_after_referral,
+    referrer_available_space_before_referral + 0.5)
+referrer_storage_space_from_referrals_after_referral = \
+    account_actions.get_storage_space_from_referrals_in_gigabytes()
+assert_equal(referrer_storage_space_from_referrals_after_referral,
+    referrer_storage_space_from_referrals_before_referral + 0.5)

=== modified file 'ubuntuone/web/tests/sst/shared/actions/account.py'
--- ubuntuone/web/tests/sst/shared/actions/account.py	2011-12-09 18:19:11 +0000
+++ ubuntuone/web/tests/sst/shared/actions/account.py	2012-08-07 18:48:18 +0000
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-# Copyright 2011 Canonical Ltd.
+# Copyright 2011, 2012 Canonical Ltd.
 #
 # This program is free software: you can redistribute it and/or modify it
 # under the terms of the GNU General Public License version 3, as published
@@ -16,6 +16,8 @@
 
 """Actions for the Account page of the Ubuntu One website."""
 
+import re
+
 from sst.actions import *
 import actions.header as header_actions
 
@@ -23,29 +25,98 @@
     header_actions.open_my_account()
     assert_page_title(), 'My account should be opened'
 
+
 def assert_page_title():
     assert_title(u'Ubuntu One : Your Account')
 
+
 def assert_account_details(full_name, email):
     _assert_name(full_name)
     _assert_email(email)
     _assert_password('********')
 
+
 def _assert_name(full_name):
     name_element = _get_user_details_data_elements()[0]
     assert_text(name_element, full_name)
 
+
+def _get_user_details_data_elements():
+    return get_elements_by_css('.user-details dd')
+
+
 def _assert_email(email):
     email_element = _get_user_details_data_elements()[1]
     assert_text(email_element, email)
 
+
 def _assert_password(password):
     password_element = _get_user_details_data_elements()[2]
     assert_text(password_element, password)
 
-def _get_user_details_data_elements():
-    return get_elements_by_css('.user-details dd')
              
 def click_edit_account_details():
-    click_link(get_element(text='Edit your account details in Ubuntu Single '
-                                'Sign On'))
+    click_link(_get_edit_account_details_link())
+
+
+def _get_edit_account_details_link():
+    get_element(text='Edit your account details in Ubuntu Single '
+        'Sign On')
+
+
+def get_available_storage_space_in_gigabytes():
+    available_storage_space = _get_available_storage_space()
+    assert available_storage_space.endswith(' GB'), \
+        'Available storage space is not in gigabytes.'
+    return float(available_storage_space[:-3])
+
+
+def _get_available_storage_space():
+    storage_summary = _get_storage_summary_string()
+    match = _get_storage_summary_match(storage_summary)
+    return match.group(3)
+
+
+def _get_storage_summary_string():
+    return _get_storage_summary_label().text
+
+
+def _get_storage_summary_label():
+    return get_elements_by_css('.storage > p')[0]
+    click_link(_get_edit_account_details_link())
+
+
+def _get_edit_account_details_link():
+    get_element(text='Edit your account details in Ubuntu Single '
+        'Sign On')
+
+
+def _get_storage_summary_match(storage_summary_string):
+    return re.match('(.*)% in use \((.*) of (.*)\)', storage_summary_string)
+
+
+def get_storage_space_from_referrals_in_gigabytes():
+    storage_space_from_referrals = _get_storage_space_from_referrals()
+    assert storage_space_from_referrals.endswith('GB'), \
+        'Storage space from referrals is not in gigabytes'
+    return float(storage_space_from_referrals[:-2])
+
+
+def _get_storage_space_from_referrals():
+    storage_space_from_referrals_string = \
+        _get_storage_space_from_referrals_string()
+    storage_space_from_referrals = '0GB'
+    if storage_space_from_referrals_string:
+        match = re.match('(.*) from referrals',
+            storage_space_from_referrals_string)
+        storage_space_from_referrals = match.group(1)
+    return storage_space_from_referrals
+
+
+def _get_storage_space_from_referrals_string():
+    storage_elements = get_elements_by_css('.storage .items>li>span')
+    storage_space_from_referrals = None
+    for element in storage_elements:
+        if 'from referrals' in element.text:
+            storage_space_from_referrals = element.text
+    return storage_space_from_referrals

=== added file 'ubuntuone/web/tests/sst/shared/actions/newdashboard.py'
--- ubuntuone/web/tests/sst/shared/actions/newdashboard.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/shared/actions/newdashboard.py	2012-08-07 18:48:18 +0000
@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2012 Canonical Ltd.
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License version 3, as published
+# by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranties of
+# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
+# PURPOSE.  See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+
+from sst.actions import *
+
+PAGE_TITLE = u'Ubuntu One : Dashboard'
+
+def open():
+    go_to('dashboard/?newdash=1')
+    assert_page_title(), 'Dashboard should be opened'
+
+
+def assert_page_title():
+    assert_title(PAGE_TITLE)
+
+
+# Referrals.
+
+def get_referrals_link():
+    return _get_referrals_element().get_attribute('value')
+
+
+def _get_referrals_element():
+    return get_element(id='ref-link')

=== modified file 'ubuntuone/web/tests/sst/shared/actions/setup.py'
--- ubuntuone/web/tests/sst/shared/actions/setup.py	2012-07-15 05:27:44 +0000
+++ ubuntuone/web/tests/sst/shared/actions/setup.py	2012-08-07 18:48:18 +0000
@@ -117,6 +117,10 @@
         skip('Can\'t create new users on production.')
     else:
         header_actions.click_login_or_sign_up()
+        _sign_up_on_sso_and_subscribe(user)
+
+
+def _sign_up_on_sso_and_subscribe(user):
         sso_actions.create_new_account(user)
         _confirm_email(user.email)
         services_actions.agree_and_subscribe_to_free_plan()
@@ -136,3 +140,13 @@
         click_button(get_elements('button.cta[type="submit"]')[0], wait=False)
     else:
         sso_actions.login(user)
+
+
+def setup_new_user_with_referrals_link(referrals_url):
+    _go_to_ubuntu_one_home()
+    _clear_previous_session()
+    go_to(referrals_url)
+    user = _get_random_user()
+    _sign_up_on_sso_and_subscribe(user)
+    dashboard_actions.wait_for_page_to_load()
+    header_actions.assert_login(user.full_name)