ben.carrasco/prowler
1
0 Fork 0
mirror of https://github.com/prowler-cloud/prowler.git synced 2026-10-30 07:14:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(cli): improve tests

Browse Source
This commit is contained in:
pedrooot committed 2024-06-04 10:43:04 +02:00
1 parent c3975cf0e4
commit 8c28962d12
2 files changed
+20 -15
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/cli/cli_test.py
+11 -10
View File
@@ -10,54 +10,55 @@ runner = CliRunner()
class TestCLI:
def test_list_services_aws(self):
result = runner.invoke(app, ["aws", "list-services"])
result = runner.invoke(app, ["aws", "--list-services"])
assert result.exit_code == 0
assert "available services." in result.output
def test_list_fixers_aws(self):
result = runner.invoke(app, ["aws", "list-fixers"])
result = runner.invoke(app, ["aws", "--list-fixers"])
assert result.exit_code == 0
assert "available fixers." in result.output
def test_list_categories_aws(self):
result = runner.invoke(app, ["aws", "list-categories"])
result = runner.invoke(app, ["aws", "--list-categories"])
assert result.exit_code == 0
assert "available categories." in result.output
def test_list_compliance_aws(self):
result = runner.invoke(app, ["aws", "list-compliance"])
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(self):
result = runner.invoke(
app, ["aws", "list-compliance-requirements", "cis_2.0_aws", "soc2_aws"]
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(self):
result = runner.invoke(app, ["aws", "list-compliance-requirements"])
result = runner.invoke(app, ["aws", "--list-compliance-requirements"])
assert result.exit_code == 2
assert "Expected at least one" in result.output
assert "requires an argument" in result.output
def test_list_compliance_requirements_one_invalid_aws(self):
invalid_name = "invalid"
result = runner.invoke(
app, ["aws", "list-compliance-requirements", "cis_2.0_aws", invalid_name]
app,
["aws", "--list-compliance-requirements", f"cis_2.0_aws,{invalid_name}"],
)
assert result.exit_code == 0
assert "Listing CIS 2.0 AWS Compliance Requirements:" in result.output
assert f"{invalid_name} is not a valid Compliance Framework" in result.output
def test_list_checks_aws(self):
result = runner.invoke(app, ["aws", "list-checks"])
result = runner.invoke(app, ["aws", "--list-checks"])
assert result.exit_code == 0
assert "available checks." in result.output
def test_list_checks_json_aws(self):
result = runner.invoke(app, ["aws", "list-checks-json"])
result = runner.invoke(app, ["aws", "--list-checks-json"])
assert result.exit_code == 0
assert "aws" in result.output
# validate the json output