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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
|
# SPDX-License-Identifier: CC0-1.0
"""
Haketilo unit tests - repository querying
"""
# This file is part of Haketilo
#
# Copyright (C) 2022 Wojtek Kosior <koszko@koszko.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the CC0 1.0 Universal License as published by
# the Creative Commons Corporation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# CC0 1.0 Universal License for more details.
import pytest
import json
from selenium.webdriver.support.ui import WebDriverWait
from ..extension_crafting import ExtraHTML
from ..script_loader import load_script
from .utils import *
def reload_with_target(driver, target):
current_url = driver.execute_script('return location.href')
driver.execute_script(
'''
window.location.href = arguments[0];
window.location.reload();
''',
f'{current_url}#{target}')
unprivileged_page_info = {
'url': 'https://example_a.com/something',
'allow': False
}
mocked_page_infos = {
'privileged': {
'url': 'moz-extension://<some-id>/file.html',
'privileged': True
},
'blocked_default': unprivileged_page_info,
'allowed_default': {
**unprivileged_page_info,
'allow': True
},
'blocked_rule': {
**unprivileged_page_info,
'mapping': '~allow'
},
'allowed_rule': {
**unprivileged_page_info,
'allow': True,
'mapping': '~allow'
},
'mapping': {
**unprivileged_page_info,
'mapping': 'm1',
'payload': {'identifier': 'res1'}
},
'error': {
**unprivileged_page_info,
'error': True
}
}
tab_mock_js = '''
;
const mocked_page_info = (%s)[/#mock_page_info-(.*)$/.exec(document.URL)[1]];
browser.tabs.sendMessage = async function(tab_id, msg) {
const this_tab_id = (await browser.tabs.getCurrent()).id;
if (tab_id !== this_tab_id)
throw `not current tab id (${tab_id} instead of ${this_tab_id})`;
if (msg[0] === "page_info") {
return mocked_page_info;
} else if (msg[0] === "repo_query") {
const response = await fetch(msg[1]);
if (!response)
return {error: "Something happened :o"};
const result = {ok: response.ok, status: response.status};
try {
result.json = await response.json();
} catch(e) {
result.error_json = "" + e;
}
return result;
} else {
throw `bad sendMessage message type: '${msg[0]}'`;
}
}
const old_tabs_query = browser.tabs.query;
browser.tabs.query = async function(query) {
const tabs = await old_tabs_query(query);
tabs.forEach(t => t.url = mocked_page_info.url);
return tabs;
}
''' % json.dumps(mocked_page_infos)
popup_ext_data = {
'background_script': broker_js,
'extra_html': ExtraHTML(
'html/popup.html',
{
'common/browser.js': tab_mock_js,
'common/indexeddb.js': '; set_repo("https://hydril.la/");'
},
wrap_into_htmldoc=False
),
'navigate_to': 'html/popup.html'
}
@pytest.mark.ext_data(popup_ext_data)
@pytest.mark.usefixtures('webextension')
@pytest.mark.parametrize('page_info_key', ['', *mocked_page_infos.keys()])
def test_popup_display(driver, execute_in_page, page_info_key):
"""
Test popup viewing while on a page. Test parametrized with different
possible values of page_info object passed in message from the content
script.
"""
reload_with_target(driver, f'mock_page_info-{page_info_key}')
by_id = driver.execute_script('''
const nodes = [...document.querySelectorAll("[id]")];
return nodes.reduce((ob, node) => Object.assign(ob, {[node.id]: node}), {});
''');
if page_info_key == '':
error_msg = 'Page info not avaialable. Try reloading the page.'
error_msg_shown = lambda d: by_id['loading_info'].text == error_msg
WebDriverWait(driver, 10).until(error_msg_shown)
return
WebDriverWait(driver, 10).until(lambda d: by_id['info_form'].is_displayed())
assert (page_info_key == 'privileged') == \
by_id['privileged_page_info'].is_displayed()
assert (page_info_key == 'privileged') ^ \
by_id['unprivileged_page_info'].is_displayed()
assert by_id['page_url'].text == mocked_page_infos[page_info_key]['url']
assert not by_id['repo_query_container'].is_displayed()
if 'blocked' in page_info_key or page_info_key in ('mapping', 'error'):
assert by_id['scripts_blocked'].text.lower() == 'yes'
elif 'allowed' in page_info_key:
assert by_id['scripts_blocked'].text.lower() == 'no'
if page_info_key == 'mapping':
assert by_id['injected_payload'].text == 'res1'
elif page_info_key != 'privileged':
assert by_id['injected_payload'].text == 'None'
mapping_text = by_id['mapping_used'].text
if page_info_key == 'mapping':
assert mapping_text == 'm1'
if 'allowed' in page_info_key:
'None (scripts allowed by' in mapping_text
elif 'blocked' in page_info_key:
'None (scripts blocked by' in mapping_text
if 'rule' in page_info_key:
'by a rule)' in mapping_text
elif 'default' in page_info_key:
'by default_policy)' in mapping_text
if page_info_key == 'error':
assert mapping_text == 'None (error occured when determining policy)'
@pytest.mark.ext_data(popup_ext_data)
@pytest.mark.usefixtures('webextension')
def test_popup_repo_query(driver, execute_in_page):
"""
Test opening and closing the repo query view in popup.
"""
reload_with_target(driver, f'mock_page_info-blocked_rule')
search_but = driver.find_element_by_id("search_resources_but")
WebDriverWait(driver, 10).until(lambda d: search_but.is_displayed())
search_but.click()
containers = dict([(name, driver.find_element_by_id(f'{name}_container'))
for name in ('page_info', 'repo_query')])
assert not containers['page_info'].is_displayed()
assert containers['repo_query'].is_displayed()
shown = lambda d: 'https://hydril.la/' in containers['repo_query'].text
WebDriverWait(driver, 10).until(shown)
# Click the "Show results" button.
selector = '.repo_query_buttons > button:first-child'
driver.find_element_by_css_selector(selector).click()
shown = lambda d: 'MAPPING_A' in containers['repo_query'].text
WebDriverWait(driver, 10).until(shown)
# Click the "Cancel" button
selector = '.repo_query_bottom_buttons > button'
driver.find_element_by_css_selector(selector).click()
assert containers['page_info'].is_displayed()
assert not containers['repo_query'].is_displayed()
@pytest.mark.ext_data(popup_ext_data)
@pytest.mark.usefixtures('webextension')
def test_popup_settings_opening(driver, execute_in_page):
"""
Test opening the settings page from popup through button click.
"""
driver.find_element_by_id("settings_but").click()
first_handle = driver.current_window_handle
WebDriverWait(driver, 10).until(lambda d: len(d.window_handles) == 2)
new_handle = [h for h in driver.window_handles if h != first_handle][0]
driver.switch_to.window(new_handle)
driver.implicitly_wait(10)
assert "Extension's options page for testing" in \
driver.find_element_by_tag_name("h1").text
|
