Fix tests

Signed-off-by: Shivam Sandbhor <shivam.sandbhor@gmail.com>

Author: sbs2001

vulnerabilities/admin.py

@@ -29,7 +29,7 @@
     Package,
     Vulnerability,
     VulnerabilityReference,
-    VulnerabilitySeverity
+    VulnerabilitySeverity,
 )
 
 
@@ -57,6 +57,7 @@ class PackageRelatedVulnerabilityAdmin(admin.ModelAdmin):
 class ImporterAdmin(admin.ModelAdmin):
     pass
 
+
 @admin.register(VulnerabilitySeverity)
 class VulnerabilitySeverityAdmin(admin.ModelAdmin):
     pass

vulnerabilities/severity_systems.py

@@ -34,7 +34,7 @@ def as_score(self, value):
         identifier="cvssv2_vector",
         name="CVSSv2 Vector",
         url="https://www.first.org/cvss/v2/",
-        notes="cvssv2 vector, used to get additional info about nature and severity of vulnerability",
+        notes="cvssv2 vector, used to get additional info about nature and severity of vulnerability",  # nopep8
     ),
     "cvssv3": ScoringSystem(
         identifier="cvssv3",
@@ -46,7 +46,7 @@ def as_score(self, value):
         identifier="cvssv3_vector",
         name="CVSSv3 Vector",
         url="https://www.first.org/cvss/v3-0/",
-        notes="cvssv3 vector, used to get additional info about nature and severity of vulnerability",
+        notes="cvssv3 vector, used to get additional info about nature and severity of vulnerability",  # nopep8
     ),
     "rhbs": ScoringSystem(
         identifier="rhbs",

vulnerabilities/tests/test_nvd.py

@@ -29,7 +29,7 @@
 from vulnerabilities.data_source import Reference
 from vulnerabilities.data_source import Advisory
 from vulnerabilities.data_source import VulnerabilitySeverity
-from vulnerabilities.severity_systems import ScoringSystem
+from vulnerabilities.severity_systems import scoring_systems
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 TEST_DATA = os.path.join(BASE_DIR, "test_data/nvd/nvd_test.json")
@@ -149,21 +149,11 @@ def test_to_advisories(self):
                             url="https://nvd.nist.gov/vuln/detail/CVE-2005-4895",
                             severities=[
                                 VulnerabilitySeverity(
-                                    system=ScoringSystem(
-                                        identifier="cvssv2",
-                                        name="CVSSv2 Base Score",
-                                        url="https://www.first.org/cvss/v2/",
-                                        notes="cvssv2 base score",
-                                    ),
+                                    system=scoring_systems["cvssv2"],
                                     value="5.0",
                                 ),
                                 VulnerabilitySeverity(
-                                    system=ScoringSystem(
-                                        identifier="cvssv2_vector",
-                                        name="CVSSv2 Vector",
-                                        url="https://www.first.org/cvss/v2/",
-                                        notes="cvssv2 vector, used to get additional info about nature and severity of vulnerability",
-                                    ),
+                                    system=scoring_systems["cvssv2_vector"],
                                     value="AV:N/AC:L/Au:N/C:N/I:N/A:P",
                                 ),
                             ],

vulnerabilities/tests/test_redhat_importer.py

@@ -103,7 +103,7 @@ def test_to_advisory(self):
                     ),
                     Reference(
                         reference_id="",
-                        url="https://access.redhat.com/hydra/rest/securitydata/cve/CVE-2016-9401.json",
+                        url="https://access.redhat.com/hydra/rest/securitydata/cve/CVE-2016-9401.json",  # nopep8
                         severities=[
                             VulnerabilitySeverity(
                                 system=scoring_systems["cvssv3"],