service.py

"""
Start the service in testsuite.
"""
 
# pylint: disable=redefined-outer-name
from collections.abc import Iterable
import logging
import pathlib
import time
import typing
from typing import Any
 
import pytest
 
from testsuite.utils import url_util
 
from pytest_userver.utils import net
 
logger = logging.getLogger(__name__)
 
 
def pytest_addoption(parser) -> None:
    group = parser.getgroup('userver')
    group.addoption(
        '--service-logs-file',
        type=pathlib.Path,
        help='Write service output to specified file',
    )
    group.addoption(
        '--service-logs-pretty',
        action='store_true',
        help='Enable pretty print and colorize service logs',
    )
    group.addoption(
        '--service-logs-pretty-verbose',
        dest='service_logs_pretty',
        action='store_const',
        const='verbose',
        help='Enable pretty print and colorize service logs in verbose mode',
    )
    group.addoption(
        '--service-logs-pretty-disable',
        action='store_false',
        dest='service_logs_pretty',
        help='Disable pretty print and colorize service logs',
    )
 
 
@pytest.fixture(scope='session')
def service_env() -> dict[str, str]:
    """
    Override this to pass extra environment variables to the service.
 
    @snippet samples/redis_service/testsuite/conftest.py  service_env
    @ingroup userver_testsuite_fixtures
    """
    return {}
 
 
@pytest.fixture(scope='session')
async def service_http_ping_url(service_config, service_baseurl) -> str | None:
    """
    Returns the service HTTP ping URL that is used by the testsuite to detect
    that the service is ready to work. Returns None if there's no such URL.
 
    By default, attempts to find server::handlers::Ping component by
    "handler-ping" name in static config. Override this fixture to change the
    behavior.
 
    @ingroup userver_testsuite_fixtures
    """
    components = service_config['components_manager']['components']
    ping_handler = components.get('handler-ping')
    if ping_handler:
        return url_util.join(service_baseurl, ping_handler['path'])
    return None
 
 
@pytest.fixture(scope='session')
def service_non_http_health_checks(  # pylint: disable=invalid-name
    service_config,
) -> net.HealthChecks:
    """
    Returns a health checks info.
 
    By default, returns pytest_userver.utils.net.get_health_checks_info().
 
    Override this fixture to change the way testsuite detects the tested
    service being alive.
 
    @ingroup userver_testsuite_fixtures
    """
 
    return net.get_health_checks_info(service_config)
 
 
@pytest.fixture(scope='session')
def service_start_timeout() -> float:
    """
    Returns service start timeout in seconds.
 
    Override this fixture to change the service start timeout.
 
    @ingroup userver_testsuite_fixtures
    """
 
    return 100.0
 
 
@pytest.fixture(scope='session')
async def service_binary_launcher():
    """
    If non-empty, defines a list of arguments starting with executable
    that is used instead of `service_binary`. The final argument list consists of
    `service_binary_launcher` plus the common service cmdline.
 
    Can be used for:
    - resource limiting (rlimit, cgroup)
    - security limiting (capabilities, LSM)
    - resource profiling (perf)
    - tracing (strace, ltrace)
 
    @see @ref pytest_userver.plugins.service.service_daemon_instance "service_daemon_instance"
    @ingroup userver_testsuite_fixtures
    """
    return []
 
 
