aboutsummaryrefslogtreecommitdiff
path: root/test/unit/test_popup.py
blob: 1fc262ca724feb1c10722308c0ed278e4f652032 (about) (plain)
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
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
# 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 *

unprivileged_page_info = {
    'url': 'https://example_a.com/something',
    'allow': False
}

mapping_page_info = {
    **unprivileged_page_info,
    'mapping': 'm1',
    'payload': {'identifier': 'res1'}
}

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': mapping_page_info,
    'error_deciding_policy': {
        **mapping_page_info,
        'error': {'haketilo_error_type': 'deciding_policy'}
    },
    'error_missing': {
        **mapping_page_info,
        'error': {'haketilo_error_type': 'missing', 'id': 'some-missing-res'}
    },
    'error_circular': {
        **mapping_page_info,
        'error': {'haketilo_error_type': 'circular', 'id': 'some-circular-res'}
    },
    'error_db': {
        **mapping_page_info,
        'error': {'haketilo_error_type': 'db'}
    },
    'error_other': {
        **mapping_page_info,
        'error': {'haketilo_error_type': 'other'}
    }
}

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.
    """
    initial_url = driver.current_url
    driver.get('about:blank')
    driver.get(f'{initial_url}#mock_page_info-{page_info_key}')

    by_id = driver.execute_script(
        '''
        const nodes = [...document.querySelectorAll("[id]")];
        const reductor = (ob, node) => Object.assign(ob, {[node.id]: node});
        return nodes.reduce(reductor, {});
        ''')

    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 'allow' in page_info_key:
        assert by_id['scripts_blocked'].text.lower() == 'no'
    elif page_info_key != 'privileged':
        assert by_id['scripts_blocked'].text.lower() == 'yes'

    payload_text = by_id['injected_payload'].text
    if page_info_key == 'mapping':
        assert payload_text == 'res1'
    elif page_info_key == 'error_missing':
        assert payload_text == \
            "None (error: resource with id 'some-missing-res' missing from the database)"
    elif page_info_key == 'error_circular':
        assert payload_text == \
            "None (error: circular dependency of resource with id 'some-circular-res' on itself)"
    elif page_info_key == 'error_db':
        assert payload_text == \
            'None (error: failure reading Haketilo internal database)'
    elif page_info_key == 'error_other':
        assert payload_text == \
            'None (error: unknown failure occured)'
    elif page_info_key != 'privileged':
        assert payload_text == 'None'

    mapping_text = by_id['mapping_used'].text

    if page_info_key == 'error_deciding_policy':
        assert mapping_text == 'None (error occured when determining policy)'
    elif page_info_key == 'mapping' or page_info_key.startswith('error'):
        assert mapping_text == 'm1'

    if 'allowed' in page_info_key:
        assert 'None (scripts allowed by' in mapping_text
    elif 'blocked' in page_info_key:
        assert 'None (scripts blocked by' in mapping_text

    if 'rule' in page_info_key:
        assert 'by a rule)' in mapping_text
    elif 'default' in page_info_key:
        assert 'by default policy)' in mapping_text

@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.
    """
    initial_url = driver.current_url
    driver.get('about:blank')
    driver.get(f'{initial_url}#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')
# Under Parabola's Iceweasel 75 the settings page's window opened during this
# test is impossible to close using driver.close() - it raises an exception with
# message 'closeTab() not supported in iceweasel'. To avoid such error during
# test cleanup, we use the mark below to tell our driver fixture to span a
# separate browser instance for this test.
@pytest.mark.second_driver()
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