Skip to content

Commit b6a7877

Browse files
committed
Fix tests
Signed-off-by: Tushar Goel <[email protected]>
1 parent adf47c6 commit b6a7877

File tree

2 files changed

+7
-18
lines changed

2 files changed

+7
-18
lines changed

vulnerabilities/api.py

Lines changed: 6 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -143,11 +143,12 @@ class VulnSerializerRefsAndSummary(BaseResourceSerializer):
143143
)
144144

145145
references = VulnerabilityReferenceSerializer(many=True, source="vulnerabilityreference_set")
146-
aliases = serializers.ListField(
147-
child=serializers.CharField(),
148-
source="aliases.values_list",
149-
read_only=True
150-
)
146+
147+
aliases = serializers.SerializerMethodField()
148+
149+
def get_aliases(self, obj):
150+
# Assuming `obj.aliases` is a queryset of `Alias` objects
151+
return [alias.alias for alias in obj.aliases.all()]
151152

152153
class Meta:
153154
model = Vulnerability
@@ -225,18 +226,6 @@ class PackageSerializer(BaseResourceSerializer):
225226
next_non_vulnerable_version = serializers.CharField(read_only=True)
226227
latest_non_vulnerable_version = serializers.CharField(read_only=True)
227228

228-
# def get_next_non_vulnerable(self, package):
229-
# next_non_vulnerable = package.fixed_package_details.get("next_non_vulnerable", None)
230-
# if next_non_vulnerable:
231-
# return next_non_vulnerable.version
232-
233-
# latest_non_vulnerable_version = serializers.SerializerMethodField("get_latest_non_vulnerable")
234-
235-
# def get_latest_non_vulnerable(self, package):
236-
# latest_non_vulnerable = package.fixed_package_details.get("latest_non_vulnerable", None)
237-
# if latest_non_vulnerable:
238-
# return latest_non_vulnerable.version
239-
240229
purl = serializers.CharField(source="package_url")
241230

242231
affected_by_vulnerabilities = serializers.SerializerMethodField("get_affected_vulnerabilities")

vulnerabilities/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -710,7 +710,7 @@ def version_class(self):
710710
@cached_property
711711
def current_version(self):
712712
return self.version_class(self.version)
713-
713+
714714
@property
715715
def next_non_vulnerable_version(self):
716716
"""

0 commit comments

Comments
 (0)