mirror of
https://github.com/ghndrx/prowler.git
synced 2026-02-10 14:55:00 +00:00
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
import os
|
|
|
|
from lib.check.check import 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(checks_file):
|
|
test_cases = [
|
|
{
|
|
"name": "Test valid check path",
|
|
"input": f"{os.path.dirname(os.path.realpath(__file__))}/fixtures/checklistA.txt",
|
|
"expected": {"check12", "check11", "extra72", "check13"},
|
|
},
|
|
{
|
|
"name": "Test valid check path",
|
|
"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"]
|