conftest.py

# /// [service_non_http_health_checker]
import asyncio

import pytest
from pytest_userver.utils import net

pytest_plugins = ['pytest_userver.plugins.core']


@pytest.fixture(name='tcp_service_port', scope='session')
def _tcp_service_port(service_config) -> int:
    components = service_config['components_manager']['components']
    tcp_hello = components.get('tcp-echo')
    assert tcp_hello, 'No "tcp-echo" component found'
    return int(tcp_hello['port'])


@pytest.fixture(scope='session')
def service_non_http_health_checks(
    service_config,
    tcp_service_port,
) -> net.HealthChecks:
    checks = net.get_health_checks_info(service_config)
    checks.tcp.append(net.HostPort(host='localhost', port=tcp_service_port))
    return checks
    # /// [service_non_http_health_checker]