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

Persist DownloadActions in DownloadManager

Browse Source 
-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=172875067
This commit is contained in:
eguven 2017-10-20 06:12:01 -07:00 committed by Oliver Woodman
parent c2d05f4405
commit 9568802c6d
1 changed files with 28 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
library/core/src/androidTest/java/com/google/android/exoplayer2/offline/DownloadManagerTest.java
View File

@ -24,7 +24,9 @@ import com.google.android.exoplayer2.offline.DownloadManager.DownloadTask.State;
import com.google.android.exoplayer2.upstream.DummyDataSource; import com.google.android.exoplayer2.upstream.DummyDataSource;
import com.google.android.exoplayer2.upstream.cache.Cache; import com.google.android.exoplayer2.upstream.cache.Cache;
import com.google.android.exoplayer2.util.ClosedSource; import com.google.android.exoplayer2.util.ClosedSource;
import com.google.android.exoplayer2.util.Util;
import java.io.DataOutputStream; import java.io.DataOutputStream;
import java.io.File;
import java.io.IOException; import java.io.IOException;
import java.util.concurrent.Executors; import java.util.concurrent.Executors;
import org.mockito.Mockito; import org.mockito.Mockito;
@ -40,6 +42,7 @@ public class DownloadManagerTest extends InstrumentationTestCase {
private static final int ASSERT_FALSE_TIME = 1000; private static final int ASSERT_FALSE_TIME = 1000;
private DownloadManager downloadManager; private DownloadManager downloadManager;
private File actionFile;
private ConditionVariable downloadFinishedCondition; private ConditionVariable downloadFinishedCondition;
private Throwable downloadError; private Throwable downloadError;
@ -48,28 +51,20 @@ public class DownloadManagerTest extends InstrumentationTestCase {
super.setUp(); super.setUp();
setUpMockito(this); setUpMockito(this);
actionFile = Util.createTempFile(getInstrumentation().getContext(), "ExoPlayerTest");
downloadManager = new DownloadManager( downloadManager = new DownloadManager(
new DownloaderConstructorHelper(Mockito.mock(Cache.class), DummyDataSource.FACTORY), new DownloaderConstructorHelper(Mockito.mock(Cache.class), DummyDataSource.FACTORY),
Executors.newCachedThreadPool()); Executors.newCachedThreadPool(), actionFile.getAbsolutePath());
downloadFinishedCondition = new ConditionVariable(); downloadFinishedCondition = new ConditionVariable();
downloadManager.setListener(new TestDownloadListener()); downloadManager.setListener(new TestDownloadListener());
} }
class TestDownloadListener implements DownloadListener { @Override
@Override public void tearDown() throws Exception {
public void onStateChange(DownloadManager downloadManager, DownloadTask downloadTask, int state, downloadManager.release();
Throwable error) { actionFile.delete();
if (state == DownloadTask.STATE_ERROR && downloadError == null) { super.tearDown();
downloadError = error;
}
((FakeDownloadAction) downloadTask.getDownloadAction()).onStateChange();
}
@Override
public void onTasksFinished(DownloadManager downloadManager) {
downloadFinishedCondition.open();
}
} }
public void testDownloadActionRuns() throws Throwable { public void testDownloadActionRuns() throws Throwable {
@ -200,6 +195,22 @@ public class DownloadManagerTest extends InstrumentationTestCase {
} }
} }
private class TestDownloadListener implements DownloadListener {
@Override
public void onStateChange(DownloadManager downloadManager, DownloadTask downloadTask, int state,
Throwable error) {
if (state == DownloadTask.STATE_ERROR && downloadError == null) {
downloadError = error;
}
((FakeDownloadAction) downloadTask.getDownloadAction()).onStateChange();
}
@Override
public void onTasksFinished(DownloadManager downloadManager) {
downloadFinishedCondition.open();
}
}
/** /**
* Sets up Mockito for an instrumentation test. * Sets up Mockito for an instrumentation test.
*/ */
@ -235,7 +246,7 @@ public class DownloadManagerTest extends InstrumentationTestCase {
@Override @Override
protected void writeToStream(DataOutputStream output) throws IOException { protected void writeToStream(DataOutputStream output) throws IOException {
throw new UnsupportedOperationException(); // do nothing.
} }
@Override @Override