be-gay-do-crimes
/
NewPipeExtractor
mirror of https://github.com/TeamNewPipe/NewPipeExtractor
1
0
Fork 0
Code Issues Releases Wiki Activity

Prefix mock file from RecordingDownloader with "generated_mock_" 

Only read those files in MockDownloader
This commit is contained in:
XiangRongLin 2021-01-10 20:46:42 +01:00
parent f91916c017
commit 1ea6c6ce54
3 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
extractor/src/test/java/org/schabi/newpipe/downloader/MockDownloader.java
View File

@ -24,12 +24,14 @@ class MockDownloader extends Downloader {
this.mocks = new HashMap<>();
File folder = new File(path);
for (File file : folder.listFiles()) {
final FileReader reader = new FileReader(file);
final TestRequestResponse response = new GsonBuilder()
.create()
.fromJson(reader, TestRequestResponse.class);
reader.close();
mocks.put(response.getRequest(), response.getResponse());
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());
}
}
}

4
extractor/src/test/java/org/schabi/newpipe/downloader/RecordingDownloader.java
View File

@ -19,6 +19,8 @@ import javax.annotation.Nonnull;
class RecordingDownloader extends Downloader {
public final static String FILE_NAME_PREFIX = "generated_mock_";
private int index = 0;
private final String path;
@ -37,7 +39,7 @@ class RecordingDownloader extends Downloader {
Downloader downloader = DownloaderTestImpl.getInstance();
Response response = downloader.execute(request);
File outputFile = new File(path + File.separator + index + ".json");
File outputFile = new File(path + File.separator + FILE_NAME_PREFIX + index + ".json");
index++;
outputFile.createNewFile();
FileWriter writer = new FileWriter(outputFile);

6
extractor/src/test/java/org/schabi/newpipe/downloader/TestRequestResponse.java
View File

@ -4,20 +4,14 @@ import org.schabi.newpipe.extractor.downloader.Request;
import org.schabi.newpipe.extractor.downloader.Response;
final class TestRequestResponse {
private final String comment;
private final Request request;
private final Response response;
public TestRequestResponse(Request request, Response response) {
this.comment = "Auto-generated for tests. See RecordingDownloader";
this.request = request;
this.response = response;
}
public String getComment() {
return comment;
}
public Request getRequest() {
return request;
}