p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Tests: Set autouse=False for patch_report_service fixture

This commit is contained in:
Mike Salvatore 2021-07-12 12:08:31 -04:00
parent e4cd06d8c3
commit c7d655ac7d
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/tests/unit_tests/monkey_island/cc/services/ransomware/test_ransomware_report.py
View File

@ -3,6 +3,9 @@ import pytest
from monkey_island.cc.services.ransomware import ransomware_report from monkey_island.cc.services.ransomware import ransomware_report
from monkey_island.cc.services.reporting.report import ReportService from monkey_island.cc.services.reporting.report import ReportService
@pytest.fixture
def patch_report_service_for_stats(monkeypatch):
TEST_SCANNED_RESULTS = [{}, {}, {}, {}] TEST_SCANNED_RESULTS = [{}, {}, {}, {}]
TEST_EXPLOITED_RESULTS = [ TEST_EXPLOITED_RESULTS = [
{"exploits": ["SSH Exploiter"]}, {"exploits": ["SSH Exploiter"]},
@ -10,26 +13,23 @@ TEST_EXPLOITED_RESULTS = [
{"exploits": ["WMI 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_scanned", lambda: TEST_SCANNED_RESULTS)
monkeypatch.setattr(ReportService, "get_exploited", lambda: TEST_EXPLOITED_RESULTS) monkeypatch.setattr(ReportService, "get_exploited", lambda: TEST_EXPLOITED_RESULTS)
def test_get_propagation_stats__num_scanned(): def test_get_propagation_stats__num_scanned(patch_report_service_for_stats):
stats = ransomware_report.get_propagation_stats() stats = ransomware_report.get_propagation_stats()
assert stats["num_scanned_nodes"] == 4 assert stats["num_scanned_nodes"] == 4
def test_get_propagation_stats__num_exploited(): def test_get_propagation_stats__num_exploited(patch_report_service_for_stats):
stats = ransomware_report.get_propagation_stats() stats = ransomware_report.get_propagation_stats()
assert stats["num_exploited_nodes"] == 3 assert stats["num_exploited_nodes"] == 3
def test_get_propagation_stats__num_exploited_per_exploit(): def test_get_propagation_stats__num_exploited_per_exploit(patch_report_service_for_stats):
stats = ransomware_report.get_propagation_stats() stats = ransomware_report.get_propagation_stats()
assert stats["num_exploited_per_exploit"]["SSH Exploiter"] == 2 assert stats["num_exploited_per_exploit"]["SSH Exploiter"] == 2