loliball/media
1
0
Fork 0
mirror of https://github.com/androidx/media.git synced 2025-04-30 06:46:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Test ResolvingDataSource resolveReportedUri functionality

Browse Source 
PiperOrigin-RevId: 688102934
This commit is contained in:
ibaker 2024-10-21 05:46:55 -07:00 committed by Copybara-Service
parent 5088e87195
commit 40cd64ab19
1 changed files with 35 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
libraries/datasource/src/test/java/androidx/media3/datasource/ResolvingDataSourceContractTest.java
View File

@ -32,17 +32,28 @@ import org.junit.runner.RunWith;
@RunWith(AndroidJUnit4.class)
public class ResolvingDataSourceContractTest extends DataSourceContractTest {
private static final String URI = "test://simple.test";
private static final String REQUESTED_URI = "test://simple.test";
private static final String RESOLVED_URI = "resolved://simple.resolved";
private static final String REQUESTED_URI_WITH_DIFFERENT_REPORTED =
"test://different.report.test";
private static final String RESOLVED_URI_WITH_DIFFERENT_REPORTED =
"resolved://different.report.test";
private static final String REPORTED_URI = "reported://reported.test";
private byte[] simpleData;
private byte[] differentReportedData;
private FakeDataSet fakeDataSet;
private FakeDataSource fakeDataSource;
@Before
public void setUp() {
simpleData = TestUtil.buildTestData(/* length= */ 20);
fakeDataSet = new FakeDataSet().newData(RESOLVED_URI).appendReadData(simpleData).endData();
differentReportedData = TestUtil.buildTestData(/* length= */ 15);
fakeDataSet =
new FakeDataSet()
.setData(RESOLVED_URI, simpleData)
.setData(RESOLVED_URI_WITH_DIFFERENT_REPORTED, differentReportedData);
}
@Override
@ -50,9 +61,15 @@ public class ResolvingDataSourceContractTest extends DataSourceContractTest {
return ImmutableList.of(
new TestResource.Builder()
.setName("simple")
.setUri(URI)
.setUri(REQUESTED_URI)
.setResolvedUri(RESOLVED_URI)
.setExpectedBytes(simpleData)
.build(),
new TestResource.Builder()
.setName("different-reported")
.setUri(REQUESTED_URI_WITH_DIFFERENT_REPORTED)
.setResolvedUri(REPORTED_URI)
.setExpectedBytes(differentReportedData)
.build());
}
@ -69,9 +86,21 @@ public class ResolvingDataSourceContractTest extends DataSourceContractTest {
new Resolver() {
@Override
public DataSpec resolveDataSpec(DataSpec dataSpec) throws IOException {
return URI.equals(dataSpec.uri.normalizeScheme().toString())
? dataSpec.buildUpon().setUri(RESOLVED_URI).build()
: dataSpec;
switch (dataSpec.uri.normalizeScheme().toString()) {
case REQUESTED_URI:
return dataSpec.buildUpon().setUri(RESOLVED_URI).build();
case REQUESTED_URI_WITH_DIFFERENT_REPORTED:
return dataSpec.buildUpon().setUri(RESOLVED_URI_WITH_DIFFERENT_REPORTED).build();
default:
return dataSpec;
}
}
@Override
public Uri resolveReportedUri(Uri uri) {
return uri.normalizeScheme().toString().equals(RESOLVED_URI_WITH_DIFFERENT_REPORTED)
? Uri.parse(REPORTED_URI)
: uri;
}
});
}