diff --git a/monkey/tests/unit_tests/monkey_island/cc/services/attack/technique_reports/test_technique_reports.py b/monkey/tests/unit_tests/monkey_island/cc/services/attack/technique_reports/test_technique_reports.py
index 6436d6df3..67ef6af0d 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/services/attack/technique_reports/test_technique_reports.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/services/attack/technique_reports/test_technique_reports.py
@@ -16,7 +16,7 @@ FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE = {
}
-class FakeAttackTechnique(AttackTechnique):
+class FakeAttackTechnique_TwoRelevantSystems(AttackTechnique):
tech_id = "T0001"
relevant_systems = ["System 1", "System 2"]
unscanned_msg = "UNSCANNED"
@@ -27,7 +27,7 @@ class FakeAttackTechnique(AttackTechnique):
pass
-EXPECTED_UNSCANNED_MSG = (
+EXPECTED_UNSCANNED_MSG_TWO_RELEVANT_SYSTEMS = (
"UNSCANNED due to one of the following reasons:\n"
"- The following configuration options were disabled:
"
"- Definition Type 1 — Config Option 1
"
@@ -35,61 +35,10 @@ EXPECTED_UNSCANNED_MSG = (
)
-EXPECTED_SCANNED_MSG = "SCANNED"
+EXPECTED_SCANNED_MSG_TWO_RELEVANT_SYSTEMS = "SCANNED"
-EXPECTED_USED_MSG = "USED"
-
-
-def test_get_message_by_status_disabled(monkeypatch):
- monkeypatch.setattr(
- (
- "monkey_island.cc.services.attack.technique_reports."
- "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
- ),
- FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
- )
- technique_msg = FakeAttackTechnique.get_message_by_status(ScanStatus.DISABLED.value)
- assert technique_msg == disabled_msg
-
-
-def test_get_message_by_status_unscanned(monkeypatch):
- monkeypatch.setattr(
- (
- "monkey_island.cc.services.attack.technique_reports."
- "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
- ),
- FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
- )
- technique_msg = FakeAttackTechnique.get_message_by_status(ScanStatus.UNSCANNED.value)
- assert technique_msg == EXPECTED_UNSCANNED_MSG
-
-
-def test_get_message_by_status_scanned(monkeypatch):
- monkeypatch.setattr(
- (
- "monkey_island.cc.services.attack.technique_reports."
- "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
- ),
- FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
- )
- technique_msg = FakeAttackTechnique.get_message_by_status(ScanStatus.SCANNED.value)
- assert technique_msg == EXPECTED_SCANNED_MSG
-
-
-def test_get_message_by_status_used(monkeypatch):
- monkeypatch.setattr(
- (
- "monkey_island.cc.services.attack.technique_reports."
- "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
- ),
- FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
- )
- technique_msg = FakeAttackTechnique.get_message_by_status(ScanStatus.USED.value)
- assert technique_msg == EXPECTED_USED_MSG
-
-
-###
+EXPECTED_USED_MSG_TWO_RELEVANT_SYSTEMS = "USED"
class FakeAttackTechnique_OneRelevantSystem(AttackTechnique):
@@ -118,6 +67,65 @@ EXPECTED_SCANNED_MSG_ONE_RELEVANT_SYSTEM = "SCANNED"
EXPECTED_USED_MSG_ONE_RELEVANT_SYSTEM = "USED"
+def test_get_message_by_status_disabled_two_relevant_systems(monkeypatch):
+ monkeypatch.setattr(
+ (
+ "monkey_island.cc.services.attack.technique_reports."
+ "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
+ ),
+ FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
+ )
+ technique_msg = FakeAttackTechnique_TwoRelevantSystems.get_message_by_status(
+ ScanStatus.DISABLED.value
+ )
+ assert technique_msg == disabled_msg
+
+
+def test_get_message_by_status_unscanned_two_relevant_systems(monkeypatch):
+ monkeypatch.setattr(
+ (
+ "monkey_island.cc.services.attack.technique_reports."
+ "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
+ ),
+ FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
+ )
+ technique_msg = FakeAttackTechnique_TwoRelevantSystems.get_message_by_status(
+ ScanStatus.UNSCANNED.value
+ )
+ assert technique_msg == EXPECTED_UNSCANNED_MSG_TWO_RELEVANT_SYSTEMS
+
+
+def test_get_message_by_status_scanned_two_relevant_systems(monkeypatch):
+ monkeypatch.setattr(
+ (
+ "monkey_island.cc.services.attack.technique_reports."
+ "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
+ ),
+ FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
+ )
+ technique_msg = FakeAttackTechnique_TwoRelevantSystems.get_message_by_status(
+ ScanStatus.SCANNED.value
+ )
+ assert technique_msg == EXPECTED_SCANNED_MSG_TWO_RELEVANT_SYSTEMS
+
+
+def test_get_message_by_status_used_two_relevant_systems(monkeypatch):
+ monkeypatch.setattr(
+ (
+ "monkey_island.cc.services.attack.technique_reports."
+ "__init__.CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE"
+ ),
+ FAKE_CONFIG_SCHEMA_PER_ATTACK_TECHNIQUE,
+ )
+ technique_msg = FakeAttackTechnique_TwoRelevantSystems.get_message_by_status(
+ ScanStatus.USED.value
+ )
+ assert technique_msg == EXPECTED_USED_MSG_TWO_RELEVANT_SYSTEMS
+
+
+###
+
+
def test_get_message_by_status_disabled_one_relevant_system(monkeypatch):
monkeypatch.setattr(
(