[container analysis] effective severity (#3478)

daniel-sanche · web-flow · commit c36b63184987 · 2020-04-23T01:13:25.000Z
Use new effective_severity field

fixes b/142836422
diff --git a/container_registry/container_analysis/samples.py b/container_registry/container_analysis/samples.py
@@ -367,7 +367,7 @@ def find_high_severity_vulnerabilities_for_image(resource_url, project_id):
     vulnerabilities = grafeas_client.list_occurrences(project_name, filter_str)
     filtered_list = []
     for v in vulnerabilities:
-        if v.severity == Severity.HIGH or v.severity == Severity.CRITICAL:
+        if v.effective_severity == Severity.HIGH or v.effective_severity == Severity.CRITICAL:
             filtered_list.append(v)
     return filtered_list
 # [END containeranalysis_filter_vulnerability_occurrences]
diff --git a/container_registry/container_analysis/samples_test.py b/container_registry/container_analysis/samples_test.py
@@ -278,6 +278,7 @@ def test_find_high_severity_vulnerabilities(self):
             'note_name': client.note_path(PROJECT_ID, note_id),
             'resource_uri': self.image_url,
             'vulnerability': {
+                'effective_severity': Severity.CRITICAL,
                 'package_issue': [
                     {
                         'affected_cpe_uri': 'your-uri-here',

Original file line number	Diff line number	Diff line change
`@@ -278,6 +278,7 @@ def test_find_high_severity_vulnerabilities(self):`
`278`	`278`	`'note_name': client.note_path(PROJECT_ID, note_id),`
`279`	`279`	`'resource_uri': self.image_url,`
`280`	`280`	`'vulnerability': {`
	`281`	`+ 'effective_severity': Severity.CRITICAL,`
`281`	`282`	`'package_issue': [`
`282`	`283`	`{`
`283`	`284`	`'affected_cpe_uri': 'your-uri-here',`