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

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

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 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;
 
 /**
  * Helper class to set up url interceptors for testing purposes.
  */
 @JNINamespace("cronet")
 public final class MockUrlRequestJobFactory {
+    private final long mInterceptorHandle;
+    private final CronetTestUtil.NetworkThreadTestConnector mNetworkThreadTestConnector;
+
     /**
      * Sets up URL interceptors.
      */
-    public static void setUp() {
-        nativeAddUrlInterceptors();
+    public MockUrlRequestJobFactory(CronetEngine cronetEngine) {
+        mNetworkThreadTestConnector = new CronetTestUtil.NetworkThreadTestConnector(cronetEngine);
+
+        mInterceptorHandle = nativeAddUrlInterceptors(
+                ((CronetUrlRequestContext) cronetEngine).getUrlRequestContextAdapter());
     }
 
     /**
+     * Remove URL Interceptors.
+     */
+    public void shutdown() {
+        nativeRemoveUrlInterceptorJobFactory(mInterceptorHandle);
+        mNetworkThreadTestConnector.shutdown();
+    }
+
+    /**
      * Constructs a mock URL that hangs or fails at certain phase.
      *
      * @param phase at which request fails. It should be a value in
      *              org.chromium.net.test.FailurePhase.
      * @param netError reported by UrlRequestJob. Passing -1, results in hang.
      */
     public static String getMockUrlWithFailure(int phase, int netError) {
         assertTrue(netError < 0);
         switch (phase) {
             case FailurePhase.START:
@@ skipping 33 matching lines @@
         return nativeGetMockUrlForSSLCertificateError();
     }
 
     /**
      * Constructs a mock URL that will hang when try to read response body from the remote.
      */
     public static String getMockUrlForHangingRead() {
         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);
 
     private static native String nativeGetMockUrlForData(String data,
             int dataRepeatCount);
 
     private static native String nativeGetMockUrlForClientCertificateRequest();
 
     private static native String nativeGetMockUrlForSSLCertificateError();
 
     private static native String nativeGetMockUrlForHangingRead();
 }