compose-viz/tests/test_cli.py
2022-05-26 11:10:19 +08:00

35 lines
1 KiB
Python

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}")