diff --git a/vulnerabilities/tests/test_api.py b/vulnerabilities/tests/test_api.py index 119ed2bd2..3444d29f8 100644 --- a/vulnerabilities/tests/test_api.py +++ b/vulnerabilities/tests/test_api.py @@ -15,11 +15,9 @@ from django.test import TestCase from django.test import TransactionTestCase from django.test.client import RequestFactory -from packageurl import PackageURL from rest_framework import status from rest_framework.test import APIClient -from vulnerabilities.api import MinimalPackageSerializer from vulnerabilities.api import PackageSerializer from vulnerabilities.api import VulnerabilityReferenceSerializer from vulnerabilities.models import Alias @@ -449,7 +447,7 @@ def test_api_packages_all_num_queries(self): response = self.csrf_client.get(f"/api/packages/all", format="json").data assert len(response) == 3 - assert response == [ + assert list(response) == [ "pkg:maven/com.fasterxml.jackson.core/jackson-databind@2.12.6.1", "pkg:maven/com.fasterxml.jackson.core/jackson-databind@2.13.1", "pkg:maven/com.fasterxml.jackson.core/jackson-databind@2.13.2",