From 8b5e1ce0ecd5dbad66ef65c1144a03790aca65a1 Mon Sep 17 00:00:00 2001
From: Christian Studer <christian.studer@circl.lu>
Date: Fri, 20 Dec 2024 10:44:19 +0100
Subject: [PATCH] fix: [tests] Updated test for the cve module following its
 recent changes

---
 tests/test_expansions.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/tests/test_expansions.py b/tests/test_expansions.py
index e091b048..2ff165ee 100644
--- a/tests/test_expansions.py
+++ b/tests/test_expansions.py
@@ -197,9 +197,19 @@ class TestExpansions(unittest.TestCase):
             self.assertIn(self.get_values(response), results)
 
     def test_cve(self):
-        query = {"module": "cve", "vulnerability": "CVE-2010-4444", "config": {"custom_API": "https://vulnerability.circl.lu/api/cve/"}}
+        query = {
+            "module": "cve",
+            "attribute": {
+                "type": "vulnerability",
+                "value": "CVE-2010-4444",
+                "uuid": "82383d84-3016-4d1c-902f-3de0533bfcec"
+            }
+        }
         response = self.misp_modules_post(query)
-        self.assertTrue(self.get_values(response).startswith("Unspecified vulnerability in Oracle Sun Java System Access Manager"))
+        try:
+            self.assertEqual(self.get_first_object_type(response), 'vulnerability')
+        except Exception:
+            print(self.get_errors(response))
 
     def test_cve_advanced(self):
         query = {"module": "cve_advanced",