@pytest.fixture(scope='session')
async def service_daemon_scope(
    create_daemon_scope,
    daemon_scoped_mark,
    service_env,
    service_http_ping_url,
    service_config_path_temp,
    service_binary,
    service_binary_launcher,
    service_non_http_health_checks,
    service_start_timeout,
):
    """
    Prepares the start of the service daemon.
    Configures the health checking to use service_http_ping_url fixture value
    if it is not None; otherwise uses the service_non_http_health_checks info.
 
    @see @ref pytest_userver.plugins.service.service_daemon_instance "service_daemon_instance"
    @ingroup userver_testsuite_fixtures
    """
    assert service_http_ping_url or service_non_http_health_checks.tcp, (
        '"service_http_ping_url" and "create_health_checker" fixtures '
        'returned None. Testsuite is unable to detect if the service is ready '
        'to accept requests.',
    )
 
    logger.debug(
        'userver fixture "service_daemon_scope" would check for "%s"',
        service_non_http_health_checks,
    )
 
    class LocalCounters:
        last_log_time = 0.0
        attempts = 0
 
    async def _checker(*, session, process) -> bool:
        LocalCounters.attempts += 1
        new_log_time = time.monotonic()
        if new_log_time - LocalCounters.last_log_time > 1.0:
            LocalCounters.last_log_time = new_log_time
            logger.debug(
                'userver fixture "service_daemon_scope" checking "%s", attempt %s',
                service_non_http_health_checks,
                LocalCounters.attempts,
            )
 
        return await net.check_availability(service_non_http_health_checks)
 
    health_check = _checker
    if service_http_ping_url:
        health_check = None
 
    # In yandex-taxi-testsuite, each poll retry duration is 0.05 seconds.
    poll_retries = int(service_start_timeout / 0.05)
 
    async with create_daemon_scope(
        args=service_binary_launcher + [str(service_binary), '--config', str(service_config_path_temp)],
        ping_url=service_http_ping_url,
        health_check=health_check,
        env=service_env,
        poll_retries=poll_retries,
    ) as scope:
        yield scope
 
 
@pytest.fixture
def extra_client_deps() -> None:
    """
    Service client dependencies hook. Feel free to override, e.g.:
 
    @code
    @pytest.fixture
    def extra_client_deps(some_fixtures_to_wait_before_service_start):
        pass
    @endcode
 
    @ingroup userver_testsuite_fixtures
    """
 
 
@pytest.fixture
def auto_client_deps(request) -> None:
    """
    Ensures that the following fixtures, if available, are run before service start:
 
    * `pgsql`
    * `mongodb`
    * `clickhouse`
    * `rabbitmq`
    * kafka (`kafka_producer`, `kafka_consumer`)
    * `redis_store`
    * `mysql`
    * @ref pytest_userver.plugins.ydb.ydbsupport.ydb "ydb"
    * @ref pytest_userver.plugins.grpc.mockserver.grpc_mockserver "grpc_mockserver"
 
    To add other dependencies prefer overriding the
    @ref pytest_userver.plugins.service.extra_client_deps "extra_client_deps"
    fixture.
 
    @ingroup userver_testsuite_fixtures
    """
    known_deps = {
        'pgsql',
        'mongodb',
        'clickhouse',
        'rabbitmq',
        'kafka_producer',
        'kafka_consumer',
        'redis_store',
        'mysql',
        'ydb',
        'grpc_mockserver',
    }
 
    try:
        fixture_lookup_error = pytest.FixtureLookupError
    except AttributeError:
        # support for an older version of the pytest
        import _pytest.fixtures
 
        fixture_lookup_error = _pytest.fixtures.FixtureLookupError
 
    resolved_deps = []
    for dep in known_deps:
        try:
            request.getfixturevalue(dep)
            resolved_deps.append(dep)
        except fixture_lookup_error:
            pass
 
    logger.debug(
        'userver fixture "auto_client_deps" resolved dependencies %s',
        resolved_deps,
    )
 
 
@pytest.fixture
def builtin_client_deps(
    testpoint,
    cleanup_userver_dumps,
    userver_log_capture,
    dynamic_config,
    mock_configs_service,
):
    """
    Service client dependencies hook, like
    @ref pytest_userver.plugins.service.extra_client_deps "extra_client_deps".
 
    Feel free to override globally in a more specific pytest plugin
    (one that comes after userver plugins),
    but make sure to depend on the original fixture:
 
    @code
    @pytest.fixture(name='builtin_client_deps')
    def _builtin_client_deps(builtin_client_deps, some_extra_fixtures):
        pass
    @endcode
 
    @ingroup userver_testsuite_fixtures
    """
 
 
