| Index: chrome/android/javatests/src/org/chromium/chrome/browser/download/ChromeDownloadDelegateTest.java
|
| diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/download/ChromeDownloadDelegateTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/download/ChromeDownloadDelegateTest.java
|
| index 8c4924f69938c26ae5de0d2ab351eb02ae952be0..77b9c75a5073580dd92fbfbbd54b9d353fd132e4 100644
|
| --- a/chrome/android/javatests/src/org/chromium/chrome/browser/download/ChromeDownloadDelegateTest.java
|
| +++ b/chrome/android/javatests/src/org/chromium/chrome/browser/download/ChromeDownloadDelegateTest.java
|
| @@ -5,29 +5,41 @@
|
| package org.chromium.chrome.browser.download;
|
|
|
| import android.content.Context;
|
| +import android.support.test.InstrumentationRegistry;
|
| import android.support.test.filters.SmallTest;
|
|
|
| +import org.junit.Assert;
|
| +import org.junit.Before;
|
| +import org.junit.Rule;
|
| +import org.junit.Test;
|
| +import org.junit.runner.RunWith;
|
| +
|
| import org.chromium.base.ThreadUtils;
|
| +import org.chromium.base.test.util.CommandLineFlags;
|
| import org.chromium.base.test.util.Feature;
|
| import org.chromium.base.test.util.RetryOnFailure;
|
| import org.chromium.chrome.browser.ChromeActivity;
|
| +import org.chromium.chrome.browser.ChromeSwitches;
|
| import org.chromium.chrome.browser.tab.Tab;
|
| -import org.chromium.chrome.test.ChromeActivityTestCaseBase;
|
| +import org.chromium.chrome.test.ChromeActivityTestRule;
|
| +import org.chromium.chrome.test.ChromeJUnit4ClassRunner;
|
|
|
| import java.util.concurrent.Callable;
|
|
|
| /**
|
| * Tests for ChromeDownloadDelegate class.
|
| */
|
| -public class ChromeDownloadDelegateTest extends ChromeActivityTestCaseBase<ChromeActivity> {
|
| -
|
| - public ChromeDownloadDelegateTest() {
|
| - super(ChromeActivity.class);
|
| - }
|
| +@RunWith(ChromeJUnit4ClassRunner.class)
|
| +@CommandLineFlags.Add({ChromeSwitches.DISABLE_FIRST_RUN_EXPERIENCE,
|
| + ChromeActivityTestRule.DISABLE_NETWORK_PREDICTION_FLAG})
|
| +public class ChromeDownloadDelegateTest {
|
| + @Rule
|
| + public ChromeActivityTestRule<ChromeActivity> mActivityTestRule =
|
| + new ChromeActivityTestRule<>(ChromeActivity.class);
|
|
|
| - @Override
|
| - public void startMainActivity() throws InterruptedException {
|
| - startMainActivityOnBlankPage();
|
| + @Before
|
| + public void setUp() throws InterruptedException {
|
| + mActivityTestRule.startMainActivityOnBlankPage();
|
| }
|
|
|
| /**
|
| @@ -47,36 +59,40 @@ public class ChromeDownloadDelegateTest extends ChromeActivityTestCaseBase<Chrom
|
| * Test to make sure {@link ChromeDownloadDelegate#shouldInterceptContextMenuDownload}
|
| * returns true only for ".dd" or ".dm" extensions with http/https scheme.
|
| */
|
| + @Test
|
| @SmallTest
|
| @Feature({"Download"})
|
| @RetryOnFailure
|
| public void testShouldInterceptContextMenuDownload() throws InterruptedException {
|
| - final Tab tab = getActivity().getActivityTab();
|
| - loadUrl("about:blank");
|
| + final Tab tab = mActivityTestRule.getActivity().getActivityTab();
|
| + mActivityTestRule.loadUrl("about:blank");
|
| ChromeDownloadDelegate delegate = ThreadUtils.runOnUiThreadBlockingNoException(
|
| new Callable<ChromeDownloadDelegate>() {
|
| @Override
|
| public ChromeDownloadDelegate call() {
|
| return new MockChromeDownloadDelegate(
|
| - getInstrumentation().getTargetContext(), tab);
|
| + InstrumentationRegistry.getInstrumentation().getTargetContext(),
|
| + tab);
|
| }
|
| });
|
| - assertFalse(delegate.shouldInterceptContextMenuDownload("file://test/test.html"));
|
| - assertFalse(delegate.shouldInterceptContextMenuDownload("http://test/test.html"));
|
| - assertFalse(delegate.shouldInterceptContextMenuDownload("ftp://test/test.dm"));
|
| - assertFalse(delegate.shouldInterceptContextMenuDownload("data://test.dd"));
|
| - assertFalse(delegate.shouldInterceptContextMenuDownload("http://test.dd"));
|
| - assertFalse(delegate.shouldInterceptContextMenuDownload("http://test/test.dd"));
|
| - assertTrue(delegate.shouldInterceptContextMenuDownload("https://test/test.dm"));
|
| + Assert.assertFalse(delegate.shouldInterceptContextMenuDownload("file://test/test.html"));
|
| + Assert.assertFalse(delegate.shouldInterceptContextMenuDownload("http://test/test.html"));
|
| + Assert.assertFalse(delegate.shouldInterceptContextMenuDownload("ftp://test/test.dm"));
|
| + Assert.assertFalse(delegate.shouldInterceptContextMenuDownload("data://test.dd"));
|
| + Assert.assertFalse(delegate.shouldInterceptContextMenuDownload("http://test.dd"));
|
| + Assert.assertFalse(delegate.shouldInterceptContextMenuDownload("http://test/test.dd"));
|
| + Assert.assertTrue(delegate.shouldInterceptContextMenuDownload("https://test/test.dm"));
|
| }
|
|
|
| + @Test
|
| @SmallTest
|
| @Feature({"Download"})
|
| public void testGetFileExtension() {
|
| - assertEquals("ext", ChromeDownloadDelegate.getFileExtension("", "file.ext"));
|
| - assertEquals("ext", ChromeDownloadDelegate.getFileExtension("http://file.ext", ""));
|
| - assertEquals("txt", ChromeDownloadDelegate.getFileExtension("http://file.ext", "file.txt"));
|
| - assertEquals("txt", ChromeDownloadDelegate.getFileExtension(
|
| - "http://file.ext", "file name.txt"));
|
| + Assert.assertEquals("ext", ChromeDownloadDelegate.getFileExtension("", "file.ext"));
|
| + Assert.assertEquals("ext", ChromeDownloadDelegate.getFileExtension("http://file.ext", ""));
|
| + Assert.assertEquals(
|
| + "txt", ChromeDownloadDelegate.getFileExtension("http://file.ext", "file.txt"));
|
| + Assert.assertEquals(
|
| + "txt", ChromeDownloadDelegate.getFileExtension("http://file.ext", "file name.txt"));
|
| }
|
| }
|
|
|