From c7a1c9d21d09cb10ad1af02cfbe3bd2a166666c1 Mon Sep 17 00:00:00 2001 From: Ilija Lazoroski Date: Fri, 19 Nov 2021 15:04:36 +0100 Subject: [PATCH] UT: Remove testing disabled attack techniques --- .../test_technique_reports.py | 19 +------------------ .../setup/mongo/test_database_initializer.py | 3 --- 2 files changed, 1 insertion(+), 21 deletions(-) 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 1da9860b9..e408a3aff 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 @@ -3,10 +3,7 @@ from enum import Enum import pytest from common.utils.attack_utils import ScanStatus -from monkey_island.cc.services.attack.technique_reports.__init__ import ( - AttackTechnique, - disabled_msg, -) +from monkey_island.cc.services.attack.technique_reports.__init__ import AttackTechnique @pytest.fixture(scope="function", autouse=True) @@ -68,13 +65,6 @@ class ExpectedMsgs_OneRelevantSystem(Enum): USED: str = "USED" -def test_get_message_by_status_disabled_two_relevant_systems(): - 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(): technique_msg = FakeAttackTechnique_TwoRelevantSystems.get_message_by_status( ScanStatus.UNSCANNED.value @@ -96,13 +86,6 @@ def test_get_message_by_status_used_two_relevant_systems(): assert technique_msg == ExpectedMsgs_TwoRelevantSystems.USED.value -def test_get_message_by_status_disabled_one_relevant_system(): - technique_msg = FakeAttackTechnique_OneRelevantSystem.get_message_by_status( - ScanStatus.DISABLED.value - ) - assert technique_msg == disabled_msg - - def test_get_message_by_status_unscanned_one_relevant_system(): technique_msg = FakeAttackTechnique_OneRelevantSystem.get_message_by_status( ScanStatus.UNSCANNED.value diff --git a/monkey/tests/unit_tests/monkey_island/cc/setup/mongo/test_database_initializer.py b/monkey/tests/unit_tests/monkey_island/cc/setup/mongo/test_database_initializer.py index d3ca3fbcc..4afdbc969 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/setup/mongo/test_database_initializer.py +++ b/monkey/tests/unit_tests/monkey_island/cc/setup/mongo/test_database_initializer.py @@ -22,9 +22,6 @@ def patch_dependencies(monkeypatch_session): monkeypatch_session.setattr( "monkey_island.cc.services.config.ConfigService.init_config", lambda: None ) - monkeypatch_session.setattr( - "monkey_island.cc.services.attack.attack_config.AttackConfig.reset_config", lambda: None - ) monkeypatch_session.setattr( "monkey_island.cc.services.database.jsonify", MagicMock(return_value=True) )