Hydrilla

# SPDX-License-Identifier: CC0-1.0

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

#

# Available under the terms of Creative Commons Zero v1.0 Universal.

import sys

from pathlib import Path

from typing import Iterable

sys.path.insert(0, str(here / 'src'))

@pytest.fixture(autouse=True)

def no_requests(monkeypatch):

    """Remove requests.sessions.Session.request for all tests."""

        monkeypatch.delattr('requests.sessions.Session.request')

@pytest.fixture

def mock_subprocess_run(monkeypatch, request):

    """

    Temporarily replace subprocess.run() with a function supplied through pytest

    marker 'subprocess_run'.

    The marker excepts 2 arguments:

    * the module inside which the subprocess attribute should be mocked and

    * a run() function to use.

    """

    where, mocked_run = request.node.get_closest_marker('subprocess_run').args

    class MockedSubprocess:

        """Minimal mocked version of the subprocess module."""

        run = mocked_run

    monkeypatch.setattr(where, 'subprocess', MockedSubprocess)

@pytest.fixture(autouse=True)

def no_gettext(monkeypatch, request):

    """

    Make gettext return all strings untranslated unless we request otherwise.

    """

    if request.node.get_closest_marker('enable_gettext'):

        return

    import hydrilla

    modules_to_process = [hydrilla]

    def add_child_modules(parent):

        """

        Recursuvely collect all modules descending from 'parent' into an array.

        """

        try:

            load_paths = parent.__path__

        except AttributeError:

            return

        for module_info in pkgutil.iter_modules(load_paths):

            if module_info.name != '__main__':

                __import__(f'{parent.__name__}.{module_info.name}')

                modules_to_process.append(getattr(parent, module_info.name))

                add_child_modules(getattr(parent, module_info.name))

    add_child_modules(hydrilla)

    for module in modules_to_process:

        if hasattr(module, '_'):

            monkeypatch.setattr(module, '_', lambda message: message)

@pytest.fixture

def tmpdir() -> Iterable[Path]:

    """

    Provide test case with a temporary directory that will be automatically

    deleted after the test.

    """

    with TemporaryDirectory() as tmpdir:

        yield Path(tmpdir)