Haketilo

# SPDX-License-Identifier: GPL-3.0-or-later

"""

Common fixtures for Haketilo unit tests

"""

# This file is part of Haketilo.

#

# Copyright (C) 2021 Wojtek Kosior <koszko@koszko.org>

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# 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

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <https://www.gnu.org/licenses/>.

#

# I, Wojtek Kosior, thereby promise not to sue for violation of this file's

import pytest

from selenium.webdriver.support import expected_conditions as EC

from .server             import do_an_internet

from .extension_crafting import make_extension

from .world_wide_library import start_serving_script, dump_scripts

from .misc_constants     import here

    httpd = do_an_internet()

    yield httpd

    httpd.shutdown()

        yield driver

        driver.quit()

    while len(driver.window_handles) > 1:

        driver.switch_to.window(driver.window_handles[-1])

        driver.close()

    driver.switch_to.window(driver.window_handles[0])

@pytest.fixture()

def driver(_driver, request):

    nav_target = request.node.get_closest_marker('get_page')

    close_all_but_one_window(_driver)

    _driver.get(nav_target.args[0] if nav_target else 'about:blank')

def webextension(driver, request):

    ext_data = request.node.get_closest_marker('ext_data')

    if ext_data is None:

        raise Exception('"webextension" fixture requires "ext_data" marker to be set')

    navigate_to = ext_data.get('navigate_to')

    if navigate_to is not None:

        del ext_data['navigate_to']

    driver.get('https://gotmyowndoma.in/')

    WebDriverWait(driver, 10).until(

    if navigate_to is not None:

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

        driver.get(testpage_url.replace('testpage.html', navigate_to))

    driver.get('https://gotmyowndoma.in/')

    ext_path.unlink()

def haketilo(driver):

    addon_id = driver.install_addon(str(here.parent / 'mozilla-build.zip'),

                                    temporary=True)

    yield

    driver.uninstall_addon(addon_id)

 * Selenium by default executes scripts in some weird one-time context. We want

 * separately-loaded scripts to be able to access global variables defined

 * before, including those declared with `const` or `let`. To achieve that, we

 * javascript served by our proxy. We use custom properties of the `window`

 * object to communicate with injected code.

    delete window.haketilo_selenium_return_value;

    delete window.haketilo_selenium_exception;

    window.returnval = val => window.haketilo_selenium_return_value = val;

    const injectee = document.createElement('script');

    injectee.src = arguments[0];

    injectee.type = "application/javascript";

    injectee.async = true;

    const prom = new Promise(cb => injectee.onload = cb);

    window.arguments = arguments[1];

    document.body.append(injectee);

    await prom;

    /*

     * To ease debugging, we want this script to signal all exceptions from the

     * injectee.

     */

               'Error in injected script! Check your geckodriver.log and ./injected_scripts/!'];

}

return inject();

           result[0] == 'haketilo_selenium_error':

        return result

# consider making them into webdriver wrapper class methods.

def execute_in_page(driver):

    def do_execute(script, *args):

    yield do_execute

def wait_elem_text(driver):

    def do_wait(id, text):

        WebDriverWait(driver, 10).until(

            EC.text_to_be_present_in_element((By.ID, id), text)

        )

    yield do_wait