loliball/media3
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support to run actions into FakeDataSource

Browse Source 
An actions is triggered when the reading reaches action's position. This can be
used to make sure the code is in a certain state while testing.

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=159545923
This commit is contained in:
eguven 2017-06-20 04:43:03 -07:00 committed by Oliver Woodman
parent ce8634507d
commit 1c8f14b5fd
1 changed files with 39 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeDataSource.java
View File

@ -49,6 +49,10 @@ import java.util.HashMap;
* until the source is closed. If the source is closed and re-opened having encountered an error,
* that error will not be thrown again.
*
* <p>Actions are inserted by calling {@link FakeData#appendReadAction(Runnable)}. An actions is
* triggered when the reading reaches action's position. This can be used to make sure the code is
* in a certain state while testing.
*
* <p>Example usage:
*
* <pre>
@ -165,6 +169,9 @@ public final class FakeDataSource implements DataSource {
} else {
currentSegmentIndex++;
}
} else if (current.isActionSegment()) {
currentSegmentIndex++;
current.action.run();
} else {
// Read at most bytesRemaining.
readLength = (int) Math.min(readLength, bytesRemaining);
@ -217,21 +224,41 @@ public final class FakeDataSource implements DataSource {
public final IOException exception;
public final byte[] data;
public final int length;
public final Runnable action;
private boolean exceptionThrown;
private boolean exceptionCleared;
private int bytesRead;
public Segment(byte[] data, IOException exception) {
public Segment(byte[] data) {
this.data = data;
this.length = data.length;
this.exception = null;
this.action = null;
}
public Segment(IOException exception) {
this.data = null;
this.length = 0;
this.exception = exception;
length = data != null ? data.length : 0;
this.action = null;
}
public Segment(Runnable action) {
this.data = null;
this.length = 0;
this.exception = null;
this.action = action;
}
public boolean isErrorSegment() {
return exception != null;
}
public boolean isActionSegment() {
return action != null;
}
}
/** Container of fake data to be served by a {@link FakeDataSource}. */
@ -270,7 +297,7 @@ public final class FakeDataSource implements DataSource {
*/
public FakeData appendReadData(byte[] data) {
Assertions.checkState(data != null && data.length > 0);
segments.add(new Segment(data, null));
segments.add(new Segment(data));
return this;
}
@ -278,7 +305,15 @@ public final class FakeDataSource implements DataSource {
* Appends an error in the underlying data.
*/
public FakeData appendReadError(IOException exception) {
segments.add(new Segment(null, exception));
segments.add(new Segment(exception));
return this;
}
/**
* Appends an action.
*/
public FakeData appendReadAction(Runnable action) {
segments.add(new Segment(action));
return this;
}