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

Make CacheDataSourceContractTest use FakeDataSource for upstream

Browse Source 
It's more flexible to use FakeDataSource, since it allows to testing
different upstream behaviors (e.g., upstream not being able to resolve
the content length).

PiperOrigin-RevId: 362072899
This commit is contained in:
olly 2021-03-10 17:48:48 +00:00 committed by Ian Baker
parent a9d397a34e
commit e91b8465b8
1 changed files with 31 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
library/core/src/test/java/com/google/android/exoplayer2/upstream/CacheDataSourceContractTest.java
View File

@ -20,6 +20,8 @@ import androidx.annotation.Nullable;
import androidx.test.core.app.ApplicationProvider;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import com.google.android.exoplayer2.testutil.DataSourceContractTest;
import com.google.android.exoplayer2.testutil.FakeDataSet;
import com.google.android.exoplayer2.testutil.FakeDataSource;
import com.google.android.exoplayer2.testutil.TestUtil;
import com.google.android.exoplayer2.upstream.cache.CacheDataSource;
import com.google.android.exoplayer2.upstream.cache.NoOpCacheEvictor;
@ -28,28 +30,36 @@ import com.google.android.exoplayer2.util.Util;
import com.google.common.collect.ImmutableList;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import org.junit.Before;
import org.junit.Rule;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
/** {@link DataSource} contract tests for {@link CacheDataSource}. */
@RunWith(AndroidJUnit4.class)
public class CacheDataSourceContractTest extends DataSourceContractTest {
private static final byte[] DATA = TestUtil.buildTestData(20);
@Rule public final TemporaryFolder tempFolder = new TemporaryFolder();
private Uri simpleUri;
private FileDataSource fileDataSource;
private Uri unknownLengthUri;
private byte[] simpleData;
private byte[] unknownLengthData;
private FakeDataSet fakeDataSet;
private FakeDataSource upstreamDataSource;
@Before
public void setUp() throws IOException {
File file = tempFolder.newFile();
Files.write(Paths.get(file.getAbsolutePath()), DATA);
simpleUri = Uri.fromFile(file);
simpleUri = Uri.parse("test://simple.test");
unknownLengthUri = Uri.parse("test://unknown-length.test");
simpleData = TestUtil.buildTestData(/* length= */ 20);
unknownLengthData = TestUtil.buildTestData(/* length= */ 40);
fakeDataSet =
new FakeDataSet()
.newData(simpleUri)
.appendReadData(simpleData)
.endData()
.newData(unknownLengthUri)
.setSimulateUnknownLength(true)
.appendReadData(unknownLengthData)
.endData();
}
@Override
@ -58,13 +68,18 @@ public class CacheDataSourceContractTest extends DataSourceContractTest {
new TestResource.Builder()
.setName("simple")
.setUri(simpleUri)
.setExpectedBytes(DATA)
.setExpectedBytes(simpleData)
.build(),
new TestResource.Builder()
.setName("unknown length")
.setUri(unknownLengthUri)
.setExpectedBytes(unknownLengthData)
.build());
}
@Override
protected Uri getNotFoundUri() {
return Uri.fromFile(tempFolder.getRoot().toPath().resolve("nonexistent").toFile());
return Uri.parse("test://not-found.test");
}
@Override
@ -73,13 +88,13 @@ public class CacheDataSourceContractTest extends DataSourceContractTest {
Util.createTempDirectory(ApplicationProvider.getApplicationContext(), "ExoPlayerTest");
SimpleCache cache =
new SimpleCache(tempFolder, new NoOpCacheEvictor(), TestUtil.getInMemoryDatabaseProvider());
fileDataSource = new FileDataSource();
return new CacheDataSource(cache, fileDataSource);
upstreamDataSource = new FakeDataSource(fakeDataSet);
return new CacheDataSource(cache, upstreamDataSource);
}
@Override
@Nullable
protected DataSource getTransferListenerDataSource() {
return fileDataSource;
return upstreamDataSource;
}
}