Support handling blob URL and resolve blob references in upload data....

webkit/tools/test_shell/simple_resource_loader_bridge.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.
 //
 // This file contains an implementation of the ResourceLoaderBridge class.
 // The class is implemented using URLRequest, meaning it is a "simple" version
 // that directly issues requests. The more complicated one used in the
 // browser uses IPC.
 //
 // Because URLRequest only provides an asynchronous resource loading API, this
@@ skipping 40 matching lines @@
 #include "net/base/upload_data.h"
 #include "net/http/http_cache.h"
 #include "net/http/http_request_headers.h"
 #include "net/http/http_response_headers.h"
 #include "net/proxy/proxy_service.h"
 #if defined(OS_WIN)
 #include "net/socket/ssl_client_socket_nss_factory.h"
 #endif
 #include "net/url_request/url_request.h"
 #include "webkit/appcache/appcache_interfaces.h"
+#include "webkit/blob/blob_storage_controller.h"
 #include "webkit/glue/resource_loader_bridge.h"
 #include "webkit/tools/test_shell/simple_appcache_system.h"
 #include "webkit/tools/test_shell/simple_socket_stream_bridge.h"
 #include "webkit/tools/test_shell/test_shell_request_context.h"
+#include "webkit/tools/test_shell/test_shell_webblobregistry_impl.h"
 
 using webkit_glue::ResourceLoaderBridge;
 using net::StaticCookiePolicy;
 using net::HttpResponseHeaders;
 
 namespace {
 
 struct TestShellRequestContextParams {
   TestShellRequestContextParams(
       const FilePath& in_cache_path,
@@ skipping 38 matching lines @@
       g_request_context_params = NULL;
     } else {
       g_request_context = new TestShellRequestContext();
       SetAcceptAllCookies(false);
     }
 
     g_request_context->AddRef();
 
     SimpleAppCacheSystem::InitializeOnIOThread(g_request_context);
     SimpleSocketStreamBridge::InitializeOnIOThread(g_request_context);
+    TestShellWebBlobRegistryImpl::InitializeOnIOThread(
+        static_cast<TestShellRequestContext*>(g_request_context)->
+            blob_storage_controller());
   }
 
   virtual void CleanUp() {
     SimpleSocketStreamBridge::Cleanup();
+    TestShellWebBlobRegistryImpl::Cleanup();
     if (g_request_context) {
       g_request_context->Release();
       g_request_context = NULL;
     }
   }
 
   void SetAcceptAllCookies(bool accept_all_cookies) {
     StaticCookiePolicy::Type policy_type = accept_all_cookies ?
         StaticCookiePolicy::ALLOW_ALL_COOKIES :
         StaticCookiePolicy::BLOCK_THIRD_PARTY_COOKIES;
@@ skipping 119 matching lines @@
   void NotifyUploadProgress(uint64 position, uint64 size) {
     if (peer_)
       peer_->OnUploadProgress(position, size);
   }
 
   // --------------------------------------------------------------------------
   // The following methods are called on the io thread.  They correspond to
   // actions performed on the owner's thread.
 
   void AsyncStart(RequestParams* params) {
+    // Might need to resolve the blob references in the upload data.
+    if (params->upload) {
+      static_cast<TestShellRequestContext*>(g_request_context)->
+          blob_storage_controller()->ResolveBlobReferencesInUploadData(
+              params->upload.get());
+    }
+
     request_.reset(new URLRequest(params->url, this));
     request_->set_method(params->method);
     request_->set_first_party_for_cookies(params->first_party_for_cookies);
     request_->set_referrer(params->referrer.spec());
     net::HttpRequestHeaders headers;
     headers.AddHeadersFromString(params->headers);
     request_->SetExtraRequestHeaders(headers);
     request_->set_load_flags(params->load_flags);
     request_->set_upload(params->upload.get());
     request_->set_context(g_request_context);
@@ skipping 299 matching lines @@
       uint64 offset,
       uint64 length,
       const base::Time& expected_modification_time) {
     DCHECK(params_.get());
     if (!params_->upload)
       params_->upload = new net::UploadData();
     params_->upload->AppendFileRange(file_path, offset, length,
                                      expected_modification_time);
   }
 
+  virtual void AppendBlobToUpload(const GURL& blob_url) {
+    DCHECK(params_.get());
+    if (!params_->upload)
+      params_->upload = new net::UploadData();
+    params_->upload->AppendBlob(blob_url);
+  }
+
   virtual void SetUploadIdentifier(int64 identifier) {
     DCHECK(params_.get());
     if (!params_->upload)
       params_->upload = new net::UploadData();
     params_->upload->set_identifier(identifier);
   }
 
   virtual bool Start(Peer* peer) {
     DCHECK(!proxy_);
 
@@ skipping 234 matching lines @@
 
 // static
 scoped_refptr<base::MessageLoopProxy>
     SimpleResourceLoaderBridge::GetIoThread() {
   if (!EnsureIOThread()) {
     LOG(DFATAL) << "Failed to create IO thread.";
     return NULL;
   }
   return g_io_thread->message_loop_proxy();
 }