[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 android.os.ConditionVariable;
+
 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 static CronetEngine sCronetEngine;
+
     /**
      * Sets up URL interceptors.
      */
-    public static void setUp() {
-        nativeAddUrlInterceptors();
+    public static void setUp(final CronetEngine cronetEngine) {
+        sCronetEngine = cronetEngine;
+        CronetTestUtil.prepareNetworkThreadForTesting(cronetEngine);
+
+        // Hop over to network thread and install URL interceptors at the native level.
+        final ConditionVariable done = new ConditionVariable();
+        CronetTestUtil.postToNetworkThread(cronetEngine, new Runnable() {
+            @Override
+            public void run() {
+                long requestContext =
+                        ((CronetUrlRequestContext) cronetEngine).getUrlRequestContextForTesting();
+                nativeAddUrlInterceptors();
+                nativeAddUrlInterceptorJobFactory(requestContext);
+                done.open();
+            }
+        });
+        done.block();
     }
 
     /**
+     * Remove URL Interceptors.
+     */
+    public static void shutdown() {
+        // Hop over to network thread and remove URL interceptors at the native level.
+        final long requestContext =
+                ((CronetUrlRequestContext) sCronetEngine).getUrlRequestContextForTesting();
+        CronetTestUtil.postToNetworkThread(sCronetEngine, new Runnable() {
+            @Override
+            public void run() {
+                nativeRemoveUrlInterceptorJobFactory(requestContext);
+            }
+        });
+
+        CronetTestUtil.cleanupNetorkThreadForTesting();
+        sCronetEngine = null;
+    }
+
+    /**
      * 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 35 matching lines @@
 
     /**
      * 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 void nativeAddUrlInterceptorJobFactory(long requestContext);
+
+    private static native void nativeRemoveUrlInterceptorJobFactory(long requestContext);
+
     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();
 }