Fix tests

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/tests/test_data/package_sort/sorted_purls.txt

@@ -21,10 +21,10 @@ pkg:conan/capnproto@0.15.2
 pkg:deb/debian/jackson-databind@2.8.6-1%2Bdeb9u7?distro=stretch
 pkg:deb/debian/jackson-databind@2.8.6-1%2Bdeb9u10?distro=stretch
 pkg:deb/debian/jackson-databind@2.9.8-3%2Bdeb10u4?distro=sid
-pkg:deb/debian/jackson-databind@2.12.1-1%2Bdeb11u1
 pkg:deb/debian/jackson-databind@2.12.1-1%2Bdeb11u1?distro=sid
-pkg:deb/debian/jackson-databind@2.13.2.2-1?distro=sid
+pkg:deb/debian/jackson-databind@2.12.1-1%2Bdeb11u1
 pkg:deb/debian/jackson-databind@2.13.2.2-1?distro=stretch
+pkg:deb/debian/jackson-databind@2.13.2.2-1?distro=sid
 pkg:deb/debian/jackson-databind@2.14.0-1?distro=sid
 pkg:deb/ubuntu/dpkg@1.13.11ubuntu7~proposed
 pkg:deb/ubuntu/dpkg@1.13.11ubuntu7.2
@@ -94,10 +94,10 @@ pkg:pypi/jinja2@2.1.1
 pkg:pypi/jinja2@2.2
 pkg:pypi/jinja2@2.2.1
 pkg:pypi/jinja2@2.10
-pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=11
-pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=12
-pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=13
-pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=2
-pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=5
-pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=7
 pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=9
+pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=7
+pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=5
+pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=2
+pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=13
+pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=12
+pkg:rpm/redhat/openssl@1.0.1e-30.el6_6?arch=11

vulnerabilities/tests/test_view.py

@@ -23,10 +23,9 @@
 from vulnerabilities.models import Vulnerability
 from vulnerabilities.models import VulnerabilitySeverity
 from vulnerabilities.templatetags.url_filters import url_quote_filter
+from vulnerabilities.utils import get_purl_version_class
 from vulnerabilities.views import PackageDetails
 from vulnerabilities.views import PackageSearch
-from vulnerabilities.views import get_purl_version_class
-from vulnerabilities.views import purl_sort_key
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 TEST_DIR = os.path.join(BASE_DIR, "test_data/package_sort")
@@ -202,12 +201,13 @@ def setUp(self):
         for pkg in input_purls:
             real_purl = PackageURL.from_string(pkg)
             attrs = {k: v for k, v in real_purl.to_dict().items() if v}
-            Package.objects.create(**attrs)
+            pkg = Package.objects.create(**attrs)
+            pkg.calculate_version_rank
 
     def test_sorted_queryset(self):
         qs_all = Package.objects.all()
         pkgs_qs_all = list(qs_all)
-        sorted_pkgs_qs_all = sorted(pkgs_qs_all, key=purl_sort_key)
+        sorted_pkgs_qs_all = pkgs_qs_all
 
         pkg_package_urls = [obj.package_url for obj in sorted_pkgs_qs_all]
         sorted_purls = os.path.join(TEST_DIR, "sorted_purls.txt")