test: Added test for cc, not passing yet

1 files changed, 13 insertions, 1 deletions

diff --git a/tests/cc/test_cc_analysis.py b/tests/cc/test_cc_analysis.py
index fa0eecd7..da922208 100644
--- a/tests/cc/test_cc_analysis.py
+++ b/tests/cc/test_cc_analysis.py
@@ -11,7 +11,7 @@ from sec_certs.dataset import CCDataset
 from sec_certs.dataset.cpe import CPEDataset
 from sec_certs.dataset.cve import CVEDataset
 from sec_certs.sample.cc import CCCertificate
-from sec_certs.sample.cpe import CPE
+from sec_certs.sample.cpe import CPE, CPEConfiguration
 from sec_certs.sample.cve import CVE
 from sec_certs.sample.protection_profile import ProtectionProfile
 from sec_certs.sample.sar import SAR
@@ -105,6 +105,7 @@ def reference_dataset(data_dir) -> CCDataset:
 def transitive_vulnerability_dataset(data_dir) -> CCDataset:
     return CCDataset.from_json(data_dir / "transitive_vulnerability_dataset.json")
 
+
 @pytest.fixture(scope="module")
 def ibm_cpe_configuration() -> CPEConfiguration:
     return CPEConfiguration(
@@ -145,6 +146,17 @@ def cpes_ibm_websphere_app_with_platform() -> set[CPE]:
     }
 
 
+def test_find_related_cves_for_cpe_configuration(
+    cc_dset: CCDataset,
+    cpes_ibm_websphere_app_with_platform: set[CPE],
+    ibm_xss_cve: CVE,
+    random_certificate: CCCertificate,
+):
+    random_certificate.heuristics.cpe_matches = {cve.uri for cve in cpes_ibm_websphere_app_with_platform}
+    cc_dset.compute_related_cves()
+    assert ibm_xss_cve.cve_id == random_certificate.heuristics.related_cves
+
+
 @pytest.fixture
 def random_certificate(cc_dset: CCDataset) -> CCCertificate:
     return cc_dset["ebd276cca70fd723"]