mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-04 07:34:44 +00:00
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
import contextlib
|
|
import subprocess
|
|
from typing import Callable, Optional, Union
|
|
|
|
import pytest
|
|
import pytest_docker.plugin
|
|
|
|
|
|
def is_responsive(container_name: str, port: int, hostname: Optional[str]) -> bool:
|
|
"""A cheap way to figure out if a port is responsive on a container"""
|
|
if hostname:
|
|
cmd = f"docker exec {container_name} /bin/bash -c 'echo -n > /dev/tcp/{hostname}/{port}'"
|
|
else:
|
|
# use the hostname of the container
|
|
cmd = f"docker exec {container_name} /bin/bash -c 'c_host=`hostname`;echo -n > /dev/tcp/$c_host/{port}'"
|
|
ret = subprocess.run(
|
|
cmd,
|
|
shell=True,
|
|
)
|
|
return ret.returncode == 0
|
|
|
|
|
|
def wait_for_port(
|
|
docker_services: pytest_docker.plugin.Services,
|
|
container_name: str,
|
|
container_port: int,
|
|
hostname: Optional[str] = None,
|
|
timeout: float = 30.0,
|
|
pause: float = 0.5,
|
|
checker: Optional[Callable[[], bool]] = None,
|
|
) -> None:
|
|
try:
|
|
docker_services.wait_until_responsive(
|
|
timeout=timeout,
|
|
pause=pause,
|
|
check=checker
|
|
if checker
|
|
else lambda: is_responsive(container_name, container_port, hostname),
|
|
)
|
|
finally:
|
|
# use check=True to raise an error if command gave bad exit code
|
|
subprocess.run(f"docker logs {container_name}", shell=True, check=True)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def docker_compose_runner(
|
|
docker_compose_command, docker_compose_project_name, docker_setup, docker_cleanup
|
|
):
|
|
@contextlib.contextmanager
|
|
def run(
|
|
compose_file_path: Union[str, list], key: str
|
|
) -> pytest_docker.plugin.Services:
|
|
with pytest_docker.plugin.get_docker_services(
|
|
docker_compose_command=docker_compose_command,
|
|
docker_compose_file=compose_file_path,
|
|
docker_compose_project_name=f"{docker_compose_project_name}-{key}",
|
|
docker_setup=docker_setup,
|
|
docker_cleanup=docker_cleanup,
|
|
) as docker_services:
|
|
yield docker_services
|
|
|
|
return run
|