def test_get_severity_return_exception(self):
"""Test Severity Procedure. Severity is invalid."""
response_obj = ComponentAnalysisResponseBuilder(self.eco, self.pkg, self.ver)
response_obj._cves = [
{'severity': ['invalid1']},
{'severity': ['invalid2']},
]
with pytest.raises(Exception) as exception:
response_obj.get_severity()
self.assertIs(exception.type, Exception)
def test_get_severity_known_values(self):
"""Test Severity with known values, expect to get medium severity."""
response_obj = ComponentAnalysisResponseBuilder(
self.eco, self.pkg, self.ver)
response_obj._cves = [{'severity': ["medium", "low"]}]
severity = response_obj.get_severity()
self.assertListEqual(severity, ["medium"])
def test_get_severity_exception(self):
"""Test Severity with unknown value, raises exception."""
response_obj = ComponentAnalysisResponseBuilder(
self.eco, self.pkg, self.ver)
response_obj._cves = [{'severity': []}]
severity = response_obj.get_severity()
self.assertListEqual(severity, [])
def test_get_severity_return_low(self):
"""Test Severity Procedure. Severity "low" has higher precedence."""
response_obj = ComponentAnalysisResponseBuilder(self.eco, self.pkg, self.ver)
response_obj._cves = [
{'severity': ['low']},
{'severity': ['low']},
]
severity = response_obj.get_severity()
self.assertListEqual(severity, ['low', 'low'])
