[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 "components/cronet/android/test/cronet_test_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 {
 
-void AddUrlInterceptors(JNIEnv* env, const JavaParamRef<jclass>& jcaller) {
+// Intercept URLRequestJob creation using URLRequestFilter from
+// libcronet_tests.so
+class UrlInterceptorJobFactoryHandle {
+ public:
+  // |jcontext_adapter| points to a URLRequestContextAdapater.
+  UrlInterceptorJobFactoryHandle(jlong jcontext_adapter)
+      : jcontext_adapter_(jcontext_adapter) {
+    TestUtil::RunAfterContextInit(
+        jcontext_adapter,
+        base::Bind(&UrlInterceptorJobFactoryHandle::InitOnNetworkThread,
+                   base::Unretained(this)));
+  }
+  // Should only be called on network thread; other threads should use
+  // ShutDown().
+  ~UrlInterceptorJobFactoryHandle() {
+    DCHECK(
+        TestUtil::GetTaskRunner(jcontext_adapter_)->BelongsToCurrentThread());
+    TestUtil::GetURLRequestContext(jcontext_adapter_)
+        ->set_job_factory(old_job_factory_);
+  }
+
+  void ShutDown() {
+    TestUtil::RunAfterContextInit(
+        jcontext_adapter_,
+        base::Bind(&UrlInterceptorJobFactoryHandle::ShutdownOnNetworkThread,
+                   base::Unretained(this)));
+  }
+
+ private:
+  void InitOnNetworkThread() {
+    net::URLRequestContext* request_context =
+        TestUtil::GetURLRequestContext(jcontext_adapter_);
+    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());
+  }
+
+  void ShutdownOnNetworkThread() { delete this; }
+
+  // The URLRequestContextAdapater this object intercepts from.
+  const jlong jcontext_adapter_;
+  // URLRequestJobFactory previously used in URLRequestContext.
+  const net::URLRequestJobFactory* old_job_factory_;
+  // URLRequestJobFactory inserted during tests to intercept URLRequests with
+  // libcronet's URLRequestFilter.
+  std::unique_ptr<net::URLRequestInterceptingJobFactory> new_job_factory_;
+};
+
+// 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.
+jlong AddUrlInterceptors(JNIEnv* env,
+                         const JavaParamRef<jclass>& jcaller,
+                         jlong jcontext_adapter) {
   net::URLRequestMockDataJob::AddUrlHandler();
   net::URLRequestFailedJob::AddUrlHandler();
   net::URLRequestHangingReadJob::AddUrlHandler();
   net::SSLCertificateErrorJob::AddUrlHandler();
+  return reinterpret_cast<jlong>(
+      new UrlInterceptorJobFactoryHandle(jcontext_adapter));
+}
+
+// Put back the old URLRequestJobFactory into the URLRequestContext.
+void RemoveUrlInterceptorJobFactory(JNIEnv* env,
+                                    const JavaParamRef<jclass>& jcaller,
+                                    jlong jinterceptor_handle) {
+  reinterpret_cast<UrlInterceptorJobFactoryHandle*>(jinterceptor_handle)
+      ->ShutDown();
 }
 
 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)));
@@ skipping 30 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