mirror of
https://github.com/ghndrx/prowler.git
synced 2026-02-10 23:05:05 +00:00
8abcc5988d
* feat(checks): Select checks to run * feat(checks): Include tests * feat(checks): Exclude checks with -e * fix(checks): Include missing path * fix(checks): Include comments
62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
import os
|
|
|
|
from lib.check.check import exclude_checks_to_run, parse_checks_from_file
|
|
|
|
|
|
class Test_Check:
|
|
# def test_import_check(self):
|
|
# test_cases = [
|
|
# {
|
|
# "name": "Test valid check path",
|
|
# "input": "providers.aws.services.iam.iam_disable_30_days_credentials.iam_disable_30_days_credentials",
|
|
# "expected": "providers.aws.services.iam.iam_disable_30_days_credentials.iam_disable_30_days_credentials",
|
|
# }
|
|
# ]
|
|
# for test in test_cases:
|
|
# assert importlib.import_module(test["input"]).__name__ == test["expected"]
|
|
|
|
def test_parse_checks_from_file(self):
|
|
test_cases = [
|
|
{
|
|
"input": f"{os.path.dirname(os.path.realpath(__file__))}/fixtures/checklistA.txt",
|
|
"expected": {"check12", "check11", "extra72", "check13"},
|
|
},
|
|
{
|
|
"input": f"{os.path.dirname(os.path.realpath(__file__))}/fixtures/checklistB.txt",
|
|
"expected": {
|
|
"extra72",
|
|
"check13",
|
|
"check11",
|
|
"check12",
|
|
"check56",
|
|
"check2423",
|
|
},
|
|
},
|
|
]
|
|
for test in test_cases:
|
|
assert parse_checks_from_file(test["input"]) == test["expected"]
|
|
|
|
def test_exclude_checks_to_run(self):
|
|
test_cases = [
|
|
{
|
|
"input": {
|
|
"check_list": {"check12", "check11", "extra72", "check13"},
|
|
"excluded_checks": {"check12", "check13"},
|
|
},
|
|
"expected": {"check11", "extra72"},
|
|
},
|
|
{
|
|
"input": {
|
|
"check_list": {"check112", "check11", "extra72", "check13"},
|
|
"excluded_checks": {"check12", "check13", "check14"},
|
|
},
|
|
"expected": {"check112", "check11", "extra72"},
|
|
},
|
|
]
|
|
for test in test_cases:
|
|
check_list = test["input"]["check_list"]
|
|
excluded_checks = test["input"]["excluded_checks"]
|
|
assert (
|
|
exclude_checks_to_run(check_list, excluded_checks) == test["expected"]
|
|
)
|