Skip to content

Commit e181c96

Browse files
committed
add unit test for highest_ssvc_decision method
Signed-off-by: tdruez <tdruez@aboutcode.org>
1 parent e9c9122 commit e181c96

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

vulnerabilities/tests/test_models.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -435,3 +435,16 @@ def test_vulnerability_model_risk_level_generated_field(self):
435435
vulnerability1.save()
436436
vulnerability1.refresh_from_db()
437437
self.assertEqual("critical", vulnerability1.risk_level)
438+
439+
def test_vulnerability_highest_ssvc_decision(self):
440+
vulnerability1 = make_vulnerability(self.dataspace)
441+
self.assertIsNone(vulnerability1.highest_ssvc_decision)
442+
443+
vulnerability1.ssvc_trees = [{"decision": "Track"}]
444+
self.assertEqual("Track", vulnerability1.highest_ssvc_decision)
445+
446+
vulnerability1.ssvc_trees = [{"decision": "Track"}, {"decision": "Act"}]
447+
self.assertEqual("Act", vulnerability1.highest_ssvc_decision)
448+
449+
vulnerability1.ssvc_trees = [{"decision": "Attend"}, {"decision": "Track*"}]
450+
self.assertEqual("Attend", vulnerability1.highest_ssvc_decision)

0 commit comments

Comments
 (0)