47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from app.services.processing import ProcessingOptions, filter_records, sort_records
|
|
|
|
|
|
def test_filter_records_supports_case_insensitive_filters():
|
|
records = [
|
|
{"policy": "ProdPolicy", "severity_level": "HIGH"},
|
|
{"policy": "OtherPolicy", "severity_level": "low"},
|
|
]
|
|
options = ProcessingOptions(
|
|
policy_cs="",
|
|
policy_ci="prod",
|
|
severity_cs="",
|
|
severity_ci="high",
|
|
sort_by="datetime",
|
|
order="asc",
|
|
mode="vendor",
|
|
)
|
|
|
|
filtered = filter_records(records, options)
|
|
|
|
assert filtered == [{"policy": "ProdPolicy", "severity_level": "HIGH"}]
|
|
|
|
|
|
def test_sort_records_by_severity_desc_uses_defined_ranking():
|
|
records = [
|
|
{"severity_level": "medium"},
|
|
{"severity_level": "critical"},
|
|
{"severity_level": "info"},
|
|
]
|
|
options = ProcessingOptions(
|
|
policy_cs="",
|
|
policy_ci="",
|
|
severity_cs="",
|
|
severity_ci="",
|
|
sort_by="severity",
|
|
order="desc",
|
|
mode="vendor",
|
|
)
|
|
|
|
sorted_records = sort_records(records, options)
|
|
|
|
assert [record["severity_level"] for record in sorted_records] == [
|
|
"critical",
|
|
"medium",
|
|
"info",
|
|
]
|