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

Tests: Convert test_get_all_mitigations() from unittest to pytest

This commit is contained in:
Mike Salvatore 2021-07-19 12:26:32 -04:00
parent ac9bd8dee7
commit f0033d0c7c
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
monkey/tests/unit_tests/monkey_island/cc/services/attack/test_mitre_api_interface.py
View File

@ -1,14 +1,11 @@
from unittest import TestCase
from monkey_island.cc.services.attack.mitre_api_interface import MitreApiInterface
class TestMitreApiInterface(TestCase):
def test_get_all_mitigations(self):
def test_get_all_mitigations():
mitigations = MitreApiInterface.get_all_mitigations()
self.assertIsNotNone((len(mitigations.items()) >= 282))
assert len(mitigations.items()) >= 282
mitigation = next(iter(mitigations.values()))
self.assertEqual(mitigation["type"], "course-of-action")
self.assertIsNotNone(mitigation["name"])
self.assertIsNotNone(mitigation["description"])
self.assertIsNotNone(mitigation["external_references"])
assert mitigation["type"] == "course-of-action"
assert mitigation["name"] is not None
assert mitigation["description"] is not None
assert mitigation["external_references"] is not None