Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
SeleniumHQ
GitHub Repository: SeleniumHQ/Selenium
 Path: blob/trunk/py/test/selenium/webdriver/firefox/ff_installs_addons_tests.py
4111 views
1
# Licensed to the Software Freedom Conservancy (SFC) under one

2
# or more contributor license agreements.  See the NOTICE file

3
# distributed with this work for additional information

4
# regarding copyright ownership.  The SFC licenses this file

5
# to you under the Apache License, Version 2.0 (the

6
# "License"); you may not use this file except in compliance

7
# with the License.  You may obtain a copy of the License at

8
#

9
#   http://www.apache.org/licenses/LICENSE-2.0

10
#

11
# Unless required by applicable law or agreed to in writing,

12
# software distributed under the License is distributed on an

13
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

14
# KIND, either express or implied.  See the License for the

15
# specific language governing permissions and limitations

16
# under the License.

17


18
import os

19
import zipfile

20


21
import pytest

22


23
from selenium.webdriver.common.by import By

24
from selenium.webdriver.support.wait import WebDriverWait

25


26
from conftest import get_extensions_location

27


28
EXTENSIONS = get_extensions_location()

29


30


31
pytestmark = pytest.mark.xfail_remote(reason="Remote WebDriver does not expose Firefox-specific addon APIs")

32


33


34
@pytest.mark.no_driver_after_test

35
def test_install_uninstall_signed_addon_xpi(driver, pages):

36
    extension = os.path.join(EXTENSIONS, "webextensions-selenium-example.xpi")

37


38
    id = driver.install_addon(extension)

39
    assert id == "[email protected]"

40


41
    pages.load("blank.html")

42
    injected = WebDriverWait(driver, timeout=2).until(

43
        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")

44
    )

45
    assert injected.text == "Content injected by webextensions-selenium-example"

46


47
    driver.uninstall_addon(id)

48
    driver.refresh()

49
    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0

50


51


52
@pytest.mark.no_driver_after_test

53
def test_install_uninstall_signed_addon_zip(driver, pages):

54
    extension = os.path.join(EXTENSIONS, "webextensions-selenium-example.zip")

55


56
    id = driver.install_addon(extension)

57
    assert id == "[email protected]"

58


59
    pages.load("blank.html")

60
    injected = WebDriverWait(driver, timeout=2).until(

61
        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")

62
    )

63
    assert injected.text == "Content injected by webextensions-selenium-example"

64


65
    driver.uninstall_addon(id)

66
    driver.refresh()

67
    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0

68


69


70
@pytest.mark.no_driver_after_test

71
def test_install_uninstall_unsigned_addon_zip(driver, pages):

72
    extension = os.path.join(EXTENSIONS, "webextensions-selenium-example-unsigned.zip")

73


74
    id = driver.install_addon(extension, temporary=True)

75
    assert id == "[email protected]"

76


77
    pages.load("blank.html")

78
    injected = WebDriverWait(driver, timeout=2).until(

79
        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")

80
    )

81
    assert injected.text == "Content injected by webextensions-selenium-example"

82


83
    driver.uninstall_addon(id)

84
    driver.refresh()

85
    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0

86


87


88
@pytest.mark.no_driver_after_test

89
def test_install_uninstall_signed_addon_dir(driver, pages):

90
    zip = os.path.join(EXTENSIONS, "webextensions-selenium-example.zip")

91


92
    target = os.path.join(EXTENSIONS, "webextensions-selenium-example-unzip")

93
    with zipfile.ZipFile(zip, "r") as zip_ref:

94
        zip_ref.extractall(target)

95


96
    id = driver.install_addon(target)

97
    assert id == "[email protected]"

98


99
    pages.load("blank.html")

100
    injected = WebDriverWait(driver, timeout=2).until(

101
        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")

102
    )

103
    assert injected.text == "Content injected by webextensions-selenium-example"

104


105
    driver.uninstall_addon(id)

106
    driver.refresh()

107
    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0

108


109


110
@pytest.mark.no_driver_after_test

111
def test_install_uninstall_unsigned_addon_dir(driver, pages):

112
    zip = os.path.join(EXTENSIONS, "webextensions-selenium-example-unsigned.zip")

113
    target = os.path.join(EXTENSIONS, "webextensions-selenium-example-unsigned-unzip")

114
    with zipfile.ZipFile(zip, "r") as zip_ref:

115
        zip_ref.extractall(target)

116


117
    id = driver.install_addon(target, temporary=True)

118
    assert id == "[email protected]"

119


120
    pages.load("blank.html")

121
    injected = WebDriverWait(driver, timeout=2).until(

122
        lambda dr: dr.find_element(By.ID, "webextensions-selenium-example")

123
    )

124
    assert injected.text == "Content injected by webextensions-selenium-example"

125


126
    driver.uninstall_addon(id)

127
    driver.refresh()

128
    assert len(driver.find_elements(By.ID, "webextensions-selenium-example")) == 0

129


130