def test__get_vulnerabilities_badge_Should_ReturnExpected_When_UseShields(
self, badge_patch, *patches):
vulnerabilities_report = {
'metrics': {
'_totals': {
'SEVERITY.UNDEFINED': 1.0,
'SEVERITY.LOW': 1.0,
'SEVERITY.MEDIUM': 1.0,
'SEVERITY.HIGH': 1.0,
}
}
}
result = get_vulnerabilities_badge(vulnerabilities_report,
use_shields=True)
expected_result = 'https://img.shields.io/badge/vulnerabilities-High-red'
self.assertEqual(result, expected_result)
def test__get_vulnerabilities_badge_Should_ReturnExpected_When_HighRed(
self, badge_patch, *patches):
vulnerabilities_report = {
'metrics': {
'_totals': {
'SEVERITY.UNDEFINED': 1.0,
'SEVERITY.LOW': 1.0,
'SEVERITY.MEDIUM': 1.0,
'SEVERITY.HIGH': 1.0,
}
}
}
result = get_vulnerabilities_badge(vulnerabilities_report)
self.assertEqual(result, badge_patch.return_value)
badge_patch.assert_called_once_with('vulnerabilities',
value='High',
default_color='red')
