1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
# 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/.
from marionette_driver import By
from marionette_driver.errors import NoSuchElementException
from firefox_puppeteer.base import BaseLib
from firefox_puppeteer.ui.base import DOMElement
class MenuBar(BaseLib):
"""Wraps the menubar DOM element inside a browser window."""
@property
def menus(self):
"""A list of :class:`MenuElement` instances corresponding to
the top level menus in the menubar.
:returns: A list of :class:`MenuElement` instances.
"""
menus = (self.marionette.find_element(By.ID, 'main-menubar')
.find_elements(By.TAG_NAME, 'menu'))
return [self.MenuElement(menu) for menu in menus]
def get_menu(self, label):
"""Get a :class:`MenuElement` instance corresponding to the specified label.
:param label: The label of the menu, e.g., **File** or **View**.
:returns: A :class:`MenuElement` instance.
"""
menu = [m for m in self.menus if m.get_attribute('label') == label]
if not menu:
raise NoSuchElementException('Could not find a menu with '
'label "{}"'.format(label))
return menu[0]
def get_menu_by_id(self, menu_id):
"""Get a :class:`MenuElement` instance corresponding to the specified
ID.
:param menu_id: The ID of the menu, e.g., **file-menu** or **view-menu**.
:returns: A :class:`MenuElement` instance.
"""
menu = [m for m in self.menus if m.get_attribute('id') == menu_id]
if not menu:
raise NoSuchElementException('Could not find a menu with '
'id "{}"'.format(menu_id))
return menu[0]
def select(self, label, item):
"""Select an item in a menu.
:param label: The label of the menu, e.g., **File** or **View**.
:param item: The label of the item in the menu, e.g., **New Tab**.
"""
return self.get_menu(label).select(item)
def select_by_id(self, menu_id, item_id):
"""Select an item in a menu.
:param menu_id: The ID of the menu, e.g. **file-menu** or **view-menu**.
:param item_id: The ID of the item in the menu, e.g. **menu_newNavigatorTab**.
"""
return self.get_menu_by_id(menu_id).select_by_id(item_id)
class MenuElement(DOMElement):
"""Wraps a menu element DOM element."""
@property
def items(self):
"""A list of menuitem DOM elements within this :class:`MenuElement` instance.
:returns: A list of items in the menu.
"""
return (self.find_element(By.TAG_NAME, 'menupopup')
.find_elements(By.TAG_NAME, 'menuitem'))
def select(self, label):
"""Click on a menu item within this menu.
:param label: The label of the menu item, e.g., **New Tab**.
"""
item = [l for l in self.items if l.get_attribute('label') == label]
if not item:
message = ("Item labeled '{}' not found in the '{}' menu"
.format(label, self.get_attribute('label')))
raise NoSuchElementException(message)
return item[0].click()
def select_by_id(self, menu_item_id):
"""Click on a menu item within this menu.
:param menu_item_id: The ID of the menu item, e.g. **menu_newNavigatorTab**.
"""
item = [l for l in self.items if l.get_attribute('id') ==
menu_item_id]
if not item:
message = ("Item with ID '{}' not found in the '{}' menu"
.format(menu_item_id, self.get_attribute('id')))
raise NoSuchElementException(message)
return item[0].click()
|
