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) @RunWith(AndroidJUnit4.class)
public class ResolvingDataSourceContractTest extends DataSourceContractTest { 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 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[] simpleData;
private byte[] differentReportedData;
private FakeDataSet fakeDataSet; private FakeDataSet fakeDataSet;
private FakeDataSource fakeDataSource; private FakeDataSource fakeDataSource;
@Before @Before
public void setUp() { public void setUp() {
simpleData = TestUtil.buildTestData(/* length= */ 20); 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 @Override
@ -50,9 +61,15 @@ public class ResolvingDataSourceContractTest extends DataSourceContractTest {
return ImmutableList.of( return ImmutableList.of(
new TestResource.Builder() new TestResource.Builder()
.setName("simple") .setName("simple")
.setUri(URI) .setUri(REQUESTED_URI)
.setResolvedUri(RESOLVED_URI) .setResolvedUri(RESOLVED_URI)
.setExpectedBytes(simpleData) .setExpectedBytes(simpleData)
.build(),
new TestResource.Builder()
.setName("different-reported")
.setUri(REQUESTED_URI_WITH_DIFFERENT_REPORTED)
.setResolvedUri(REPORTED_URI)
.setExpectedBytes(differentReportedData)
.build()); .build());
} }
@ -69,9 +86,21 @@ public class ResolvingDataSourceContractTest extends DataSourceContractTest {
new Resolver() { new Resolver() {
@Override @Override
public DataSpec resolveDataSpec(DataSpec dataSpec) throws IOException { public DataSpec resolveDataSpec(DataSpec dataSpec) throws IOException {
return URI.equals(dataSpec.uri.normalizeScheme().toString()) switch (dataSpec.uri.normalizeScheme().toString()) {
? dataSpec.buildUpon().setUri(RESOLVED_URI).build() case REQUESTED_URI:
: dataSpec; 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;
} }
}); });
} }