compose-viz/tests/test_cli.py

import os

import pytest
from typer.testing import CliRunner

from compose_viz import cli

runner = CliRunner()


@pytest.mark.parametrize(
    "test_file_path",
    [
        "tests/ymls/builds/docker-compose.yml",
        "tests/ymls/depends_on/docker-compose.yml",
        "tests/ymls/extends/docker-compose.yml",
        "tests/ymls/links/docker-compose.yml",
        "tests/ymls/networks/docker-compose.yml",
        "tests/ymls/ports/docker-compose.yml",
        "tests/ymls/volumes/docker-compose.yml",
        "examples/full-stack-node-app/docker-compose.yml",
        "examples/non-normative/docker-compose.yml",
    ],
)
def test_cli(test_file_path: str) -> None:
    input_path = f"{test_file_path}"
    output_filename = "compose-viz-test"
    default_format = "png"
    result = runner.invoke(cli.app, ["-o", output_filename, input_path])

    assert result.exit_code == 0
    assert f"Successfully parsed {input_path}\n" in result.stdout
    assert os.path.exists(f"{output_filename}.{default_format}")

    os.remove(f"{output_filename}.{default_format}")