testcontainers-python/tests/test_docker_compose.py at master · SnowMoon-Dev/testcontainers-python · GitHub

History

92 lines (68 loc) · 3.34 KB

Raw

from unittest.mock import patch

import pytest

import subprocess

from testcontainers.compose import DockerCompose

from testcontainers.core.docker_client import DockerClient

from testcontainers.core.exceptions import NoSuchPortExposed

from testcontainers.core.waiting_utils import wait_for_logs

def test_can_spawn_service_via_compose():

with DockerCompose('tests') as compose:

host = compose.get_service_host("hub", 4444)

port = compose.get_service_port("hub", 4444)

assert host == "0.0.0.0"

assert port == "4444"

def test_can_pull_images_before_spawning_service_via_compose():

with DockerCompose("tests", pull=True) as compose:

host = compose.get_service_host("hub", 4444)

port = compose.get_service_port("hub", 4444)

assert host == "0.0.0.0"

assert port == "4444"

def test_can_build_images_before_spawning_service_via_compose():

with patch.object(DockerCompose, "_call_command") as call_mock:

with DockerCompose("tests", build=True) as compose:

...

assert compose.build

docker_compose_cmd = call_mock.call_args_list[0][1]["cmd"]

assert "docker-compose" in docker_compose_cmd

assert "up" in docker_compose_cmd

assert "--build" in docker_compose_cmd

def test_can_throw_exception_if_no_port_exposed():

with DockerCompose("tests") as compose:

with pytest.raises(NoSuchPortExposed):

compose.get_service_host("hub", 5555)

def test_compose_wait_for_container_ready():

with DockerCompose("tests") as compose:

docker = DockerClient()

compose.wait_for("http://%s:4444/wd/hub" % docker.host())

def test_compose_can_wait_for_logs():

with DockerCompose(filepath="tests", compose_file_name="docker-compose-4.yml") as compose:

wait_for_logs(compose, "Hello from Docker!")

def test_can_parse_multiple_compose_files():

with DockerCompose(filepath="tests",

compose_file_name=["docker-compose.yml", "docker-compose-2.yml"]) as compose:

host = compose.get_service_host("mysql", 3306)

port = compose.get_service_port("mysql", 3306)

assert host == "0.0.0.0"

assert port == "3306"

host = compose.get_service_host("hub", 4444)

port = compose.get_service_port("hub", 4444)

assert host == "0.0.0.0"

assert port == "4444"

def test_can_get_logs():

with DockerCompose("tests") as compose:

docker = DockerClient()

compose.wait_for("http://%s:4444/wd/hub" % docker.host())

stdout, stderr = compose.get_logs()

assert stdout, 'There should be something on stdout'

def test_can_pass_env_params_by_env_file():

with DockerCompose('tests', compose_file_name='docker-compose-3.yml',

env_file='.env.test') as _:

check_env_is_set_cmd = 'docker exec tests_mysql_1 printenv | grep TEST_ASSERT_KEY'.split()

out = subprocess.run(check_env_is_set_cmd, stdout=subprocess.PIPE)

assert out.stdout.decode('utf-8').splitlines()[0], 'test_is_passed'

def test_can_exec_commands():

with DockerCompose("tests") as compose:

result = compose.exec_in_container('hub', ['echo', 'my_test'])

assert result[0] == 'my_test\n', "The echo should be successful"

assert result[1] == '', "stderr should be empty"

assert result[2] == 0, 'The exit code should be successful'

testcontainers-python/tests/test_docker_compose.py at master · SnowMoon-Dev/testcontainers-python · GitHub

Expand file tree

Breadcrumbs

test_docker_compose.py

Latest commit

History

Breadcrumbs

test_docker_compose.py

File metadata and controls