Agent: Add unit tests for ransomware report service

monkey/tests/unit_tests/monkey_island/cc/services/test_ransomware_report.py

@@ -0,0 +1,37 @@
+import pytest
+
+import monkey_island.cc.services.ransomware_report as ransomware_report
+from monkey_island.cc.services.reporting.report import ReportService
+
+TEST_SCANNED_RESULTS = [{}, {}, {}, {}]
+TEST_EXPLOITED_RESULTS = [
+    {"exploits": ["SSH Exploiter"]},
+    {"exploits": ["SSH Exploiter", "SMB Exploiter"]},
+    {"exploits": ["WMI Exploiter"]},
+]
+
+
+@pytest.fixture(scope="function", autouse=True)
+def patch_report_service(monkeypatch):
+    monkeypatch.setattr(ReportService, "get_scanned", lambda: TEST_SCANNED_RESULTS)
+    monkeypatch.setattr(ReportService, "get_exploited", lambda: TEST_EXPLOITED_RESULTS)
+
+
+def test_get_propagation_stats__num_scanned():
+    stats = ransomware_report.get_propagation_stats()
+
+    assert stats["num_scanned_nodes"] == 4
+
+
+def test_get_propagation_stats__num_exploited():
+    stats = ransomware_report.get_propagation_stats()
+
+    assert stats["num_exploited_nodes"] == 3
+
+
+def test_get_propagation_stats__count_per_exploit():
+    stats = ransomware_report.get_propagation_stats()
+
+    assert stats["count_per_exploit"]["SSH Exploiter"] == 2
+    assert stats["count_per_exploit"]["SMB Exploiter"] == 1
+    assert stats["count_per_exploit"]["WMI Exploiter"] == 1