Skip to content

Commit 9e82f00

Browse files
committed
Fix tests
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent 8207a32 commit 9e82f00

File tree

1 file changed

+26
-18
lines changed

1 file changed

+26
-18
lines changed

vulnerabilities/utils.py

Lines changed: 26 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -612,25 +612,33 @@ def compute_content_id(advisory_data):
612612
normalized_data["url"] = advisory_data.url
613613

614614
elif isinstance(advisory_data, AdvisoryData):
615-
normalized_data = {
616-
"aliases": normalize_list(advisory_data.aliases),
617-
"summary": normalize_text(advisory_data.summary),
618-
"affected_packages": [
619-
pkg.to_dict() for pkg in normalize_list(advisory_data.affected_packages) if pkg
620-
],
621-
"weaknesses": normalize_list(advisory_data.weaknesses),
622-
}
623615
if advisory_data.references_v2:
624-
normalized_data["references"] = [
625-
ref.to_dict() for ref in normalize_list(advisory_data.references_v2) if ref
626-
]
627-
normalized_data["severities"] = [
628-
sev.to_dict() for sev in normalize_list(advisory_data.severities) if sev
629-
]
630-
if advisory_data.references:
631-
normalized_data["references"] = [
632-
ref.to_dict() for ref in normalize_list(advisory_data.references) if ref
633-
]
616+
normalized_data = {
617+
"aliases": normalize_list(advisory_data.aliases),
618+
"summary": normalize_text(advisory_data.summary),
619+
"affected_packages": [
620+
pkg.to_dict() for pkg in normalize_list(advisory_data.affected_packages) if pkg
621+
],
622+
"references": [
623+
ref.to_dict() for ref in normalize_list(advisory_data.references_v2) if ref
624+
],
625+
"severities": [
626+
sev.to_dict() for sev in normalize_list(advisory_data.severities) if sev
627+
],
628+
"weaknesses": normalize_list(advisory_data.weaknesses),
629+
}
630+
elif advisory_data.references or advisory_data.references == []:
631+
normalized_data = {
632+
"aliases": normalize_list(advisory_data.aliases),
633+
"summary": normalize_text(advisory_data.summary),
634+
"affected_packages": [
635+
pkg.to_dict() for pkg in normalize_list(advisory_data.affected_packages) if pkg
636+
],
637+
"references": [
638+
ref.to_dict() for ref in normalize_list(advisory_data.references) if ref
639+
],
640+
"weaknesses": normalize_list(advisory_data.weaknesses),
641+
}
634642

635643
normalized_data["url"] = advisory_data.url
636644

0 commit comments

Comments
 (0)