From 21a1db0f2d7518914940e4560fd7b0cfd1d06de7 Mon Sep 17 00:00:00 2001 From: Felddy Date: Thu, 29 Aug 2024 12:27:50 -0400 Subject: [PATCH] Implement tests with and without schemas --- tests/test_kevsync.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/tests/test_kevsync.py b/tests/test_kevsync.py index 78ed9a3..c39edc7 100644 --- a/tests/test_kevsync.py +++ b/tests/test_kevsync.py @@ -19,7 +19,7 @@ async def test_connection_motor(db_uri, db_name): assert server_info["ok"] == 1.0, "Direct database ping failed" -async def test_fetch_kev_data(): +async def test_fetch_kev_data_without_schema(): kev_data = await fetch_kev_data(DEFAULT_KEV_URL) assert "vulnerabilities" in kev_data, "Expected 'vulnerabilities' in KEV data" assert ( @@ -27,6 +27,14 @@ async def test_fetch_kev_data(): ), "Expected at least one vulnerability item in KEV data" +async def test_fetch_kev_data_with_schema(): + kev_data = await fetch_kev_data(DEFAULT_KEV_URL, DEFAULT_KEV_SCHEMA_URL) + assert "vulnerabilities" in kev_data, "Expected 'vulnerabilities' in KEV data" + assert ( + len(kev_data["vulnerabilities"]) > 0 + ), "Expected at least one vulnerability item in KEV data" + + async def test_create_kev_doc(): await create_kev_doc(VULN_1) # Attempt to find the newly created document