From eecfe09f2c145ec0f7256b5981cc49ba683f8b2e Mon Sep 17 00:00:00 2001 From: XiangRongLin <41164160+XiangRongLin@users.noreply.github.com> Date: Sun, 17 Jan 2021 19:52:58 +0100 Subject: [PATCH] Check for non-existent folder in MockDownloader --- .../newpipe/downloader/MockDownloader.java | 20 ++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/extractor/src/test/java/org/schabi/newpipe/downloader/MockDownloader.java b/extractor/src/test/java/org/schabi/newpipe/downloader/MockDownloader.java index 21019a4ec..888d334df 100644 --- a/extractor/src/test/java/org/schabi/newpipe/downloader/MockDownloader.java +++ b/extractor/src/test/java/org/schabi/newpipe/downloader/MockDownloader.java @@ -27,15 +27,17 @@ class MockDownloader extends Downloader { public MockDownloader(@Nonnull String path) throws IOException { this.path = path; this.mocks = new HashMap<>(); - File folder = new File(path); - for (File file : folder.listFiles()) { - if (file.getName().startsWith(RecordingDownloader.FILE_NAME_PREFIX)) { - final FileReader reader = new FileReader(file); - final TestRequestResponse response = new GsonBuilder() - .create() - .fromJson(reader, TestRequestResponse.class); - reader.close(); - mocks.put(response.getRequest(), response.getResponse()); + final File[] files = new File(path).listFiles(); + if (files != null) { + for (File file : files) { + if (file.getName().startsWith(RecordingDownloader.FILE_NAME_PREFIX)) { + final FileReader reader = new FileReader(file); + final TestRequestResponse response = new GsonBuilder() + .create() + .fromJson(reader, TestRequestResponse.class); + reader.close(); + mocks.put(response.getRequest(), response.getResponse()); + } } } }