Haketilo

# SPDX-License-Identifier: CC0-1.0

"""

Haketilo unit tests - entire settings page

"""

# 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

from .utils import *

from selenium.webdriver.support.ui import WebDriverWait

from selenium.webdriver.support import expected_conditions as EC

from selenium.webdriver.common.by import By

from ..extension_crafting import ExtraHTML

from ..script_loader import load_script

from .utils import *

ext_data = {

    'background_script': broker_js,

    'extra_html': ExtraHTML('html/settings.html', wrap_into_htmldoc=False)

}

@pytest.mark.ext_data(ext_data)

@pytest.mark.usefixtures('webextension')

def test_settings_page_tabs(driver, execute_in_page):

    """

    Test navigation through the tabs of the settings page.

    """

    # First, put some sample data in IndexedDB.

    execute_in_page(load_script('common/indexeddb.js'))

    execute_in_page(

        '''

        initial_data = arguments[0];

        returnval(get_db().then(() => {}));

        ''',

        make_complete_sample_data())

    # Now navigate to settings page.

    testpage_url = driver.execute_script('return window.location.href;')

    driver.get(testpage_url.replace('testpage.html', 'html/settings.html'))

    execute_in_page('init_settings_page();')

    WebDriverWait(driver, 10)\

        .until(EC.visibility_of_element_located((By.ID, "main_view")))

    assert driver.find_elements_by_id('loader') == []

    assert not driver.find_element_by_id('indexeddb_error').is_displayed()

    names = ['blocking', 'mappings', 'resources', 'new_payload', 'repos']

    tabs = dict([(n, driver.find_element_by_id(f'{n}_tab')) for n in names])

    heads = dict([(n, driver.find_element_by_id(f'{n}_head')) for n in names])

    for i, tab_name in enumerate(['new_payload', *names]):

        if (i > 0):

            heads[tab_name].click()

        assert 'active_head' in heads[tab_name].get_attribute('class')

        assert 'active_tab' in tabs[tab_name].get_attribute('class')

        assert tabs[tab_name].is_displayed()

        for tab_name_2 in [n for n in names if n != tab_name]:

            assert 'active_head' not in heads[tab_name_2].get_attribute('class')

            assert 'active_tab' not in tabs[tab_name_2].get_attribute('class')

            assert not tabs[tab_name_2].is_displayed()

@pytest.mark.ext_data({

    **ext_data,

    'navigate_to': 'html/settings.html'

})

@pytest.mark.usefixtures('webextension')

@pytest.mark.parametrize('incognito', [True, False])

def test_settings_page_indexeddb_error(driver, execute_in_page, incognito):

    """

    Test if failure to access IndexedDB in settings page results in the

    appropriate message being shown.

    """

    execute_in_page(

        '''{

        /*

         * Mock an unavailable IndexedDB. Calling onerror() without having set

         * "errorCode" on the request is the behavior observed under Mozilla.

         */

        indexedDB.open = function() {

            const dummy_open_request = {};

            const dummy_event = {target: dummy_open_request};

            setTimeout(() => dummy_open_request.onerror(dummy_event), 0);

            return dummy_open_request;

        }

        const dummy_tab = {incognito: arguments[0]};

        browser.tabs.getCurrent = () => Promise.resolve(dummy_tab);

        init_settings_page();

        }''',

        incognito)

    WebDriverWait(driver, 10)\

        .until(EC.visibility_of_element_located((By.ID, 'indexeddb_error')))

    assert driver.find_elements_by_id('loader') == []

    assert not driver.find_element_by_id('main_view').is_displayed()

    if incognito:

        assert driver.find_element_by_id('indexeddb_private_mode_explanation')\

                     .is_displayed()