from pathlib import Path
import subprocess


def capture(command):
    proc = subprocess.Popen(
        command,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
    )
    out, err = proc.communicate()
    return out, err, proc.returncode


def test_podman_compose_extends_w_file_subdir():
    """
    Test that podman-compose can execute podman-compose -f <file> up with extended File which
    includes a build context
    :return:
    """
    main_path = Path(__file__).parent.parent

    command_up = [
        "coverage",
        "run",
        str(main_path.joinpath("podman_compose.py")),
        "-f",
        str(main_path.joinpath("tests", "extends_w_file_subdir", "docker-compose.yml")),
        "up",
        "-d",
    ]

    command_check_container = [
        "podman",
        "container",
        "ps",
        "--sort",
        "status",
        "--all",
        "--format",
        '"{{.Image}}"',
    ]

    command_down = [
        "podman",
        "rmi",
        "--force",
        "localhost/subdir_test:me",
        "docker.io/library/busybox",
    ]

    out, _, returncode = capture(command_up)
    assert 0 == returncode
    # check container was created and exists
    out, _, returncode = capture(command_check_container)
    assert 0 == returncode
    assert b'"localhost/subdir_test:me"\n' in out
    out, _, returncode = capture(command_down)
    # cleanup test image(tags)
    assert 0 == returncode
    print('ok')
    # check container did not exists anymore
    out, _, returncode = capture(command_check_container)
    print(out)
    assert 0 == returncode
    assert b'"localhost/subdir_test:me"\n' not in out


def test_podman_compose_extends_w_empty_service():
    """
    Test that podman-compose can execute podman-compose -f <file> up with extended File which
    includes an empty service. (e.g. if the file is used as placeholder for more complex configurations.)
    :return:
    """
    main_path = Path(__file__).parent.parent

    command_up = [
        "python3",
        str(main_path.joinpath("podman_compose.py")),
        "-f",
        str(
            main_path.joinpath("tests", "extends_w_empty_service", "docker-compose.yml")
        ),
        "up",
        "-d",
    ]

    _, _, returncode = capture(command_up)
    assert 0 == returncode