Move initialization of ChromeURLRequestContexts to the IO thread.

chrome/common/chrome_plugin_unittest.cc

 // Copyright (c) 2006-2008 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.
 // Tests exercising the Chrome Plugin API.
 
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
 #include "chrome/browser/chrome_plugin_host.h"
+#include "chrome/browser/net/url_request_context_getter.h"
 #include "chrome/browser/profile.h"
 #include "chrome/common/chrome_plugin_lib.h"
 #include "chrome/test/chrome_plugin/test_chrome_plugin.h"
 #include "net/base/io_buffer.h"
 #include "net/http/http_response_headers.h"
 #include "net/url_request/url_request_test_job.h"
 #include "net/url_request/url_request_unittest.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 const wchar_t kDocRoot[] = L"chrome/test/data";
 const char kPluginFilename[] = "test_chrome_plugin.dll";
 const int kResponseBufferSize = 4096;
 
+class TestURLRequestContextGetter : public URLRequestContextGetter {
+ public:
+  virtual URLRequestContext* GetURLRequestContext() {
+    if (!context_)
+      context_ = new TestURLRequestContext();
+    return context_;
+  }
+ private:
+  scoped_refptr<URLRequestContext> context_;
+};
+
 class ChromePluginTest : public testing::Test, public URLRequest::Delegate {
  public:
   ChromePluginTest()
       : request_(NULL),
         response_buffer_(new net::IOBuffer(kResponseBufferSize)),
         plugin_(NULL),
         expected_payload_(NULL),
-        request_context_(new TestURLRequestContext()) {
+        request_context_getter_(new TestURLRequestContextGetter()) {
     test_funcs_.test_make_request = NULL;
   }
 
   // Loads/unloads the chrome test plugin.
   void LoadPlugin();
   void UnloadPlugin();
 
   // Runs the test and expects the given payload as a response.  If expectation
   // is NULL, the request is expected to fail.
   void RunTest(const GURL& url, const TestResponsePayload* expected_payload);
 
   // URLRequest::Delegate implementations
   virtual void OnResponseStarted(URLRequest* request);
   virtual void OnReadCompleted(URLRequest* request, int bytes_read);
 
   // Helper called when the URLRequest is done.
   void OnURLRequestComplete();
 
   // testing::Test
   virtual void SetUp() {
     LoadPlugin();
     URLRequest::RegisterProtocolFactory("test", &URLRequestTestJob::Factory);
 
     // We need to setup a default request context in order to issue HTTP
     // requests.
     DCHECK(!Profile::GetDefaultRequestContext());
-    Profile::set_default_request_context(request_context_.get());
+    Profile::set_default_request_context(request_context_getter_.get());
   }
   virtual void TearDown() {
     UnloadPlugin();
     URLRequest::RegisterProtocolFactory("test", NULL);
 
     Profile::set_default_request_context(NULL);
 
     // Clear the request before flushing the message loop since killing the
     // request can result in the generation of more tasks.
     request_.reset();
 
     // Flush the message loop to make Purify happy.
     message_loop_.RunAllPending();
   }
  protected:
   MessageLoopForIO message_loop_;
 
   // Note: we use URLRequest (instead of URLFetcher) because this allows the
   // request to be intercepted.
   scoped_ptr<URLRequest> request_;
   scoped_refptr<net::IOBuffer> response_buffer_;
   std::string response_data_;
 
   ChromePluginLib* plugin_;
   TestFuncParams::PluginFuncs test_funcs_;
   const TestResponsePayload* expected_payload_;
-  scoped_refptr<URLRequestContext> request_context_;
+  scoped_refptr<URLRequestContextGetter> request_context_getter_;
 };
 
 static void STDCALL CPT_Complete(CPRequest* request, bool success,
                                  const std::string& raw_headers,
                                  const std::string& body) {
   GURL url(request->url);
   if (url == GURL(kChromeTestPluginPayloads[0].url)) {
     // This URL should fail, because the plugin should not have intercepted it.
     EXPECT_FALSE(success);
     MessageLoop::current()->Quit();
@@ skipping 180 matching lines @@
 
 // Tests that the plugin does not intercept its own requests.
 TEST_F(ChromePluginTest, DoesNotInterceptOwnRequest) {
   const TestResponsePayload& payload = kChromeTestPluginPayloads[0];
 
   EXPECT_EQ(CPERR_SUCCESS, test_funcs_.test_make_request(
       "GET", GURL(payload.url)));
 
   MessageLoop::current()->Run();
 }