diff options
| -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) |