Island: Rename count_per_exploit -> num_exploited_per_exploit

This commit is contained in:
Mike Salvatore 2021-07-12 11:50:39 -04:00
parent 4564596cd0
commit 06439d92f9
2 changed files with 5 additions and 5 deletions

View File

@ -10,7 +10,7 @@ def get_propagation_stats() -> Dict:
return { return {
"num_scanned_nodes": len(scanned), "num_scanned_nodes": len(scanned),
"num_exploited_nodes": len(exploited), "num_exploited_nodes": len(exploited),
"count_per_exploit": _get_exploit_counts(exploited), "num_exploited_per_exploit": _get_exploit_counts(exploited),
} }

View File

@ -29,9 +29,9 @@ def test_get_propagation_stats__num_exploited():
assert stats["num_exploited_nodes"] == 3 assert stats["num_exploited_nodes"] == 3
def test_get_propagation_stats__count_per_exploit(): def test_get_propagation_stats__num_exploited_per_exploit():
stats = ransomware_report.get_propagation_stats() stats = ransomware_report.get_propagation_stats()
assert stats["count_per_exploit"]["SSH Exploiter"] == 2 assert stats["num_exploited_per_exploit"]["SSH Exploiter"] == 2
assert stats["count_per_exploit"]["SMB Exploiter"] == 1 assert stats["num_exploited_per_exploit"]["SMB Exploiter"] == 1
assert stats["count_per_exploit"]["WMI Exploiter"] == 1 assert stats["num_exploited_per_exploit"]["WMI Exploiter"] == 1