tests

Signed-off-by: Jan Kowalleck <[email protected]>

Author: jkowalleck

tests/test_model_vulnerability.py

@@ -41,7 +41,7 @@
 from tests import reorder
 
 
-class TestModelVulnerability(TestCase):
+class TestModelVulnerabilitySeverity(TestCase):
 
     def test_v_severity_from_cvss_scores_single_critical(self) -> None:
         self.assertEqual(
@@ -85,13 +85,33 @@ def test_v_severity_from_cvss_scores_multiple_high(self) -> None:
             VulnerabilitySeverity.HIGH
         )
 
+class TestModelVulnerabilityScoreSource(TestCase):
+
+    def test_v_source_parse_other(self) -> None:
+        self.assertEqual(
+            VulnerabilityScoreSource.get_from_vector('loremIpsum'),
+            VulnerabilityScoreSource.OTHER
+        )
+
+    def test_v_source_parse_cvss4_0(self) -> None:
+        self.assertEqual(
+            VulnerabilityScoreSource.get_from_vector('CVSS:4.0/AV:N/AC:L/AT:P/PR:N/UI:P/VC:H/VI:H/VA:H/SC:N/SI:N/SA:N/E:U'),
+            VulnerabilityScoreSource.CVSS_V4
+        )
+
     def test_v_source_parse_cvss3_1(self) -> None:
+        self.assertEqual(
+            VulnerabilityScoreSource.get_from_vector('CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:N/I:N/A:H'),
+            VulnerabilityScoreSource.CVSS_V3_1
+        )
+
+    def test_v_source_parse_cvss3_0(self) -> None:
         self.assertEqual(
             VulnerabilityScoreSource.get_from_vector('CVSS:3.0/AV:L/AC:L/PR:N/UI:R/S:C/C:L/I:N/A:N'),
             VulnerabilityScoreSource.CVSS_V3
         )
 
-    def test_v_source_parse_cvss2_1(self) -> None:
+    def test_v_source_parse_cvss2_0(self) -> None:
         self.assertEqual(
             VulnerabilityScoreSource.get_from_vector('CVSS:2.0/AV:N/AC:L/Au:N/C:N/I:N/A:C'),
             VulnerabilityScoreSource.CVSS_V2
@@ -103,7 +123,7 @@ def test_v_source_parse_owasp_1(self) -> None:
             VulnerabilityScoreSource.OWASP
         )
 
-    def test_v_source_get_localised_vector_cvss3_1(self) -> None:
+    def test_v_source_get_localised_vector_cvss3_0(self) -> None:
         self.assertEqual(
             VulnerabilityScoreSource.CVSS_V3.get_localised_vector(
                 vector='CVSS:3.0/AV:L/AC:L/PR:N/UI:R/S:C/C:L/I:N/A:N'
@@ -166,6 +186,8 @@ def test_v_source_get_localised_vector_other_2(self) -> None:
             'SOMETHING_OR_OTHER'
         )
 
+class TestModelVulnerability(TestCase):
+
     def test_empty_vulnerability(self) -> None:
         v = Vulnerability()
         self.assertIsNone(v.bom_ref.value)