simono41/compose-viz
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

test: update test_cli

Browse source
This commit is contained in:
Xyphuz 2022-05-18 22:47:54 +08:00
parent 04535bb834
commit 16768590e4
1 changed files with 73 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

78
tests/test_cli.py
View file

@ -1,4 +1,5 @@
import os
import pytest
from typer.testing import CliRunner
from compose_viz import cli
@ -6,11 +7,78 @@ from compose_viz import cli
runner = CliRunner()
def test_cli():
input_path = "tests/in/000001.yaml"
result = runner.invoke(cli.app, [input_path])
@pytest.mark.parametrize("file_number", [
"000001",
"000010",
"000011",
"000100",
"000101",
"000110",
"000111",
"001000",
"001001",
"001010",
"001011",
"001100",
"001101",
"001110",
"001111",
"010000",
"010001",
"010010",
"010011",
"010100",
"010101",
"010110",
"010111",
"011000",
"011001",
"011010",
"011011",
"011100",
"011101",
"011110",
"011111",
"100000",
"100001",
"100010",
"100011",
"100100",
"100101",
"100110",
"100111",
"101000",
"101001",
"101010",
"101011",
"101100",
"101101",
"101110",
"101111",
"110000",
"110001",
"110010",
"110011",
"110100",
"110101",
"110110",
"110111",
"111000",
"111001",
"111010",
"111011",
"111100",
"111101",
"111110",
"111111",
])
def test_cli(file_number: str):
input_path = f"tests/in/{file_number}.yaml"
output_path = f"{file_number}.png"
result = runner.invoke(cli.app, ["-o", output_path, input_path])
assert result.exit_code == 0
assert f"Successfully parsed {input_path}\n" in result.stdout
assert os.path.exists("compose-viz.png")
assert os.path.exists(output_path)
os.remove(output_path)