Fixed unit test that gets mitigations from API

This commit is contained in:
VakarisZ 2020-04-01 09:52:38 +03:00
parent f6782b72fa
commit 483a3576d4
1 changed files with 2 additions and 2 deletions

View File

@ -7,8 +7,8 @@ class TestMitreApiInterface(TestCase):
def test_get_all_mitigations(self): def test_get_all_mitigations(self):
mitigations = MitreApiInterface.get_all_mitigations() mitigations = MitreApiInterface.get_all_mitigations()
self.assertTrue((len(mitigations) >= 282)) self.assertTrue((len(mitigations.items()) >= 282))
mitigation = mitigations[0] mitigation = next(iter(mitigations.values()))
self.assertEqual(mitigation['type'], "course-of-action") self.assertEqual(mitigation['type'], "course-of-action")
self.assertTrue(mitigation['name']) self.assertTrue(mitigation['name'])
self.assertTrue(mitigation['description']) self.assertTrue(mitigation['description'])