status-go/tests-functional/conftest.py

import os
from dataclasses import dataclass, field
from typing import List
import pytest


def pytest_addoption(parser):
    parser.addoption(
        "--status_backend_url",
        action="store",
        help="",
        default=None,
    )
    parser.addoption(
        "--anvil_url",
        action="store",
        help="",
        default="http://0.0.0.0:8545",
    )
    parser.addoption(
        "--password",
        action="store",
        help="",
        default="Strong12345",
    )
    parser.addoption(
        "--docker_project_name",
        action="store",
        help="",
        default="tests-functional",
    )
    parser.addoption(
        "--codecov_dir",
        action="store",
        help="",
        default=None,
    )
    parser.addoption(
        "--user_dir",
        action="store",
        help="",
        default=None,
    )


@dataclass
class Option:
    status_backend_port_range: List[int] = field(default_factory=list)
    status_backend_containers: List[str] = field(default_factory=list)
    base_dir: str = ""


option = Option()


def pytest_configure(config):
    global option
    option = config.option

    executor_number = int(os.getenv("EXECUTOR_NUMBER", 5))
    base_port = 7000
    range_size = 100
    max_port = 65535
    min_port = 1024

    start_port = base_port + (executor_number * range_size)
    end_port = start_port + 20000

    # Ensure generated ports are within the valid range
    if start_port < min_port or end_port > max_port:
        raise ValueError(f"Generated port range ({start_port}-{end_port}) is outside the allowed range ({min_port}-{max_port}).")

    option.status_backend_port_range = list(range(start_port, end_port))
    option.status_backend_containers = []

    option.base_dir = os.path.dirname(os.path.abspath(__file__))


@pytest.fixture(scope="function", autouse=True)
def close_status_backend_containers(request):
    yield
    if hasattr(request.node.instance, "reuse_container"):
        return
    for container in option.status_backend_containers:
        container.kill()  # type: ignore