add tests still wip

This commit is contained in:
pedrooot
2024-05-27 12:02:19 +02:00
parent 2cefd32b5d
commit b474247758
+71
View File
@@ -0,0 +1,71 @@
from typer.testing import CliRunner
from cli.cli import (
app, # Asegúrate de que "your_module" sea el nombre del módulo donde está definido tu código
)
runner = CliRunner()
def test_banner_show():
result = runner.invoke(app, ["banner", "--show"])
assert result.exit_code == 0
assert "Banner is not shown." not in result.output
def test_banner_no_show():
result = runner.invoke(app, ["banner", "--no-show"])
assert result.exit_code == 0
assert "Banner is not shown." in result.output
def test_list_services_aws():
result = runner.invoke(app, ["aws", "list-services"])
assert result.exit_code == 0
assert "available services." in result.output
def test_list_fixers_aws():
result = runner.invoke(app, ["aws", "list-fixers"])
assert result.exit_code == 0
assert "available fixers." in result.output
def test_list_categories_aws():
result = runner.invoke(app, ["aws", "list-categories"])
assert result.exit_code == 0
assert "available categories." in result.output
def test_list_compliance_aws():
result = runner.invoke(app, ["aws", "list-compliance"])
assert result.exit_code == 0
assert "available Compliance Frameworks." in result.output
def test_list_compliance_requirements_aws():
result = runner.invoke(
app, ["aws", "list-compliance-requirements", "cis_2.0_aws", "soc2_aws"]
)
assert result.exit_code == 0
assert "Listing CIS 2.0 AWS Compliance Requirements:" in result.output
assert "Listing SOC2 AWS Compliance Requirements:" in result.output
def test_list_compliance_requirements_no_compliance_aws():
result = runner.invoke(app, ["aws", "list-compliance-requirements"])
assert result.exit_code == 0
assert "No compliance frameworks specified." in result.output
def test_list_checks_aws():
result = runner.invoke(app, ["aws", "list-checks"])
assert result.exit_code == 0
assert "available checks." in result.output
def test_list_checks_json_aws():
result = runner.invoke(app, ["aws", "list-checks-json"])
assert result.exit_code == 0
assert "aws" in result.output
assert result.output.startswith("{") and result.output.endswith("}\n")