sjs/media
1
0
Fork 0
mirror of https://github.com/samsonjs/media.git synced 2026-04-27 15:07:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add FakeDataSource.setTestData() to simplify adding random test data

Browse source 
-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=159676653
This commit is contained in:
eguven 2017-06-21 03:54:00 -07:00 committed by Oliver Woodman
parent 40039ed0a1
commit ed060a8f33
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeDataSource.java
View file

@ -343,26 +343,36 @@ public final class FakeDataSource implements DataSource {
dataMap = new HashMap<>(); dataMap = new HashMap<>();
} }
/** Sets the default data, overwrites if there is one already. */
public FakeData newDefaultData() { public FakeData newDefaultData() {
defaultData = new FakeData(this, null); defaultData = new FakeData(this, null);
return defaultData; return defaultData;
} }
/** Sets random data with the given {@code length} for the given {@code uri}. */
public FakeDataSet setRandomData(String uri, int length) {
return setData(uri, TestUtil.buildTestData(length));
}
/** Sets the given {@code data} for the given {@code uri}. */
public FakeDataSet setData(String uri, byte[] data) {
return newData(uri).appendReadData(data).endData();
}
/** Returns a new {@link FakeData} with the given {@code uri}. */
public FakeData newData(String uri) { public FakeData newData(String uri) {
FakeData data = new FakeData(this, uri); FakeData data = new FakeData(this, uri);
dataMap.put(uri, data); dataMap.put(uri, data);
return data; return data;
} }
public FakeDataSet setData(String uri, byte[] data) { /** Returns the data for the given {@code uri}, or {@code defaultData} if no data is set. */
return newData(uri).appendReadData(data).endData();
}
public FakeData getData(String uri) { public FakeData getData(String uri) {
FakeData data = dataMap.get(uri); FakeData data = dataMap.get(uri);
return data != null ? data : defaultData; return data != null ? data : defaultData;
} }
/** Returns a list of all data including {@code defaultData}. */
public ArrayList<FakeData> getAllData() { public ArrayList<FakeData> getAllData() {
ArrayList<FakeData> fakeDatas = new ArrayList<>(dataMap.values()); ArrayList<FakeData> fakeDatas = new ArrayList<>(dataMap.values());
if (defaultData != null) { if (defaultData != null) {