diff --git a/library/core/src/test/java/com/google/android/exoplayer2/upstream/CacheDataSourceContractTest.java b/library/core/src/test/java/com/google/android/exoplayer2/upstream/CacheDataSourceContractTest.java new file mode 100644 index 0000000000..b75ff45f13 --- /dev/null +++ b/library/core/src/test/java/com/google/android/exoplayer2/upstream/CacheDataSourceContractTest.java @@ -0,0 +1,76 @@ +/* + * Copyright (C) 2021 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.google.android.exoplayer2.upstream; + +import android.net.Uri; +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.TestUtil; +import com.google.android.exoplayer2.upstream.cache.CacheDataSource; +import com.google.android.exoplayer2.upstream.cache.NoOpCacheEvictor; +import com.google.android.exoplayer2.upstream.cache.SimpleCache; +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; + + @Before + public void setUp() throws IOException { + File file = tempFolder.newFile(); + Files.write(Paths.get(file.getAbsolutePath()), DATA); + simpleUri = Uri.fromFile(file); + } + + @Override + protected ImmutableList getTestResources() { + return ImmutableList.of( + new TestResource.Builder() + .setName("simple") + .setUri(simpleUri) + .setExpectedBytes(DATA) + .build()); + } + + @Override + protected Uri getNotFoundUri() { + return Uri.fromFile(tempFolder.getRoot().toPath().resolve("nonexistent").toFile()); + } + + @Override + protected DataSource createDataSource() throws IOException { + File tempFolder = + Util.createTempDirectory(ApplicationProvider.getApplicationContext(), "ExoPlayerTest"); + SimpleCache cache = + new SimpleCache(tempFolder, new NoOpCacheEvictor(), TestUtil.getInMemoryDatabaseProvider()); + return new CacheDataSource(cache, new FileDataSource()); + } +}