diff --git a/monkey/tests/unit_tests/monkey_island/cc/services/reporting/exploitations/test_monkey_exploitation.py b/monkey/tests/unit_tests/monkey_island/cc/services/reporting/exploitations/test_monkey_exploitation.py index cdee46ba1..f40e09c62 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/services/reporting/exploitations/test_monkey_exploitation.py +++ b/monkey/tests/unit_tests/monkey_island/cc/services/reporting/exploitations/test_monkey_exploitation.py @@ -9,12 +9,16 @@ from monkey_island.cc.services.reporting.exploitations.monkey_exploitation impor ) -def test_get_exploits_used_on_node(): +def test_get_exploits_used_on_node__2_exploits(): exploits = get_exploits_used_on_node(NODE_DICT) assert sorted(exploits) == sorted(["Elastic Groovy Exploiter", "Drupal Server Exploiter"]) + +def test_get_exploits_used_on_node__duplicate_exploits(): exploits = get_exploits_used_on_node(NODE_DICT_DUPLICATE_EXPLOITS) assert exploits == ["Drupal Server Exploiter"] + +def test_get_exploits_used_on_node__failed(): exploits = get_exploits_used_on_node(NODE_DICT_FAILED_EXPLOITS) assert exploits == []