Add tests for num queries for views

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

Author: TG1999

vulnerabilities/tests/test_view.py

@@ -17,9 +17,12 @@
 from univers import versions
 
 from vulnerabilities import models
-from vulnerabilities.models import AffectedByPackageRelatedVulnerability, Alias, FixingPackageRelatedVulnerability, VulnerabilitySeverity
+from vulnerabilities.models import AffectedByPackageRelatedVulnerability
+from vulnerabilities.models import Alias
+from vulnerabilities.models import FixingPackageRelatedVulnerability
 from vulnerabilities.models import Package
 from vulnerabilities.models import Vulnerability
+from vulnerabilities.models import VulnerabilitySeverity
 from vulnerabilities.templatetags.url_filters import url_quote_filter
 from vulnerabilities.views import PackageDetails
 from vulnerabilities.views import PackageSearch
@@ -279,21 +282,11 @@ def test_url_quote_filter(self, input_value, expected_output):
 
 class VulnerabilitySearchTestCaseWithPackages(TestCase):
     def setUp(self):
-        self.vuln1 = Vulnerability.objects.create(
-            vulnerability_id="VCID-1", summary="Vuln 1"
-        )
-        self.vuln2 = Vulnerability.objects.create(
-            vulnerability_id="VCID-2", summary="Vuln 2"
-        )
-        self.vuln3 = Vulnerability.objects.create(
-            vulnerability_id="VCID-3", summary="Vuln 3"
-        )
-        self.vuln4 = Vulnerability.objects.create(
-            vulnerability_id="VCID-4", summary="Vuln 4"
-        )
-        self.vuln5 = Vulnerability.objects.create(
-            vulnerability_id="VCID-5", summary="Vuln 5"
-        )
+        self.vuln1 = Vulnerability.objects.create(vulnerability_id="VCID-1", summary="Vuln 1")
+        self.vuln2 = Vulnerability.objects.create(vulnerability_id="VCID-2", summary="Vuln 2")
+        self.vuln3 = Vulnerability.objects.create(vulnerability_id="VCID-3", summary="Vuln 3")
+        self.vuln4 = Vulnerability.objects.create(vulnerability_id="VCID-4", summary="Vuln 4")
+        self.vuln5 = Vulnerability.objects.create(vulnerability_id="VCID-5", summary="Vuln 5")
 
         self.package1 = Package.objects.create(type="pypi", name="django", version="1.0.0")
         self.package2 = Package.objects.create(type="pypi", name="django", version="2.0.0")
@@ -317,10 +310,14 @@ def setUp(self):
         )
 
         self.severity1 = VulnerabilitySeverity.objects.create(
-            scoring_system="CVSSv3", value="9.8", scoring_elements="AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H"
+            scoring_system="CVSSv3",
+            value="9.8",
+            scoring_elements="AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H",
         )
         self.severity2 = VulnerabilitySeverity.objects.create(
-            scoring_system="CVSSv3", value="7.5", scoring_elements="AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H"
+            scoring_system="CVSSv3",
+            value="7.5",
+            scoring_elements="AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H",
         )
 
         self.vuln1.severities.add(self.severity1)