import os
from typing import List, Optional

from ci.ray_ci.linux_container import LinuxContainer
from ci.ray_ci.tester_container import TesterContainer


class LinuxTesterContainer(TesterContainer, LinuxContainer):
    def __init__(
        self,
        docker_tag: str,
        shard_count: int = 1,
        gpus: int = 0,
        network: Optional[str] = None,
        test_envs: Optional[List[str]] = None,
        shard_ids: Optional[List[int]] = None,
        skip_ray_installation: bool = False,
        build_type: Optional[str] = None,
        python_version: Optional[str] = None,
        install_mask: Optional[str] = None,
        tmp_filesystem: Optional[str] = None,
        privileged: bool = False,
    ) -> None:
        LinuxContainer.__init__(
            self,
            docker_tag,
            envs=test_envs,
            volumes=[
                f"{os.environ.get('RAYCI_CHECKOUT_DIR')}:/ray-mount",
                "/var/run/docker.sock:/var/run/docker.sock",
            ],
            python_version=python_version,
            tmp_filesystem=tmp_filesystem,
            privileged=privileged,
        )
        TesterContainer.__init__(
            self,
            shard_count,
            gpus,
            bazel_log_dir="/tmp/bazel_event_logs",
            network=network,
            test_envs=test_envs,
            shard_ids=shard_ids,
            skip_ray_installation=skip_ray_installation,
            build_type=build_type,
            install_mask=install_mask,
        )