Cronet: a framework for providing alternative Cronet implementations

components/cronet/android/test/javatests/src/org/chromium/net/CronetTestBase.java

Index: components/cronet/android/test/javatests/src/org/chromium/net/CronetTestBase.java
diff --git a/components/cronet/android/test/javatests/src/org/chromium/net/CronetTestBase.java b/components/cronet/android/test/javatests/src/org/chromium/net/CronetTestBase.java
index c0b8d3c0b3ec465497fc4dae04d51958e2718325..efdbdb4fd7b4e2f42da788998ea2fadee32010f2 100644
--- a/components/cronet/android/test/javatests/src/org/chromium/net/CronetTestBase.java
+++ b/components/cronet/android/test/javatests/src/org/chromium/net/CronetTestBase.java
@@ -18,6 +18,7 @@ import java.lang.annotation.RetentionPolicy;
 import java.lang.annotation.Target;
 import java.lang.reflect.AnnotatedElement;
 import java.net.URL;
+import java.util.List;
 
 /**
  * Base test class for all CronetTest based tests.
@@ -132,10 +133,8 @@ public class CronetTestBase extends AndroidTestCase {
                 super.runTest();
                 if (!method.isAnnotationPresent(OnlyRunNativeCronet.class)) {
                     if (mCronetTestFramework != null) {
-                        ExperimentalCronetEngine.Builder builder =
-                                new ExperimentalCronetEngine.Builder(getContext());
+                        ExperimentalCronetEngine.Builder builder = createJavaEngineBuilder();
                         builder.setUserAgent(UserAgent.from(getContext()));
-                        builder.enableLegacyMode(true);
                         mCronetTestFramework.mCronetEngine = (CronetEngineBase) builder.build();
                         // Make sure that the instantiated engine is JavaCronetEngine.
                         assert mCronetTestFramework.mCronetEngine.getClass()
@@ -152,6 +151,33 @@ public class CronetTestBase extends AndroidTestCase {
         }
     }
 
+    /**
+     * Creates and returns {@link ExperimentalCronetEngine.Builder} that creates
+     * Java (platform) based {@link CronetEngine.Builder}.
+     *
+     * @return the {@code CronetEngine.Builder} that builds Java-based {@code Cronet engine}.
+     */
+    ExperimentalCronetEngine.Builder createJavaEngineBuilder() {
+        CronetProvider provider = findJavaCronetProvider();
+        return (ExperimentalCronetEngine.Builder) provider.createBuilder();
+    }
+
+    /**
+     * Finds {@link CronetProvider} that can be used to create Java-based {@link CronetEngine}.
+     *
+     * @return the found provider.
+     */
+    private CronetProvider findJavaCronetProvider() {
+        List<CronetProvider> availableProviders =
+                CronetProviders.getInstance().getAvailableProviders(getContext());
+        for (CronetProvider provider : availableProviders) {
+            if (provider.getName().equals(CronetProviders.PROVIDER_NAME_PLATFORM)) {
+                return provider;
+            }
+        }
+        throw new AssertionError("Unable to find Java Provider");
+    }
+
     public void assertResponseEquals(UrlResponseInfo expected, UrlResponseInfo actual) {
         assertEquals(expected.getAllHeaders(), actual.getAllHeaders());
         assertEquals(expected.getAllHeadersAsList(), actual.getAllHeadersAsList());