Index: components/cronet/android/test/javatests/src/org/chromium/net/urlconnection/CronetHttpURLConnectionTest.java |
diff --git a/components/cronet/android/test/javatests/src/org/chromium/net/urlconnection/CronetHttpURLConnectionTest.java b/components/cronet/android/test/javatests/src/org/chromium/net/urlconnection/CronetHttpURLConnectionTest.java |
index c66b4fdbe24289e177b1d027a935970b75d76648..74b4c877f585a8a02ac83b5ac6598f2a0657134c 100644 |
--- a/components/cronet/android/test/javatests/src/org/chromium/net/urlconnection/CronetHttpURLConnectionTest.java |
+++ b/components/cronet/android/test/javatests/src/org/chromium/net/urlconnection/CronetHttpURLConnectionTest.java |
@@ -9,6 +9,7 @@ import android.test.suitebuilder.annotation.SmallTest; |
import org.chromium.base.annotations.SuppressFBWarnings; |
import org.chromium.base.test.util.Feature; |
+import org.chromium.net.CronetEngine; |
import org.chromium.net.CronetTestBase; |
import org.chromium.net.CronetTestFramework; |
import org.chromium.net.MockUrlRequestJobFactory; |
@@ -41,6 +42,7 @@ import java.util.regex.Pattern; |
* See {@link CronetTestBase#runTest()} for details. |
*/ |
public class CronetHttpURLConnectionTest extends CronetTestBase { |
+ private CronetEngine mCronetEngine; |
@Override |
protected void setUp() throws Exception { |
@@ -50,7 +52,8 @@ public class CronetHttpURLConnectionTest extends CronetTestBase { |
CronetTestFramework.LIBRARY_INIT_KEY, |
CronetTestFramework.LibraryInitType.HTTP_URL_CONNECTION, |
}; |
- startCronetTestFrameworkWithUrlAndCommandLineArgs(null, commandLineArgs); |
+ mCronetEngine = startCronetTestFrameworkWithUrlAndCommandLineArgs(null, |
+ commandLineArgs).mCronetEngine; |
assertTrue(NativeTestServer.startNativeTestServer(getContext())); |
} |
@@ -109,7 +112,7 @@ public class CronetHttpURLConnectionTest extends CronetTestBase { |
@OnlyRunCronetHttpURLConnection |
public void testReadTimeout() throws Exception { |
// Add url interceptors. |
- MockUrlRequestJobFactory.setUp(); |
+ MockUrlRequestJobFactory.setUp(mCronetEngine); |
URL url = new URL(MockUrlRequestJobFactory.getMockUrlForHangingRead()); |
HttpURLConnection connection = (HttpURLConnection) url.openConnection(); |
connection.setReadTimeout(1000); |
@@ -707,7 +710,7 @@ public class CronetHttpURLConnectionTest extends CronetTestBase { |
String data = "MyBigFunkyData"; |
int dataLength = data.length(); |
int repeatCount = 100000; |
- MockUrlRequestJobFactory.setUp(); |
+ MockUrlRequestJobFactory.setUp(mCronetEngine); |
URL url = new URL(MockUrlRequestJobFactory.getMockUrlForData(data, repeatCount)); |
HttpURLConnection connection = |
(HttpURLConnection) url.openConnection(); |