diff --git a/dependency-check-core/src/test/java/org/owasp/dependencycheck/data/nvdcve/BaseDBTestCase.java b/dependency-check-core/src/test/java/org/owasp/dependencycheck/data/nvdcve/BaseDBTestCase.java index fb8f3bb03..db812d114 100644 --- a/dependency-check-core/src/test/java/org/owasp/dependencycheck/data/nvdcve/BaseDBTestCase.java +++ b/dependency-check-core/src/test/java/org/owasp/dependencycheck/data/nvdcve/BaseDBTestCase.java @@ -45,8 +45,10 @@ public abstract class BaseDBTestCase extends BaseTest { public static void ensureDBExists() throws Exception { - java.io.File dataPath = Settings.getDataFile(Settings.KEYS.DATA_DIRECTORY); - if (!dataPath.exists() || (dataPath.isDirectory() && dataPath.listFiles().length < 3)) { + java.io.File dataPath = Settings.getDataDirectory(); + String fileName = String.format(Settings.getString(Settings.KEYS.DB_FILE_NAME), Settings.getString(Settings.KEYS.DB_VERSION)); + java.io.File dataFile = new File(dataPath, fileName); + if (!dataPath.exists() || !dataFile.exists()) { dataPath.mkdirs(); FileInputStream fis = null; ZipInputStream zin = null;