1
2import pytest
4
5
6pytest_plugins = ['pytest_userver.plugins.core']
7
8
9@pytest.fixture(name='tcp_service_port', scope='session')
10def _tcp_service_port(service_config) -> int:
11 components = service_config['components_manager']['components']
12 tcp_hello = components.get('tcp-hello')
13 assert tcp_hello, 'No "tcp-hello" component found'
14 return int(tcp_hello['port'])
15
16
17@pytest.fixture(scope='session')
18def service_non_http_health_checks(
19 service_config, tcp_service_port,
20) -> net.HealthChecks:
21 checks = net.get_health_checks_info(service_config)
22 checks.tcp.append(net.HostPort(host='localhost', port=tcp_service_port))
23 return checks
24