import pytest

from testcontainers.core.container import DockerContainer

from testcontainers.core.docker_client import DockerClient

from testcontainers.core.config import ConnectionMode

FAKE_ID = "ABC123"

class FakeContainer:

@property

def id(self) -> str:

return FAKE_ID

@pytest.fixture

def container(monkeypatch: pytest.MonkeyPatch) -> DockerContainer:

"""

Fake initialized container

"""

client = DockerClient()

container = DockerContainer("foobar")

monkeypatch.setattr(container, "_docker", client)

monkeypatch.setattr(container, "_container", FakeContainer())

return container

@pytest.mark.parametrize("mode", ["docker_host", "gateway_ip", "bridge_ip"])

def test_get_container_host_ip(container: DockerContainer, monkeypatch: pytest.MonkeyPatch, mode: str) -> None:

"""

Depending on the connection mode the correct function is executed to the host_ip

"""

connection_mode = ConnectionMode(mode)

def result_fake(result: str, require_container_id):

def fake_for_mode(*container_id: str):

if require_container_id:

assert len(container_id) == 1

assert container_id[0] == FAKE_ID

else:

assert len(container_id) == 0

return result

return fake_for_mode

client = container._docker

monkeypatch.setattr(client, "get_connection_mode", lambda: connection_mode)

monkeypatch.setattr(client, "gateway_ip", result_fake("gateway_ip", True))

monkeypatch.setattr(client, "bridge_ip", result_fake("bridge_ip", True))

monkeypatch.setattr(client, "host", result_fake("docker_host", False))

assert container.get_container_host_ip() == mode

@pytest.mark.parametrize("mode", [ConnectionMode.gateway_ip, ConnectionMode.docker_host])

def test_get_exposed_port_mapped(

container: DockerContainer, monkeypatch: pytest.MonkeyPatch, mode: ConnectionMode

) -> None:

def fake_mapped(container_id: int, port: int) -> int:

assert container_id == FAKE_ID

assert port == 8080

return 45678

client = container._docker

monkeypatch.setattr(client, "port", fake_mapped)

monkeypatch.setattr(client, "get_connection_mode", lambda: mode)

assert container.get_exposed_port(8080) == 45678

def test_get_exposed_port_original(container: DockerContainer, monkeypatch: pytest.MonkeyPatch) -> None:

client = container._docker

monkeypatch.setattr(client, "get_connection_mode", lambda: ConnectionMode.bridge_ip)

assert container.get_exposed_port(8080) == 8080