[Cronet] Test the libcronet that's shipped, not libcronet_test

components/cronet/android/test/src/org/chromium/net/MockUrlRequestJobFactory.java

Index: components/cronet/android/test/src/org/chromium/net/MockUrlRequestJobFactory.java
diff --git a/components/cronet/android/test/src/org/chromium/net/MockUrlRequestJobFactory.java b/components/cronet/android/test/src/org/chromium/net/MockUrlRequestJobFactory.java
index 96750028ea374fcdf297bbdc0fd113dffd9af363..db8dbb4aab502699a5f2c14b3ebfb469d7ae4b8c 100644
--- a/components/cronet/android/test/src/org/chromium/net/MockUrlRequestJobFactory.java
+++ b/components/cronet/android/test/src/org/chromium/net/MockUrlRequestJobFactory.java
@@ -7,6 +7,7 @@ package org.chromium.net;
 import static junit.framework.Assert.assertTrue;
 
 import org.chromium.base.annotations.JNINamespace;
+import org.chromium.net.impl.CronetUrlRequestContext;
 import org.chromium.net.test.FailurePhase;
 
 /**
@@ -14,11 +15,25 @@ import org.chromium.net.test.FailurePhase;
  */
 @JNINamespace("cronet")
 public final class MockUrlRequestJobFactory {
+    private final long mInterceptorHandle;
+    private final CronetTestUtil.PrepareNetworkThread mPrepareNetworkThread;
+
     /**
      * Sets up URL interceptors.
      */
-    public static void setUp() {
-        nativeAddUrlInterceptors();
+    public MockUrlRequestJobFactory(CronetEngine cronetEngine) {
+        mPrepareNetworkThread = new CronetTestUtil.PrepareNetworkThread(cronetEngine);
+
+        mInterceptorHandle = nativeAddUrlInterceptors(
+                ((CronetUrlRequestContext) cronetEngine).getUrlRequestContextAdapter());
+    }
+
+    /**
+     * Remove URL Interceptors.
+     */
+    public void shutdown() {
+        nativeRemoveUrlInterceptorJobFactory(mInterceptorHandle);
+        mPrepareNetworkThread.shutdown();
     }
 
     /**
@@ -75,7 +90,9 @@ public final class MockUrlRequestJobFactory {
         return nativeGetMockUrlForHangingRead();
     }
 
-    private static native void nativeAddUrlInterceptors();
+    private static native long nativeAddUrlInterceptors(long requestContextAdapter);
+
+    private static native void nativeRemoveUrlInterceptorJobFactory(long interceptorHandle);
 
     private static native String nativeGetMockUrlWithFailure(int phase, int netError);