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")); |
} |
} |