Add: Unit tests for models

tests/nvd/models/test_cve_change.py

@@ -0,0 +1,73 @@
+# Copyright (C) 2023 Greenbone AG
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+# pylint: disable=line-too-long
+# ruff: noqa: E501
+
+import unittest
+from datetime import datetime
+from uuid import UUID
+
+from pontos.nvd.models.cve_change import CVEChange, Detail, EventName
+from tests.nvd import get_cve_change_data
+
+
+class CVEChangeTestCase(unittest.TestCase):
+    def test_required_only(self):
+        cve_change = CVEChange.from_dict(get_cve_change_data())
+
+        self.assertEqual(cve_change.cve_id, "CVE-2022-0001")
+        self.assertEqual(cve_change.event_name, "Initial Analysis")
+        self.assertEqual(
+            cve_change.cve_change_id,
+            UUID("5160FDEB-0FF0-457B-AA36-0AEDCAB2522E"),
+        )
+        self.assertEqual(cve_change.source_identifier, "nvd@nist.gov")
+        self.assertEqual(
+            cve_change.created, datetime(2022, 3, 18, 20, 13, 8, 123000)
+        )
+        self.assertEqual(
+            cve_change.details,
+            [
+                Detail(
+                    action="Added",
+                    type="CVSS V2",
+                    new_value="NIST (AV:L/AC:L/Au:N/C:P/I:N/A:N)",
+                ),
+                Detail(
+                    action="Added",
+                    type="CVSS V3.1",
+                    new_value="NIST AV:L/AC:L/PR:L/UI:N/S:C/C:H/I:N/A:N",
+                ),
+                Detail(
+                    action="Changed",
+                    type="Reference Type",
+                    old_value="http://www.openwall.com/lists/oss-security/2022/03/18/2 No Types Assigned",
+                    new_value="http://www.openwall.com/lists/oss-security/2022/03/18/2 Mailing List, Third Party Advisory",
+                ),
+            ],
+        )
+
+
+class EventNameTestCase(unittest.TestCase):
+    def test_init(self):
+        self.assertEqual(
+            EventName("Initial Analysis"), EventName.INITAL_ANALYSIS
+        )
+
+    def test__str__(self):
+        self.assertEqual(str(EventName.INITAL_ANALYSIS), "Initial Analysis")