Installing an add-on using Selenium

test_install.py

# 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 time

from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as expected
from selenium.webdriver.support.ui import WebDriverWait as Wait
import pytest

from pages.desktop.home import Home


@pytest.fixture
def firefox_profile(firefox_profile):
    firefox_profile.set_preference('browser.tabs.remote.force-enable', True)
    firefox_profile.set_preference('ui.popup.disable_autohide', True)
    firefox_profile.update_preferences()
    return firefox_profile


@pytest.mark.nondestructive
@pytest.mark.capabilities(marionette=True)
def test_install_extension(base_url, selenium, firefox_profile, capabilities):
    """Install an extension"""
    page = Home(selenium, base_url).open()
    addon_page = page.most_popular.extensions[0].click()
    addon_page.click_add_to_firefox()
    selenium.set_context('chrome')
    notification = Wait(selenium, 10).until(
        expected.presence_of_element_located((
            By.ID, 'addon-install-confirmation-notification')))
    time.sleep(10)
    # Wait(selenium, 10).until(expected.visibility_of(notification))
    notification.find_element(
        By.ID, 'addon-install-confirmation-accept').click()
    notification = Wait(selenium, 10).until(
        expected.presence_of_element_located((
            By.ID, 'addon-install-complete-notification')))
    # Wait(selenium, 10).until(expected.visibility_of(notification))
    time.sleep(10)
    notification.find_element(
        By.ANON_ATTRIBUTE, {'anonid': 'closebutton'}).click()

Requires pytest, pytest-selenium, and a command line such as:

py.test --driver Firefox \
--driver-path ~/Downloads/wires-0.7.1 \
--capability binary /Applications/FirefoxDeveloperEdition.app/Contents/MacOS/firefox-bin \
--base-url https://addons-dev.allizom.org \
test_install.py