diff options
| author | J08nY | 2024-06-10 23:47:30 +0200 |
|---|---|---|
| committer | J08nY | 2024-06-10 23:54:30 +0200 |
| commit | 77bf4d941260c871f4da37ed2f8cc1a04ea1a3b9 (patch) | |
| tree | f232718fa9a9f2cb65ff630880e75fa68e2aa63a | |
| parent | 54c9737a6536123501a012dd21548bb5eae8d599 (diff) | |
| download | sec-certs-77bf4d941260c871f4da37ed2f8cc1a04ea1a3b9.tar.gz sec-certs-77bf4d941260c871f4da37ed2f8cc1a04ea1a3b9.tar.zst sec-certs-77bf4d941260c871f4da37ed2f8cc1a04ea1a3b9.zip | |
Fix NVD CPE Match API usage.
NIST recently lowered the limit on resultsPerPage for the
CPE Match API, hence we have to adapt:
https://nvd.nist.gov/general/news/cpematch-resultsperpage-update
| -rw-r--r-- | src/sec_certs/utils/nvd_dataset_builder.py | 2 | ||||
| -rw-r--r-- | tests/test_nvd_dataset_builder.py | 5 |
2 files changed, 3 insertions, 4 deletions
diff --git a/src/sec_certs/utils/nvd_dataset_builder.py b/src/sec_certs/utils/nvd_dataset_builder.py index fe43bcb3..55132339 100644 --- a/src/sec_certs/utils/nvd_dataset_builder.py +++ b/src/sec_certs/utils/nvd_dataset_builder.py @@ -323,7 +323,7 @@ class CveNvdDatasetBuilder(NvdDatasetBuilder[CVEDataset]): class CpeMatchNvdDatasetBuilder(NvdDatasetBuilder[dict]): _ENDPOINT: Final[str] = "CPEMatch" _ENDPOINT_URL: Final[str] = "https://services.nvd.nist.gov/rest/json/cpematch/2.0" - _RESULTS_PER_PAGE: Final[int] = 5000 + _RESULTS_PER_PAGE: Final[int] = 500 _VERSION_KEYS: Final[list[str]] = [ "versionStartIncluding", "versionStartExcluding", diff --git a/tests/test_nvd_dataset_builder.py b/tests/test_nvd_dataset_builder.py index e8adc845..fda94d1f 100644 --- a/tests/test_nvd_dataset_builder.py +++ b/tests/test_nvd_dataset_builder.py @@ -1,10 +1,9 @@ from datetime import date, datetime, timedelta -from importlib import resources +from importlib.resources import files from typing import Any import pytest -import tests.data.common from sec_certs.configuration import config from sec_certs.dataset import CCDataset, CPEDataset, CVEDataset from sec_certs.utils.nvd_dataset_builder import ( @@ -17,7 +16,7 @@ from sec_certs.utils.nvd_dataset_builder import ( @pytest.fixture(autouse=True) def load_test_config(): - with resources.path(tests.data.common, "settings_tests.yml") as path: + with files("tests.data.common") / "settings_tests.yml" as path: config.load_from_yaml(path) |