Browser test for LoadDataWithBaseURL reload

content/public/renderer/document_state.h

 // Copyright (c) 2012 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.
 
 #ifndef CONTENT_PUBLIC_RENDERER_DOCUMENT_STATE_H_
 #define CONTENT_PUBLIC_RENDERER_DOCUMENT_STATE_H_
 
 #include <string>
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/supports_user_data.h"
 #include "base/time/time.h"
 #include "content/common/content_export.h"
 #include "net/http/http_response_info.h"
 #include "third_party/WebKit/public/web/WebDataSource.h"
+#include "url/gurl.h"
 
 namespace content {
 
 class NavigationState;
 
 // The RenderView stores an instance of this class in the "extra data" of each
 // WebDataSource (see RenderView::DidCreateDataSource).
 class CONTENT_EXPORT DocumentState
     : NON_EXPORTED_BASE(public blink::WebDataSource::ExtraData),
       public base::SupportsUserData {
@@ skipping 147 matching lines @@
   }
   bool was_referred_by_prefetcher() const {
     return was_referred_by_prefetcher_;
   }
 
   void set_was_after_preconnect_request(bool value) {
     was_after_preconnect_request_ = value;
   }
   bool was_after_preconnect_request() { return was_after_preconnect_request_; }
 
+  // For LoadDataWithBaseURL navigations, |was_load_data_with_base_url_request_|
+  // is set to true and |data_url_| is set to the data URL of the navigation.
+  // Otherwise, |was_load_data_with_base_url_request_| is false and |data_url_|
+  // is empty.
+  void set_was_load_data_with_base_url_request(bool value) {
+    was_load_data_with_base_url_request_ = value;
+  }
+  bool was_load_data_with_base_url_request() const {
+    return was_load_data_with_base_url_request_;
+  }
+  const GURL& data_url() const {
+    return data_url_;
+  }
+  void set_data_url(const GURL& data_url) {
+    data_url_ = data_url;
+  }
+
   // Record the nature of this load, for use when histogramming page load times.
   LoadType load_type() const { return load_type_; }
   void set_load_type(LoadType load_type) { load_type_ = load_type; }
 
   NavigationState* navigation_state() { return navigation_state_.get(); }
   void set_navigation_state(NavigationState* navigation_state);
 
   bool can_load_local_resources() const { return can_load_local_resources_; }
   void set_can_load_local_resources(bool can_load) {
     can_load_local_resources_ = can_load;
@@ skipping 15 matching lines @@
   bool was_alternate_protocol_available_;
   net::HttpResponseInfo::ConnectionInfo connection_info_;
   bool was_fetched_via_proxy_;
   net::HostPortPair proxy_server_;
 
   // A prefetcher is a page that contains link rel=prefetch elements.
   bool was_prefetcher_;
   bool was_referred_by_prefetcher_;
   bool was_after_preconnect_request_;
 
+  bool was_load_data_with_base_url_request_;
+  GURL data_url_;
+
   LoadType load_type_;
 
   scoped_ptr<NavigationState> navigation_state_;
 
   bool can_load_local_resources_;
 };
 
 #endif  // CONTENT_PUBLIC_RENDERER_DOCUMENT_STATE_H_
 
 }  // namespace content