Skip to content

Commit 51dbfaa

Browse files
committed
feat: Integrate NVD proxy mirror
1 parent 00f8a64 commit 51dbfaa

1 file changed

Lines changed: 19 additions & 2 deletions

File tree

test/test_vulnerabilities.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,26 @@
44
import test
55
from owasp_dt.api.metrics import get_vulnerability_metrics
66
from owasp_dt.api.vulnerability import get_all_vulnerabilities
7+
from owasp_dt_v2.api.extensions import update_extension_config
78
from owasp_dt_v2.api.vuln_data_sources import trigger_vuln_data_source_mirror_run
8-
9-
9+
from owasp_dt_v2.models import UpdateExtensionConfigRequest, UpdateExtensionConfigRequestConfig
10+
11+
12+
def test_change_nvd_mirror(client_v2: owasp_dt.Client):
13+
config = {
14+
"enabled": True,
15+
"cveFeedsUrl": "https://nvd.reiche.world/feeds"
16+
}
17+
request_body = UpdateExtensionConfigRequest(config=UpdateExtensionConfigRequestConfig.from_dict(config))
18+
resp = update_extension_config.sync_detailed(
19+
client=client_v2,
20+
extension_point_name="vuln-data-source",
21+
extension_name="nvd",
22+
body=request_body
23+
)
24+
assert resp.status_code in [200, 304]
25+
26+
@pytest.mark.depends(on=['test_change_nvd_mirror'])
1027
def test_trigger_mirror_nvd(client_v2: owasp_dt.Client):
1128
resp = trigger_vuln_data_source_mirror_run.sync_detailed(client=client_v2, name="nvd")
1229
assert resp.status_code in [202, 409]

0 commit comments

Comments
 (0)