Merge lp:~elopio/ubuntuone-testing/sst-contacts into lp:ubuntuone-testing

[Leo Arias]

Added the actions and tests for contacts on the website (Fixes LP: #877934)

Added the actions and tests for contacts on the website (LP: #877934)

[Rick McBride (rmcbride)]

Excellent!

review: Approve

=== added directory 'ubuntuone/web/tests/sst/contacts'
=== added file 'ubuntuone/web/tests/sst/contacts/u1webc001_addcontact.py'
--- ubuntuone/web/tests/sst/contacts/u1webc001_addcontact.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/contacts/u1webc001_addcontact.py	2011-10-19 06:57:26 +0000
@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2011 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/>.
+
+"""Test case u1webc-001 Add Contact.
+Add a contact to Ubuntu One through the website. 
+
+"""
+
+import uuid
+from sst.actions import *
+import actions.setup as setup_actions
+import actions.contacts as contacts_actions
+from data.contact import Contact
+from data.contact import Name
+from data.contact import Email
+
+setup_actions.setup()
+contacts_actions.open()
+contact_uuid = uuid.uuid1()
+name = Name(first_name='Test first name %s' % contact_uuid, 
+            last_name='Test last name %s' % contact_uuid)
+email = Email('Other', '[e-mail]' % contact_uuid)
+contact = Contact(name, (email,))
+contacts_actions.add_contact(contact)
+contacts_actions.assert_contact_information(contact)
 
=== added file 'ubuntuone/web/tests/sst/contacts/u1webc002_editcontact.py'
--- ubuntuone/web/tests/sst/contacts/u1webc002_editcontact.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/contacts/u1webc002_editcontact.py	2011-10-19 06:57:26 +0000
@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2011 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/>.
+
+"""Test case u1webc-002 Edit Contact.
+Edit a contact stored on Ubuntu One through the website. 
+
+"""
+
+import uuid
+from sst.actions import *
+import actions.setup as setup_actions
+import actions.contacts as contacts_actions
+from data.contact import Contact
+from data.contact import Name
+from data.contact import Email
+
+setup_actions.setup()
+contacts_actions.open()
+contact_uuid = uuid.uuid1()
+name = Name(first_name='Test first name %s' % contact_uuid, 
+            last_name='Test last name %s' % contact_uuid)
+email = Email('Other', '[e-mail]' % contact_uuid)
+contact = Contact(name, (email,))
+contacts_actions.add_contact(contact)
+contacts_actions.assert_contact_information(contact)
+edited_name = Name(first_name='Edited ' + name.first_name,
+                   last_name='Edited ' + name.last_name)
+edited_email = Email(email.description, 'edited' + email.address)
+edited_contact = Contact(edited_name, (edited_email,))
+# This will also assert that the original email is not present in the list.
+# If it were, test will fail because two elements with the same id were found.
+contacts_actions.edit_contact(edited_contact)
+contacts_actions.assert_contact_information(edited_contact)
 
=== added file 'ubuntuone/web/tests/sst/contacts/u1webc003_deletecontact.py'
--- ubuntuone/web/tests/sst/contacts/u1webc003_deletecontact.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/contacts/u1webc003_deletecontact.py	2011-10-19 06:57:26 +0000
@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2011 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/>.
+
+"""Test case u1webc-003 Delete Contact.
+Delete a contact from Ubuntu One through the website. 
+
+"""
+
+import uuid
+from sst.actions import *
+import actions.setup as setup_actions
+import actions.contacts as contacts_actions
+from data.contact import Contact
+from data.contact import Name
+from data.contact import Email
+
+setup_actions.setup()
+contacts_actions.open()
+contact_uuid = uuid.uuid1()
+name = Name(first_name='Test first name %s' % contact_uuid, 
+            last_name='Test last name %s' % contact_uuid)
+email = Email('Other', '[e-mail]' % contact_uuid)
+contact = Contact(name, (email,))
+contacts_actions.add_contact(contact)
+contacts_actions.assert_contact_information(contact)
+contact_identifier = contacts_actions.get_contact_identifier()
+contacts_actions.assert_contact_exists(contact_identifier)
+contacts_actions.delete_contact()
+fails(contacts_actions.assert_contact_exists, contact_identifier)
 
=== added file 'ubuntuone/web/tests/sst/shared/actions/contacts.py'
--- ubuntuone/web/tests/sst/shared/actions/contacts.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/shared/actions/contacts.py	2011-10-19 06:57:26 +0000
@@ -0,0 +1,346 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2011 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/>.
+
+"""Actions for the Contacts page of the Ubuntu One website."""
+
+from sst.actions import *
+from loading import *
+from data.contact import Contact
+
+def open():
+    """Open the Contacts page and assert it's title."""
+    goto('/contacts/')
+    assert_title()
+
+def assert_title():
+    """Assert that the title of the page is the expected."""
+    title_is(u'Ubuntu One : Contacts')
+
+def add_contact(contact):
+    """Add a new contact.
+
+    Keyword arguments:
+    contact -- The contact that will be added.
+
+    """
+    link_add = get_element(css_class='add-button')
+    link_click(link_add)
+    wait_for_action_to_complete()
+    _fill_contact(contact)
+    link_save = get_elements(css_class='save-button')[0]
+    element_click(link_save)
+    wait_for_action_to_complete()
+
+def edit_contact(edited_contact):
+    """Edit the information of the opened contact.
+
+    Keyword arguments:
+    edited_contact -- The contact with the edited information.
+
+    """
+    link_edit = get_element(css_class='edit-button')
+    element_click(link_edit)
+    wait_for_action_to_complete()
+    _fill_contact(edited_contact)
+    link_save = get_elements(css_class='save-button')[0]
+    element_click(link_save)
+    wait_for_action_to_complete()
+
+def _fill_contact(contact):
+    """Fill the contact information.
+
+    Keyword arguments:
+    contact -- The contact with the information that will be filled.
+
+    """
+    if contact.name:
+        _fill_contact_name(contact.name)
+    _fill_contact_emails(contact.emails)
+    _fill_contact_phones(contact.phones)
+    _fill_contact_addresses(contact.addresses)
+    _fill_contact_instant_messaging_addresses(
+        contact.instant_messaging_addresses)
+    _fill_contact_urls(contact.urls)
+    if contact.role:
+        _fill_contact_role(contact.role)
+    if contact.about_information:
+        _fill_contact_about_information(contact.about_information)
+
+def _fill_contact_name(name):
+    """Fill the contact name information.
+
+    Keyword arguments:
+    name -- The name of the contact.
+
+    """
+    if name.title:
+        textfield_write('names-title', name.title)
+    if name.first_name:
+        textfield_write('names-first_name', name.first_name)
+    if name.middle_name:
+        textfield_write('names-middle_name', name.middle_name)
+    if name.last_name:
+        textfield_write('names-last_name', name.last_name)
+    if name.suffix:
+        textfield_write('names-suffix', name.suffix)
+    
+def _fill_contact_emails(emails):
+    """Fill the contact emails information.
+
+    Keyword arguments:
+    emails -- A sequence with the emails of the contact.
+
+    """
+    for index, email in enumerate(emails):
+        set_select('%d-email_addresses-description' % index, 
+                        email.description)
+        textfield_write('%d-email_addresses-address' % index, email.address)
+        element_click('add-email-address')
+
+def _fill_contact_phones(phones):
+    """Fill the contact phones information.
+
+    Keyword arguments:
+    phones -- A sequence with the phones of the contact.
+
+    """
+    for index, phone in enumerate(phones):
+        set_select('%d-phone_numbers-description' % index, phone.description)
+        textfield_write('%d-phone_numbers-number' % index, phone.number)
+        element_click('add-phone-number')
+        
+def _fill_contact_addresses(addresses):
+    """Fill the contact addresses information.
+
+    Keyword arguments:
+    addresses -- A sequence with the addresses of the contact.
+    
+    """
+    for index, address in enumerate(addresses):
+        set_select('%d-addresses-description' % index, address.description)
+        textfield_write('%d-addresses-address1' % index, address.address1)
+        textfield_write('%d-addresses-address2' % index, address.address2)
+        textfield_write('%d-addresses-pobox' % index, address.po_box)
+        textfield_write('%d-addresses-city' % index, address.city)
+        textfield_write('%d-addresses-state' % index, address.state)
+        textfield_write('%d-addresses-postalcode' % index, address.postal_code)
+        textfield_write('%d-addresses-country' % index, address.country)
+        element_click('add-postal-address')
+
+def _fill_contact_instant_messaging_addresses(instant_messaging_addresses):
+    """Fill the contact instant messaging information.
+
+    Keyword arguments:
+    instant_messaging_addresses -- A sequence of instant messaging addresses.
+
+    """
+    for index, address in enumerate(instant_messaging_addresses):
+        set_select('%d-im_addresses-description' % index, address.description)
+        textfield_write('%d-im_addresses-address' % index, address.address)
+        element_click('add-im-address')
+
+def _fill_contact_urls(urls):
+    """Fill the contact URLs information.
+
+    Keyword arguments:
+    urls -- A sequence with the URLs of the contact.
+
+    """
+    for index, url in enumerate(urls):
+        textfield_write('%d-urls-description' % index, url.description)
+        textfield_write('%d--urls-address' % index, url.url)
+        element_click('add-url')
+
+def _fill_contact_role(role):
+    """Fill the contact role information.
+
+    Keyword arguments:
+    role -- The job role information of the contact.
+
+    """
+    if role.company:
+        textfield_write('work-company', role.company)
+    if role.department:
+        textfield_write('work-department', role.department)
+    if role.job_title:
+        textfield_write('work-job_title', role.job_title)
+    if role.manager_name:
+        textfield_write('work-manager_name', role.manager_name)
+    if role.assitant_name:
+        textfield_write('work-assistant_name', role.assistant_name)
+
+def _fill_contact_about_information(about_information):
+    """Fill the additional information about the contact.
+
+    Keyword arguments:
+    about_information -- The additional information about the contact.
+
+    """
+    if about_information.nickname:
+        textfield_write('personal-nick_name', about_information.nickname)
+    if about_information.birth_date:
+        textfield_write('personal-birth_date', about_information.birth_date)
+    if about_information.spouse_name:
+        textfield_write('personal-spouse_name', about_information.spouse_name)
+    if about_information.wedding_date:
+        textfield_write('personal-wedding_date', 
+                        about_information.wedding_date)
+    if about_information.notes:
+        textfield_write('personal-notes', about_information.notes)
+
+def assert_contact_information(contact):
+    """Assert that the information of the displayed contact is the expected.
+
+    Keyword arguments:
+    contact -- The expected contact information.
+
+    """
+    _assert_contact_name(contact.name)
+    _assert_contact_emails(contact.emails)
+    contact_identifier = get_contact_identifier()
+    _assert_contact_in_list(contact_identifier, contact)
+    # TODO _assert_contact_phones(contact.phones)
+    # TODO _assert_contact_addresses(contact.addresses)
+    # TODO _assert_contact_instant_messaging_addresses(
+    # contact.instant_messaging_addresses)
+    # TODO _assert_contact_urls(contact.urls)
+    # TODO _assert_contact_role(contact.role)
+    # TODO _assert_contact_about_information(contact.about_information)
+
+def get_contact_identifier():
+    """Get the identifier of the opened contact."""
+    button_edit = get_element_by_css('.edit-button')
+    return button_edit.get_attribute('id')
+
+def _get_element_contact_record(identifier):
+    """Get the contact record element from the contacts list.
+
+    Keyword arguments:
+    identifier -- The identifier of the contact.
+    
+    """
+    return get_element(id=identifier, css_class='contact-record')
+
+def _assert_contact_in_list(contact_identifier, contact_information):
+    """Assert that the contact appears on the right place on the contacts list.
+
+    Keyword arguments:
+    contact_identifier -- The identifier of the contact.
+    contact_information -- The expected contact information.
+
+    """
+    # TODO to assert that the contact is under the right letter we need it to
+    # have separate divs for each letter.
+    # first_letter = _get_contact_letter_in_list(contact.name)
+    display_name = _get_contact_display_name_in_list(contact_information.name)
+    element_contact = _get_element_contact_record(contact_identifier)
+    element_name = element_contact.find_element_by_css_selector('.name a')
+    text_is(element_name, display_name)
+    if contact_information.emails:
+        element_email = element_contact.find_element_by_css_selector('.email')
+        text_is(element_email, contact_information.emails[0].address)
+
+def _get_contact_display_name_in_list(name):
+    """Get the expected name to display in the contacts list.
+
+    Keyword arguments:
+    name -- The name information of the contact.
+
+    """
+    if name and (name.last_name or name.first_name):
+        names = []
+        if name.last_name:
+            names.append(name.last_name)
+        if name.first_name:
+            names.append(name.first_name)
+        display_name = ', '.join(names)
+    else:
+        display_name = 'New contact'
+    return display_name
+
+def _get_contact_letter_in_list(name):
+    """Get the letter under which it's expected to find the contact in the 
+    list.
+
+    Keyword arguments:
+    name -- The name information of the contact.
+
+    """
+    if name and (name.last_name or name.first_name):
+        if name.last_name:
+            first_letter = name.last_name[0]
+        else:
+            first_letter = name.first_name[0]
+    else:
+        first_letter = '0' # TODO ask if this is the correct behavior.
+
+def _assert_contact_name(name):
+    """Assert that the name of the displayed contact is the expected.
+
+    Keyword arguments:
+    name -- The name information of the contact.
+
+    """
+    if not name:
+        display_name = 'New contact'
+    else:
+        names = []
+        if name.title:
+            names.append(name.title)
+        if name.first_name:
+            names.append(name.first_name)
+        if name.middle_name:
+            names.append(name.middle_name)
+        if name.last_name:
+            names.append(name.last_name)
+        if name.suffix:
+            names.append(name.suffix)
+        display_name = ' '.join(names)
+    label = get_element_by_css('#names>h1')
+    text_is(label, display_name)
+
+def _assert_contact_emails(emails):
+    """Assert that the email addresses of the displayed contact are the 
+    expected.
+
+    Keyword arguments:
+    emails -- A sequence with the expected email addresses.
+
+    """
+    elements_email_descriptions = get_elements_by_css('#email_addresses dt')
+    elements_email_addresses = get_elements_by_css('#email_addresses a')
+    for index, email in enumerate(emails):
+        text_is(elements_email_descriptions[index], '%s:' % email.description)
+        text_is(elements_email_addresses[index], email.address)
+
+def delete_contact():
+    """Delete the opened contact."""
+    contact_identifier = get_contact_identifier()
+    element_contact = _get_element_contact_record(contact_identifier)
+    element_contact_checkbox = element_contact. \
+        find_element_by_css_selector('.contact-delete-checkbox')
+    checkbox_set(element_contact_checkbox, True)
+    element_click('delete-selected')
+    alert_accept('Delete selected contacts?')
+    wait_for_action_to_complete()
+
+def assert_contact_exists(identifier):
+    """Assert that a contact exists.
+
+    Keyword arguments:
+    identifier -- The identifier of the contact.
+    """
+    _get_element_contact_record(identifier)
 
=== added file 'ubuntuone/web/tests/sst/shared/actions/loading.py'
--- ubuntuone/web/tests/sst/shared/actions/loading.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/shared/actions/loading.py	2011-10-19 06:57:26 +0000
@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2011 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/>.
+
+"""Actions to wait for data loading to complete."""
+
+from sst.actions import *
+
+def wait_for_action_to_complete():
+    """Wait for the last action to complete.
+
+    This function waits for the waiting bar to disappear.
+    
+    """
+    # Added a little sleep because if this function is called to fast before
+    # the bar appears tests will fail.
+    sleep(3)
+    waiting_bar = get_element(css_class='yui3-busy-waiting')
+    waitfor(fails, is_displayed, waiting_bar)
 
=== added directory 'ubuntuone/web/tests/sst/shared/data'
=== added file 'ubuntuone/web/tests/sst/shared/data/__init__.py'
=== added file 'ubuntuone/web/tests/sst/shared/data/contact.py'
--- ubuntuone/web/tests/sst/shared/data/contact.py	1970-01-01 00:00:00 +0000
+++ ubuntuone/web/tests/sst/shared/data/contact.py	2011-10-19 06:57:26 +0000
@@ -0,0 +1,204 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2011 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/>.
+
+"""Data classes for a Contact."""
+
+class Contact:
+    "Class for Contact information."
+
+    def __init__(self, name=None, emails=(), phones=(), addresses=(), 
+                 instant_messaging_addresses=(), urls=(), role=None, 
+                 about_information=None):
+        """Class constructor.
+
+        Keyword arguments:
+        name -- The name of the contact. Default is None.
+        emails -- A sequence with the emails of the contact. Default is an 
+            empty tuple.
+        phones -- A sequence with the phones of the contact. Default is an 
+            empty tuple.
+        addresses -- A sequence with the addresses of the contac. Default is 
+            an empty tuple.
+        instant_messaging_addresses -- A sequence with the instant messaging
+            addresses of the contact.  Default is an empty tuple.
+        urls -- A sequence with the URLs of the contact.  Default is an empty 
+            tuple.
+        role -- The job role of the contact. Default is None.
+        about_information -- Additional information of the contact. Default is
+           None.
+
+        """
+        self.name = name
+        self.emails = emails
+        self.phones = phones
+        self.addresses = addresses
+        self.instant_messaging_addresses = instant_messaging_addresses
+        self.urls = urls
+        self.role = role
+        self.about_information = about_information
+        
+class Name:
+    """Class for the name of a contact."""
+
+    def __init__(self, title=None, first_name=None, middle_name=None, 
+                 last_name=None, suffix=None):
+        """Class constructor.
+
+        Keyword arguments:
+        title -- The title of the contact. Default is None.
+        first_name -- The first name of the contact. Default is None.
+        middle_name -- The middle name of the contact. Default is None.
+        last_name -- The last name of the contact. Default is None.
+        suffix -- The suffix of the contact. Default is None.
+
+        """
+        self.title = title
+        self.first_name = first_name
+        self.middle_name = middle_name
+        self.last_name = last_name
+        self.suffix = suffix
+
+class Email:
+    """Class for the email of a contact."""
+
+    def __init__(self, description, address):
+        """Class constructor.
+    
+        Keyword arguments:
+        description -- The description of the email. It can be 'Other',
+            'Home' or Work.
+        address -- The email address.
+
+        """
+        self.description = description
+        self.address = address
+
+class Phone:
+    """Class for the phone of a contact."""
+
+    def __init__(self, description, number):
+        """Class constructor.
+
+        Keyword arguments:
+        description -- The description of the phone. It can be 'Home', 
+            'Mobile', 'Home fax', 'Work', 'Car', 'Pager', 'Work fax', 
+            'Assitant', 'Call back', 'Company', 'Other fax', 'Primary', 
+            'Telex' or 'Other'.
+        number -- The phone number.
+
+        """
+        self.description = description
+        self.number = number
+
+class Address:
+    """Class for the address of a contact."""
+
+    def __init__(self, description, address1=None, address2=None, po_box=None,
+                 city=None, state=None, postal_code=None, country=None):
+        """Class constructor.
+
+        Keyword arguments:
+        description -- The description of the address. It can be 'Home', 
+            'Work' or 'Other'.
+        address1 -- The first line of the address. Default is None.
+        address2 -- The second line of the address. Default is None.
+        po_box -- The post office box number. Default is None.
+        city -- The name of the city. Default is None.
+        state -- The name of the state. Default is None.
+        postal_code -- The postal code number. Default is None.
+        country -- The name of the country. Default is None.
+
+        """
+        self.description = description
+        self.address1 = address1
+        self.address2 = address2
+        self.po_box = po_box
+        self.city = city
+        self.state = state
+        self.postal_code = postal_code
+        self.country = country
+
+class InstantMessagingAddress:
+    """Class for the Instant Messaging address of a contact."""
+
+    def  __init__(self, description, address):
+        """Class constructor.
+
+        Keyword arguments:
+        description -- The description of the instan messaging address. It can
+            be 'AIM', 'Gadu-Gadu', 'GroupWise', 'ICQ', 'IRC', 'Jabber', 'MSN', 
+            'Skype', 'Yahoo', or 'Other'.
+        address -- The instant messaging address.
+
+        """
+        self.description = description
+        self.address = address
+
+class Url:
+    """Class for the Universal Resource Locators of a contact."""
+    
+    def __init__(self, description, url):
+        """Class constructor.
+
+        Keyword arguments:
+        description -- The description of the URL.
+        url -- The Universal Resource Locator.
+        
+        """
+        self.description = description
+        self.url = url
+
+class Role:
+    """Class for the role of a contact."""
+
+    def __init__(self, company=None, department=None, job_title=None, 
+                 manager_name=None, assistant_name=None):
+        """Class constructor.
+
+        Keyword arguments:
+        company -- The name of the company. Default is None.
+        department -- The name of the department. Default is None.
+        job_title -- The name of the job possition. Default is None.
+        manager_name -- The name of the manager. Default is None.
+        assitant_name -- The name of the assistant. Default is None.
+
+        """
+        self.company = company
+        self.department = department
+        self.job_title = job_title
+        self.manager_name = manager_name
+        self.assitant_name = assitant_name
+    
+class AboutInformation:
+    """Class for additional information about the contact."""
+    
+    def __init__(self, nickname=None, birth_date=None, spouse_name=None, 
+                 wedding_date=None, notes=None):
+        """Class constructor.
+
+        Keyword arguments:
+        nickname -- The nickname. Default is None.
+        birth_date -- The date of birth. Default is None.
+        spouse_name -- The name of the spouse. Default is None.
+        wedding_date -- The date of the wedding. Default is None.
+        notes -- Additional notes about the contact. Default is None.
+
+        """
+        self.nickname = nickname
+        self.birth_date = birth_date
+        self.spouse_name = spouse_name
+        self.wedding_date = wedding_date
+        self.notes = notes