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

components/cronet/android/test/mock_url_request_job_factory.cc

 // 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.
 
 #include "mock_url_request_job_factory.h"
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_string.h"
+#include "base/memory/ptr_util.h"
 #include "jni/MockUrlRequestJobFactory_jni.h"
 #include "net/test/url_request/ssl_certificate_error_job.h"
 #include "net/test/url_request/url_request_failed_job.h"
 #include "net/test/url_request/url_request_hanging_read_job.h"
 #include "net/test/url_request/url_request_mock_data_job.h"
+#include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_filter.h"
+#include "net/url_request/url_request_intercepting_job_factory.h"
 #include "url/gurl.h"
 
 using base::android::JavaParamRef;
 using base::android::ScopedJavaLocalRef;
 
 namespace cronet {
 
+// URLRequestJobFactory previously used in URLRequestContext.
+static const net::URLRequestJobFactory* old_job_factory;
+// URLRequestJobFactory inserted during tests to intercept URLRequests with
+// libcronet's URLRequestFilter.
+static std::unique_ptr<net::URLRequestJobFactory> new_job_factory;
+
 void AddUrlInterceptors(JNIEnv* env, const JavaParamRef<jclass>& jcaller) {
   net::URLRequestMockDataJob::AddUrlHandler();
   net::URLRequestFailedJob::AddUrlHandler();
   net::URLRequestHangingReadJob::AddUrlHandler();
   net::SSLCertificateErrorJob::AddUrlHandler();
 }
 
+// URL interceptors are registered with the URLRequestFilter in
+// libcronet_tests.so.  However tests are run on libcronet.so.  Use the
+// URLRequestFilter in libcronet_tests.so with the URLRequestContext in
+// libcronet.so by installing a URLRequestInterceptingJobFactory
+// that calls into libcronet_tests.so's URLRequestFilter.
+void AddUrlInterceptorJobFactory(JNIEnv* env,
+                                 const JavaParamRef<jclass>& jcaller,
+                                 jlong jrequestContext) {
+  // Intercept |request_context| job creation using URLRequestFilter
+  net::URLRequestContext* request_context =
+      reinterpret_cast<net::URLRequestContext*>(jrequestContext);
+  old_job_factory = request_context->job_factory();
+  new_job_factory.reset(new net::URLRequestInterceptingJobFactory(
+      const_cast<net::URLRequestJobFactory*>(old_job_factory),
+      net::URLRequestFilter::GetInstance()));
+  request_context->set_job_factory(new_job_factory.get());
+}
+
+// Put back the old URLRequestJobFactory into the URLRequestContext.
+void RemoveUrlInterceptorJobFactory(JNIEnv* env,
+                                    const JavaParamRef<jclass>& jcaller,
+                                    jlong jrequestContext) {
+  reinterpret_cast<net::URLRequestContext*>(jrequestContext)
+      ->set_job_factory(old_job_factory);
+  new_job_factory.reset();
+}
+
 ScopedJavaLocalRef<jstring> GetMockUrlWithFailure(
     JNIEnv* jenv,
     const JavaParamRef<jclass>& jcaller,
     jint jphase,
     jint jnet_error) {
   GURL url(net::URLRequestFailedJob::GetMockHttpUrlWithFailurePhase(
       static_cast<net::URLRequestFailedJob::FailurePhase>(jphase),
       static_cast<int>(jnet_error)));
   return base::android::ConvertUTF8ToJavaString(jenv, url.spec());
 }
@@ skipping 28 matching lines @@
     const JavaParamRef<jclass>& jcaller) {
   GURL url(net::URLRequestHangingReadJob::GetMockHttpUrl());
   return base::android::ConvertUTF8ToJavaString(jenv, url.spec());
 }
 
 bool RegisterMockUrlRequestJobFactory(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 }  // namespace cronet