@pytest.fixture
async def service_daemon_instance(
    ensure_daemon_started,
    service_daemon_scope,
    builtin_client_deps,
    auto_client_deps,
    # User defined client deps must be last in order to use
    # fixtures defined above.
    extra_client_deps,
):
    """
    Calls `ensure_daemon_started` on
    @ref pytest_userver.plugins.service.service_daemon_scope "service_daemon_scope"
    to actually start the service. Makes sure that all the dependencies are prepared
    before the service starts.
 
    @see @ref pytest_userver.plugins.service.extra_client_deps "extra_client_deps"
    @see @ref pytest_userver.plugins.service.auto_client_deps "auto_client_deps"
    @see @ref pytest_userver.plugins.service.builtin_client_deps "builtin_client_deps"
    @ingroup userver_testsuite_fixtures
    """
    # TODO also run userver_client_cleanup here
    return await ensure_daemon_started(service_daemon_scope)
 
 
@pytest.fixture(scope='session')
def daemon_scoped_mark(request) -> dict[str, Any] | None:
    """
    Depend on this fixture directly or transitively to make your fixture a per-daemon fixture.
 
    Example:
 
    @code
    @pytest.fixture(scope='session')
    def users_cache_state(daemon_scoped_mark, ...):
        return UsersCacheState(users_list=[])
    @endcode
 
    For tests marked with `@pytest.mark.uservice_oneshot(...)`, the service will be restarted,
    and all the per-daemon fixtures will be recreated.
 
    This fixture returns kwargs passed to the `uservice_oneshot` mark (which may be an empty dict).
    For normal tests, this fixture returns `None`.
 
    @ingroup userver_testsuite_fixtures
    """
    # === How daemon-scoped fixtures work ===
    # pytest always keeps no more than 1 instance of each parametrized fixture.
    # When a parametrized fixture is requested, FixtureDef checks using __eq__ whether the current value
    # of fixture param equals the cached value. If they differ, then the fixture and all the dependent fixtures
    # are torn down (in reverse-dependency order), then the fixture is set up again.
    # TLDR: when the param changes, the whole tree of daemon-specific fixtures is torn down.
    return getattr(request, 'param', None)
 
 
# @cond
 
 
def pytest_configure(config):
    config.addinivalue_line(
        'markers',
        'uservice_oneshot: use a per-test service daemon instance',
    )
 
 
def _contains_oneshot_marker(parametrize: Iterable[pytest.Mark]) -> bool:
    """
    Check if at least one of 'parametrize' marks is of the form:
 
    @pytest.mark.parametrize(
        "foo, bar",
        [
            ("a", 10),
            pytest.param("b", 20, marks=pytest.mark.uservice_oneshot),  # <====
        ]
    )
    """
    return any(
        True
        for parametrize_mark in parametrize
        if len(parametrize_mark.args) >= 2
        for parameter_set in parametrize_mark.args[1]
        if hasattr(parameter_set, 'marks')
        for mark in parameter_set.marks
        if mark.name == 'uservice_oneshot'
    )
 
 
def pytest_generate_tests(metafunc: pytest.Metafunc) -> None:
    oneshot_marker = metafunc.definition.get_closest_marker('uservice_oneshot')
    parametrize_markers = metafunc.definition.iter_markers('parametrize')
    if oneshot_marker is not None or _contains_oneshot_marker(parametrize_markers):
        # Set a dummy parameter value. Actual param is patched in pytest_collection_modifyitems.
        metafunc.parametrize(
            (daemon_scoped_mark.__name__,),
            [(None,)],
            indirect=True,
            # TODO use pytest.HIDDEN_PARAM after it becomes available
            #  https://github.com/pytest-dev/pytest/issues/13228
            ids=['uservice_oneshot'],
            scope='function',
        )
 
 
# TODO use dependent parametrize instead of patching param value after it becomes available
#  https://github.com/pytest-dev/pytest/issues/13233
def pytest_collection_modifyitems(items: list[pytest.Item]):
    for item in items:
        oneshot_marker = item.get_closest_marker('uservice_oneshot')
        if oneshot_marker and isinstance(item, pytest.Function):
            func_item = typing.cast(pytest.Function, item)
            func_item.callspec.params[daemon_scoped_mark.__name__] = dict(oneshot_marker.kwargs, function=func_item)
 
 
# @endcond