test(CVSS2, CVSS3): add tests for temporal and environmental vector m…

…ethods
RedHatProductSecurity · Sep 28, 2024 · 9e2dca0 · 9e2dca0
1 parent 5dfd6a1
commit 9e2dca0
Show file tree

Hide file tree

Showing 2 changed files with 133 additions and 0 deletions.
diff --git a/tests/test_cvss2.py b/tests/test_cvss2.py
@@ -190,6 +190,62 @@ def test_rh_vector(self):
         v = "ABC/AV:L/AC:H/Au:M/C:C/I:P/A:P"
         self.assertRaises(CVSS2RHMalformedError, CVSS2.from_rh_vector, v)
 
+    def test_temporal_vector(self):
+        """
+        Test for retrieving only the Temporal CVSS Vector.
+        """
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/E:U/RL:U/RC:UR/CDP:ND/TD:ND/CR:L/IR:M/AR:ND"
+        self.assertEqual(
+            "E:U/RL:U/RC:UR",
+            CVSS2(v).temporal_vector(),
+        )
+
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/CDP:ND/TD:ND/CR:L/IR:M/AR:ND"
+        self.assertEqual(
+            "E:ND/RL:ND/RC:ND",
+            CVSS2(v).temporal_vector(),
+        )
+
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/RL:U/RC:UR/CDP:ND/TD:ND/CR:L/IR:M/AR:ND"
+        self.assertEqual(
+            "E:ND/RL:U/RC:UR",
+            CVSS2(v).temporal_vector(),
+        )
+
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/E:U/RC:UR/CDP:ND/TD:ND/CR:L/IR:M/AR:ND"
+        self.assertEqual(
+            "E:U/RL:ND/RC:UR",
+            CVSS2(v).temporal_vector(),
+        )
+
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/E:U/RL:U/CDP:ND/TD:ND/CR:L/IR:M/AR:ND"
+        self.assertEqual(
+            "E:U/RL:U/RC:ND",
+            CVSS2(v).temporal_vector(),
+        )
+
+    def test_environmental_vector(self):
+        """
+        Test for retrieving only the Environmental CVSS Vector.
+        """
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/E:U/RL:U/CDP:N/TD:N/CR:L/IR:M/AR:L"
+        self.assertEqual(
+            "CDP:N/TD:N/CR:L/IR:M/AR:L",
+            CVSS2(v).environmental_vector(),
+        )
+
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/E:U/RL:U"
+        self.assertEqual(
+            "CDP:ND/TD:ND/CR:ND/IR:ND/AR:ND",
+            CVSS2(v).environmental_vector(),
+        )
+
+        v = "AV:A/AC:H/Au:S/C:C/I:C/A:P/E:U/RL:U/TD:N/CR:L/IR:M/AR:L"
+        self.assertEqual(
+            "CDP:ND/TD:N/CR:L/IR:M/AR:L",
+            CVSS2(v).environmental_vector(),
+        )
+
     def test_parse_from_text_cvss2(self):
         """
         Tests for parsing CVSS from text.

diff --git a/tests/test_cvss3.py b/tests/test_cvss3.py
@@ -230,6 +230,83 @@ def test_rh_vector(self):
         v = "ABC/CVSS:3.0/AV:A/AC:H/PR:N/UI:R/S:C/C:L/I:H/A:L"
         self.assertRaises(CVSS3RHMalformedError, CVSS3.from_rh_vector, v)
 
+    def test_temporal_vector(self):
+        """
+        Test for retrieving only the Temporal CVSS Vector.
+        """
+        v = (
+            "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/"
+            "E:H/RL:O/RC:R/CR:H/IR:X/AR:X/MAC:H/MPR:X/MUI:X/MC:L/MA:X"
+        )
+        self.assertEqual(
+            "E:H/RL:O/RC:R",
+            CVSS3(v).temporal_vector(),
+        )
+
+        v = (
+            "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/"
+            "CR:H/IR:X/AR:X/MAC:H/MPR:X/MUI:X/MC:L/MA:X"
+        )
+        self.assertEqual(
+            "E:X/RL:X/RC:X",
+            CVSS3(v).temporal_vector(),
+        )
+
+        v = (
+            "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/"
+            "RL:O/RC:R/CR:H/IR:X/AR:X/MAC:H/MPR:X/MUI:X/MC:L/MA:X"
+        )
+        self.assertEqual(
+            "E:X/RL:O/RC:R",
+            CVSS3(v).temporal_vector(),
+        )
+
+        v = (
+            "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/"
+            "E:H/RC:R/CR:H/IR:X/AR:X/MAC:H/MPR:X/MUI:X/MC:L/MA:X"
+        )
+        self.assertEqual(
+            "E:H/RL:X/RC:R",
+            CVSS3(v).temporal_vector(),
+        )
+
+        v = (
+            "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/"
+            "E:H/RL:O/CR:H/IR:X/AR:X/MAC:H/MPR:X/MUI:X/MC:L/MA:X"
+        )
+        self.assertEqual(
+            "E:H/RL:O/RC:X",
+            CVSS3(v).temporal_vector(),
+        )
+
+    def test_environmental_vector(self):
+        """
+        Test for retrieving only the Environmental CVSS Vector.
+        """
+        v = (
+            "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/"
+            "E:H/RL:O/RC:R/CR:H/IR:X/AR:X/MAC:H/MPR:X/MUI:X/MC:L/MA:X"
+        )
+        self.assertEqual(
+            "CR:H/IR:X/AR:X/MAV:P/MAC:H/MPR:H/MUI:R/MS:C/MC:L/MI:H/MA:N",
+            CVSS3(v).environmental_vector(),
+        )
+
+        v = "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/E:H/RL:O/RC:R"
+        self.assertEqual(
+            "CR:X/IR:X/AR:X/MAV:P/MAC:H/MPR:H/MUI:R/MS:C/MC:H/MI:H/MA:N",
+            CVSS3(v).environmental_vector(),
+        )
+
+        v = (
+            "CVSS:3.0/S:C/C:H/I:H/A:N/AV:P/AC:H/PR:H/UI:R/"
+            "E:H/RL:O/RC:R/MAV:N/MAC:L/MPR:N/MUI:R/MS:C/MC:L/MI:N/MA:H/CR:M/IR:L/AR:M"
+        )
+        self.assertEqual(
+            "CR:M/IR:L/AR:M/MAV:N/MAC:L/MPR:N/MUI:R/MS:C/MC:L/MI:N/MA:H",
+            CVSS3(v).environmental_vector(),
+        )
+
     def test_parse_from_text_cvss3(self):
         i = "CVSS:3.0/AV:N/AC:L/PR:N/UI:N/S:C/C:H/I:H/A:H"
         e = [CVSS3(i)]